/*

  Copyright 2010 Nathan Phillip Brink

  This file is a part of DistRen.

  DistRen is free software: you can redistribute it and/or modify

  it under the terms of the GNU Affero General Public License as published by

  the Free Software Foundation, either version 3 of the License, or

  (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_PROTOCOL_H

#define DISTREN_PROTOCOL_H

#include <stdint.h>

/**

   Server types:

 */

#define DISTREN_SERVERTYPE_SUBMIT (0x1)

#define DISTREN_SERVERTYPE_DISTRIBUTE (0x2)

#define DISTREN_SERVERTYPE_RENDER (0x4)

/**

   This file defines the constants and structs that the client uses to talk to the server and that the servers use to talk to eachother.

 */

/**

   generic, shared requests

 */

enum distren_request_type

  {

    DISTREN_REQUEST_VERSION = 1, /*< identifies the version of software being

				   used by the sender and tells if it is a client or server */

    DISTREN_REQUEST_PING = 2,

    DISTREN_REQUEST_PONG = 3,

    DISTREN_REQUEST_DISCONNECT = 4,

    /**

       client->server only requests

    */

    DISTREN_REQUEST_SUBMIT = 5,

    /**

       anything->server requests

     */

    DISTREN_REQUEST_JOBINFO = 6, /*< retrieves information about a job based on its number */

    /**

       server->anything

     */

    DISTREN_REQUEST_JOBINFO_RESPONSE = 7, /*< returns information about a job */

    /**

       server->server

    */

    DISTREN_REQUEST_RENDERFRAME = 8,

    DISTREN_REQUEST_PROGRESS = 10, /*< tells another server of the progress of the first server's work at rendering */

    DISTREN_REQUEST_GETWORK = 11,

  };

struct distren_request

{

  uint32_t magic;

  uint32_t len;

  uint8_t enumsize;

  enum distren_request_type type;

};

/**

   initializes request

 */

int distren_request_new(struct distren_request **req, uint32_t len, enum distren_request_type type);

/**

   frees request

 */

int distren_request_free(struct distren_request *req);

#endif

    mem->size += realsize;

    mem->memory[mem->size] = 0;

  }

  return realsize;

}

struct _web_memorystruct _web_getrequest(char *url){

  // fprintf(stderr,"Preparing to get request at %s",url);

  CURL *curl;

  CURLcode res;

  struct _web_memorystruct chunk;

  chunk.memory=NULL; /* we expect realloc(NULL, size) to work */

  chunk.size = 0;    /* no data at this point */

  curl = curl_easy_init();

  if(curl) {

    curl_easy_setopt(curl, CURLOPT_URL, url);

    curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, _web_writememorycallback);

    curl_easy_setopt(curl, CURLOPT_WRITEDATA, (void *)&chunk);

    curl_easy_setopt(curl, CURLOPT_USERAGENT, "libcurl-agent/1.0");

    res = curl_easy_perform(curl);

    curl_easy_cleanup(curl);

  }

  /* we're done with libcurl, so clean it up */

  curl_global_cleanup();

  return chunk; // 0 is OK, 1 is 404 or other error

}

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

  char *url;

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

  struct _web_memorystruct data = _web_getrequest(url);

  free(url);

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

  if(data.memory)

    free(data.memory);

}

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

  fprintf(stderr,"Resetting frame %d in job %d on server... ",framenum,jobnum);

  char *url;

  _distren_asprintf(&url,"http://dev.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);

  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://dev.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);

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

  if(data.memory)

    free(data.memory);

}

int _web_getwork(int slavekey, char *slavepass, int *jobnum, int *framenum){

  char *url;

  _distren_asprintf(&url,"http://dev.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")){

    fprintf(stderr,"*** Slave %d does not exist!\n",slavekey);

    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

      *jobnum = atoi(tmp);

      tmp = strtok (NULL, ",");

      if(tmp == NULL)

        return 0; // no work

      *framenum = atoi(tmp);

      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://dev.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);

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

  if(data.memory)

    free(data.memory);

}

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

  int ret;

  int frameToRender = 1;

  char *frame_str;

  _distren_asprintf(&frame_str, "%d", frameToRender); // Render frame 1

  char *output;

  _distren_asprintf(&output, "%s/benchmark#.jpg", datadir); // Where to save benchmark output

  char *realOutput;

  _distren_asprintf(&realOutput, "%s/benchmark%d.jpg", datadir, frameToRender); // Where to save benchmark output

  char *input;

  _distren_asprintf(&input, "%s/benchmark.blend", datadir); // Input file

  fprintf(stderr,"Downloading benchmark data...\n");

  curlget("http://data.distren.org/benchmark.blend", input); // Download to input file location

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

  char *cmd[] = { command, "-b", input, "-o", output, "-f", frame_str, "-t", "0", (char *)NULL }; // arguments for blender

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

  fprintf(stderr,"Running benchmark...\n");

  long startTime;

  long endTime;

  time(&startTime);

  char buf[20];

  struct execio *testrem;

  size_t readlen;

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

  if(ret)

    {

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

      return 1;

    }

  buf[19] = '\0';

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

    {

       buf[readlen] = '\0';

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

    }

  execio_close(testrem);

  time(&endTime);

  struct stat buffer;

  int ostatus = stat(realOutput, &buffer);

  if(ostatus == -1){

    ret = 1; // Return error if output wasn't generated

  }

  else

    remove(output);

  *benchmarkTime = abs(difftime(startTime,endTime));

  float tmp = *benchmarkTime;

  tmp = (1/tmp) * 50000;

  *renderPower = (int)tmp;

  free(realOutput);

  free(frame_str);

  free(input);

  free(output);

  return ret;

}

/*

  Copyright 2010 Nathan Phillip Brink, Ethan Zonca, Matthew Orlando

  This file is a part of DistRen.

  DistRen is free software: you can redistribute it and/or modify

  it under the terms of the GNU Affero General Public License as published by

  the Free Software Foundation, either version 3 of the License, or

  (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);

int exec_blender(char *input, char *output, int frame);

void xmlinit();

void xmlcleanup();

int distren_mkdir_recurse(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();

/* 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);

int _web_getwork(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