#!/bin/bash

#

# Startup script for DistRen SimpleSlave

#

# chkconfig: 35 90 12

# description: Distren SimpleSlave Client

# processname: distrensimpleslave

# Source function library

. /etc/rc.d/init.d/functions

prog="DistRen SimpleSlave"

args=""

logfile="/var/log/distrensimpleslave"

lockfile="/var/lock/subsys/distrensimpleslave"

start() {

        echo -n $"Starting $prog: "

        daemon $binary

        touch $lockfile

        success $"simpleslave startup"

        echo

}

stop() {

        echo -n $"Shutting down $prog: "

        killproc $binary

        rm -f $lockfile

        echo

}

# Logic

case "$1" in

  start)

        start

        ;;

  stop)

        stop

  if(!password)

    {

      fprintf(stderr, "password not set, try the -p flag\n");

      return 1;

    }

  if(!hostname)

    {

      fprintf(stderr, "hostname not set\n");

      return 1;

    }

  /* Notifies the user if there no username in .conf */

  if(checkUsername(username))

    return 1;

  if(!strncmp(password, "!password",10))

    {

      fprintf(stderr, "You haven't specified a password. Please edit distrenslave.conf or try the -p flag!\n");

      return 1;

    }

  // Variables needed for main loop

  int jobnum = 0;

  int framenum = 0;

  int slavekey = atoi(username); // @TODO: Make this more friendly

  char *urltoTar;      /* Full URL to the server-side location of job#.tgz */

  char *pathtoTar;     /* Full path to the location of the job#.tgz */

  char *pathtoTardir;

  // Temporary for uncompressed testing

  // char *urltoJobfile;

  char *urltoOutput;   /* Full URL where output is posted */

  char *pathtoOutput;  /* Full path to the output (rendered) file */

  char *pathtoOutdir;  /* Full path to output directory */

  char *pathtoRenderOutput; /* Contains blender framenum placeholder */

  char *urltoJobfile; /* Used if we aren't using compression */

  char *pathtoJob; /* Path to job data folder */

  char *pathtoJobfile; /* Full path to the job's main file */

  char *outputExt = "jpg";     /* Output Extension (e.g., JPG) */

  int haveWork = 0;

  int quit = 0;

  fprintf(stderr,"\nDistRen SimpleSlave Pre-Alpha %s\n- Experimental build: Use at your own risk!\n\n", PACKAGE_VERSION);

    {

      if(slaveBenchmark(datadir, &benchmarkTime, &renderPower))

	{

	  fprintf(stderr,"Benchmark failed! Exiting.\n");

	  return 1;

	}

      else

	{

	  fprintf(stderr,"Benchmark successful, time taken was %d seconds, giving you a render power of %d.\n",

		  benchmarkTime, renderPower);

	  _web_setrenderpower(slavekey, password, renderPower);

	  return 0;

	}

    }

  if(!DEBUG)

    fprintf(stderr, "Running...");

  // Main loop

  while(!quit)

    {

    // request work

    if(DEBUG)

      fprintf(stderr,"Requesting work from %s...\n", hostname);

    /* If we got a frame */

    if(haveWork)

      {

        fprintf(stderr,"\nGot work from server...\n");

        if(DEBUG)

          fprintf(stderr, "Preparing to render frame %d in job %d\n", framenum, jobnum);

        prepareJobPaths(jobnum, framenum, outputExt, datadir, &urltoTar, &pathtoTar, &pathtoTardir, &pathtoJob, &pathtoJobfile, &urltoJobfile, &urltoOutput, &pathtoOutput, &pathtoRenderOutput, &pathtoOutdir);

        //free(outputExt);

        mkdir(pathtoTardir, 0700);

        int dlret = downloadTar(urltoTar, pathtoTar);

        //int dlret = downloadTar(urltoJobfile, pathtoJobfile);

        if(dlret == 0)

          fprintf(stderr,"Data retrieved successfully!\n");

        else if(dlret == 3){

          _web_resetframe(slavekey, password, jobnum, framenum);  // Unassign the frame on the server so other slaves can render it

          return 0; // ouput dir doesn't exist

        }

        else

          if(DEBUG)

            //fprintf(stderr,"Using existing uncompressed data %s...\n", pathtoJobfile);

            fprintf(stderr,"Using existing tarball %s...\n", pathtoTar);

	    fprintf(stderr,"Using already extracted data.\n");

        }

        struct stat tbuffer;

        int tstatus = stat(pathtoJobfile, &tbuffer);

	if(tstatus == -1){

          fprintf(stderr,"The main job file does not exist! Data may be corrupt, or the archive is malformatted.\n");

          return 1;

        }

        /* ignore return because directory may exist already */

        if(DEBUG)

          fprintf(stderr,"Creating output directory %s\n", pathtoOutdir);

        mkdir(pathtoOutdir, 0700);

        if(DEBUG)

          fprintf(stderr,"Marking frame started on server... ");

        _web_startframe(slavekey, password, jobnum, framenum);

        /* Execute blender */

        if(DEBUG){

          fprintf(stderr,"Executing blender on file %s\n", pathtoJobfile);

          fprintf(stderr,"Directing output to file %s\n", pathtoOutput);

        }

          {

            fprintf(stderr,"Error running Blender. Check your installation and/or your PATH.\n");

            _web_resetframe(slavekey, password, jobnum, framenum);  // Unassign the frame on the server so other slaves can render it

            return 1;

          }

        free(pathtoJobfile);

        pathtoJobfile = NULL;

        struct stat buffer;

        int fstatus = stat(pathtoOutput, &buffer);

        if(fstatus == -1){

          fprintf(stderr,"*** %s doesn't exist! Scene may not have camera, or your blender installation is not working.\n", pathtoOutput);

          _web_resetframe(slavekey, password, jobnum, framenum);  // Unassign the frame on the server so other slaves can render it

          return 1;

        }

        else{

          /* Post-execution */

          if(DEBUG)

            fprintf(stderr, "Finished frame %d in job %d, uploading...\n", framenum, jobnum);

          else

            fprintf(stderr,"Finished.\n");

          uploadOutput(pathtoOutput, urltoOutput, jobnum, framenum, slavekey); // @TODO: Handle return value

          free(urltoOutput);

  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

  GNU Affero General Public License for more details.

  You should have received a copy of the GNU Affero General Public License

  along with DistRen.  If not, see <http://www.gnu.org/licenses/>.

*/

#include "common/config.h"

#include "distrenjob.h"

#include "slavefuncs.h"

#include "common/asprintf.h"

#include "common/execio.h"

#include "common/protocol.h"

#include "common/remoteio.h"

#include <curl/curl.h>

#include <curl/easy.h>

#include <curl/types.h>

#include <archive.h>

#include <archive_entry.h>

#include <errno.h>

#include <stdio.h>

#include <string.h>

#include <unistd.h>

#include <stdlib.h>

#include <sys/stat.h>

#include <fcntl.h>

#include <time.h>

// Function debug

#define DEBUG 0

/**

   Grabs the xml DOM node reached by an XPath.

   @param path an XPath that lead to DOM node

   @return the first node associated with the path or NULL if there is no match

 */

xmlNodePtr xml_quickxpath(xmlXPathContextPtr xpathctxt, xmlChar *path)

{

  xmlNodePtr toreturn;

  xmlXPathObjectPtr xmlxpathobjptr;

  xmlxpathobjptr = xmlXPathEval(path, xpathctxt);

  if(!xmlxpathobjptr

     || !xmlxpathobjptr->nodesetval->nodeNr)

    {

      fprintf(stderr, "XPath resolution failed for ``%s'' in ``%s'' (``%s'')\n", path, xpathctxt->doc->name, xpathctxt->doc->URL);

      return (xmlNodePtr)NULL;

    }

  toreturn = *(xmlxpathobjptr->nodesetval->nodeTab);

  xmlXPathFreeObject(xmlxpathobjptr);

    while(fgets(buffer,maxlinelen+2,fp1))

      {

        buff_ptr = buffer;

        while ((find_ptr = strstr(buff_ptr,wordtoreplace)))

        {

           while(buff_ptr < find_ptr)

             fputc((int)*buff_ptr++,fp2);

           fputs(replacewith,fp2);

           buff_ptr += find_len;

         }

         fputs(buff_ptr,fp2);

      }

    rename(fileRepl, fileOrig);

  }

  fclose(fp2);

  fclose(fp1);

  fprintf(stderr,"Wrote conf file...\n");

return 1; // Success

}

/* Executors */

/** Executor function for Blender operations */

{

  int ret;

  char *frame_str;

  _distren_asprintf(&frame_str, "%i", frame);

  char *command = "blender"; // @TODO: We currently expect this to be in PATH

  if(DEBUG)

    fprintf(stderr,"Preparing to execute command: %s -b %s -o %s -f %s\n", command, input, output, frame_str);

  char buf[20];

  struct execio *testrem;

  size_t readlen;

  if(DEBUG)

    fprintf(stderr,"Executing: %s\n", frame_str);

  ret = execio_open(&testrem, command, cmd);

  if(ret)

    {

      fprintf(stderr, "Error running blender\n");

      return 1;

    }

  buf[19] = '\0';

  while(!execio_read(testrem, buf, 19, &readlen))

    {

       buf[readlen] = '\0';

       if(DEBUG)

         fprintf(stderr, "read \"%s\"\n", buf);

    }

  _distren_asprintf(&url,"http://distren.org/slave/act.php?mode=resetframe&slavekey=%d&slavepass=%s&jobnum=%d&framenum=%d", slavekey, slavepass, jobnum, framenum);

  struct _web_memorystruct data = _web_getrequest(url);

  free(url);

  if(DEBUG)

    fprintf(stderr,"%s\n", data.memory);

  if(data.memory)

    free(data.memory);

}

void _web_startframe(int slavekey, char *slavepass, int jobnum, int framenum){

  if(DEBUG)

    fprintf(stderr,"Marking frame %d started on server... ",framenum);

  char *url;

  _distren_asprintf(&url,"http://distren.org/slave/act.php?mode=startframe&slavekey=%d&slavepass=%s&jobnum=%d&framenum=%d", slavekey, slavepass, jobnum, framenum);

  struct _web_memorystruct data = _web_getrequest(url);

  free(url);

  if(DEBUG)

    fprintf(stderr,"%s\n", data.memory);

  if(data.memory)

    free(data.memory);

}

  char *url;

  _distren_asprintf(&url,"http://distren.org/slave/act.php?mode=getwork&slavekey=%d&slavepass=%s", slavekey, slavepass);

  struct _web_memorystruct data = _web_getrequest(url);

  free(url);

  if(!data.memory || !strcmp(data.memory,",")){

    fprintf(stderr,"*** No work available on server! In other news, really weird things are happening. Check it out. You shouldn't be seeing this.\n");

    return 0;

  }

  else if(!strcmp(data.memory, "ERROR_BADKEY")){

    free(data.memory);

    return 0;

  }

  else if(!strcmp(data.memory, "ERROR_NORENDERPOWER")){

    free(data.memory);

    return 0;

  }

  // Compare to PACKAGE_VERSION

  else{

    char *tmp;

    char *serverversion;

    tmp = strtok (data.memory,",,");

    if(tmp != NULL){ // make sure work is available

      tmp = strtok (NULL, ",");

      if(tmp == NULL)

        return 0; // no work

      tmp = strtok (NULL, ",");

      if(tmp == NULL)

        return 0; // no work

      serverversion = tmp;

      // @TODO: This should be called every time, not just on fail.

      if(strcmp(PACKAGE_VERSION,serverversion)){

        fprintf(stderr,"Your distren package is out of date! Please acquire a newer version. (%s local vs %s remote)\n", PACKAGE_VERSION, serverversion);

        return 0;

      }

      if(DEBUG)

        fprintf(stderr,"Software versions: %s local vs %s remote\n", PACKAGE_VERSION, serverversion);

      free(data.memory);

      return 1;

    }

    else

      return 0; // error

  }

}

void _web_setrenderpower(int slavekey, char *slavepass, int renderpower){

  fprintf(stderr,"Setting render power on server... ");

  char *url;

  _distren_asprintf(&url,"http://distren.org/slave/act.php?mode=setrenderpower&slavekey=%d&slavepass=%s&renderpower=%d", slavekey, slavepass, renderpower);

  struct _web_memorystruct data = _web_getrequest(url);

  free(url);

  (at your option) any later version.

  DistRen is distributed in the hope that it will be useful,

  but WITHOUT ANY WARRANTY; without even the implied warranty of

  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

  GNU Affero General Public License for more details.

  You should have received a copy of the GNU Affero General Public License

  along with DistRen.  If not, see <http://www.gnu.org/licenses/>.

*/

#ifndef _DISTREN_SLAVEFUNCS_H

#define _DISTREN_SLAVEFUNCS_H

#include "distrenjob.h"

#include "common/remoteio.h"

#include <curl/curl.h>

#include <libxml/xpath.h>

#include <stdio.h>

struct msg;

int sendSignal(struct remoteio *rem, char signal);

int sendExtSignal(struct remoteio *rem, char signal, char *data);

xmlNodePtr xml_quickxpath(xmlXPathContextPtr xpathctxt, xmlChar *path);

int software_updatecheck();

int delete_jobdata(int jobnum, char *datadir);

size_t curl_writetodisk(void *ptr, size_t size, size_t nmemb, FILE *stream);

CURLcode curlget(char *url, char *out);

CURLcode curlpost(char *filename, char *url, int jobnum, int framenum, int slavekey);

int ssh_keygen();

int register_user(char *username, char *email);

int login_user(char *username);

int conf_replace(char *conffile, char *wordtoreplace, char *replacewith);

void xmlinit();

void xmlcleanup();

int distren_mkdir_recurse(const char *dirname);

int job_build_path(char *filename, unsigned int jobnum);

int downloadTar(char *url, char *destinationPath);

int uploadOutput(char *pathtoOutput, char *urltoOutput, int jobnum, int framenum, int slavekey);

int unpackJob(char *outdir, char *pathtoTar);

void prepareJobPaths(int jobnum, int framenum, char *outputExt, char *datadir, char **urltoTar,char **pathtoTar,char **pathtoTardir,char **pathtoJob, char **pathtoJobfile,char **urltoJobfile,char **urltoOutput,char **pathtoOutput, char **pathtoRenderOutput, char **pathtoOutdir);

int checkUsername(char *username);

void slaveTest();

/* Standard slave */

void finishframe(struct remoteio *rem, int jobnum, int framenum);

void resetframe(struct remoteio *rem, int jobnum, int framenum);

void startframe(struct remoteio *rem, int jobnum, int framenum);

int greet_server(struct remoteio *rem);

int getwork(struct remoteio *rem, int *jobnum, int *framenum);

void setrenderpower(struct remoteio *rem, int renderpower);

int getrenderpower(struct remoteio *rem);

int checkslaveversion(struct remoteio *rem);

/* Simple slave */

struct _web_memorystruct;

void *_web_myrealloc(void *ptr, size_t size);

size_t _web_writememorycallback(void *ptr, size_t size, size_t nmemb, void *data);

struct _web_memorystruct _web_getrequest(char *url);

void _web_finishframe(int slavekey, char *slavepass, int jobnum, int framenum);

void _web_startframe(int slavekey, char *slavepass, int jobnum, int framenum);

void _web_resetframe(int slavekey, char *slavepass, int jobnum, int framenum);

void _web_setrenderpower(int slavekey, char *slavepass, int renderpower);

int slaveBenchmark(char *datadir, int *benchmarkTime, int *renderPower);

#endif