/*

  Copyright 2008 Nathan Phillip Brink, Ethan Zonca, Matt 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/>.

*/

#include <stdio.h> /* sprintf, printf */

#include <stdlib.h> /* malloc, free */

#include "options.h"

int main(int argc, char *argv[])

{

  struct execio *testrem;

  char *execargv[] = 

    {

      "ssh",

      "protofusion.org",

    };

  fprintf(stderr, "testing execio (It shouldn't work) :-)\n");

  fprintf(stderr, "execio_open returns %d\n", execio_open(&testrem, "ssh", execargv));

  return 0;

};

/* this function should probably not exist. asprintf should be used instead of sprintf */

size_t intstrlen(size_t theint)

{

  size_t len = 0; /* if the number is single digit, this will be incremented */

  do

    {

      theint /= 10;

      len ++;

    }

  while(theint > 0);

  return len;

}

/*

  Copyright 2008 Nathan Phillip Brink, Ethan Zonca

  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/>.

*/

#include "execio.h"

#include <unistd.h>

#include <sys/types.h>

#include <sys/wait.h>

#include <signal.h>

#include <malloc.h>

#include <stdio.h>

int execio_open(struct execio **rem, const char *progname, char *const argv[])

{

  /* pipe used to write to child */

  int pipe_write[2];

  /* pipe used to read from child */

  int pipe_read[2];

  pid_t child;

  /* for wait(2) if needed */

  int childstatus;

  int counter;

  int counter2;

  /* create two pipes to facilitate communication with child */

  if(pipe(pipe_write))

    return 1;

  if(pipe(pipe_read))

    {

      close(pipe_write[0]);

      close(pipe_write[1]);

      return 1;

    }

  /* parent */

  child = fork();

  if(child == -1)

    {

      close(pipe_write[0]);

      close(pipe_write[1]);

      close(pipe_read[0]);

      close(pipe_read[1]);

      return 1;

    }

  if(child)

    {

      /* close sides of pipe we won't use */

      close(pipe_write[0]);

      close(pipe_read[1]);

      /* setup execio struct */

      (*rem) = malloc(sizeof(struct execio));

      if(!(*rem))

	{

	  /* we should tell the child we're dead - use wait and close our end of the pipes! */

	  close(pipe_write[1]);

	  close(pipe_read[0]);

	  /* we should probably pass of the wait() call to a thread that just does boring things like that. Especially for when the server tries to connect to other servers... */

	  /* maybe we should just kill the child */

	  kill(child, SIGTERM);

	  /* the waitpid(2) seems to indicate that only when the child is terminated will this wait return. */

	  waitpid(child, &childstatus, 0); 

	}

      (*rem)->pipe_write = pipe_write[1];

      (*rem)->pipe_read = pipe_read[0];

      (*rem)->child = child;

      return 0;

    }

  /* child */

  else

    {

      /* close unused pipes */

      close(pipe_write[1]);

      close(pipe_read[0]);

      /*

	reset stdin, stdout, and stderr to the appropriate files. OK, not stderr :-) 

      */

      dup2(pipe_read[1], 0);

      dup2(pipe_write[0], 1);

      /* 

	 close all other file descriptors. We want to keep 0, 1, and 2 open. We don't know that the last open() or pipe() always gives the highest fd number. However, I will assume that it does. Maybe this is a bad idea:

       */

      counter = pipe_write[0];

      if(counter < pipe_write[1])

	counter = pipe_write[1];

      if(counter < pipe_read[0])

	counter = pipe_read[0];

      if(counter < pipe_read[1])

	counter = pipe_read[1];

      counter2 = 0;

      while(counter > 2)

	{

	  if(!close(counter))

	    counter2 ++; /* record how many descriptors we still had open :-) */

	  counter --;

	}

      /* stderr is the only stream we haven't confiscated atm - just for fun - I will confiscate it later, though, to support parsing error messages */

      /*

	now exec: execvp uses interpreter to find the file to exec

       */

      execvp(progname, argv);

      return 1; /* this line should never be reached because we exec -- unless if the exec returns something bad. Then we'd have to tell execio over the pipe about that somehow... */

    }

}

size_t execio_read(struct execio *eio, void *buf, size_t len)

{

  return 0;

}

size_t execio_write(struct execio *eio, void *buf, size_t len)

{

  return 0;

}

int execio_close(struct execio *eio)

{

  return 0;

}

/*

  Copyright 2008 Nathan Phillip Brink, Ethan Zonca

  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_EXECIO_H

#define _DISTREN_EXECIO_H

/*

  This file tries to abstract away getting a socket/fd that talks to a spawned program

 */

#include <unistd.h>

enum execio_state

  {

    EXECIO_STATE_ERROR,

    EXECIO_STATE_EOF

  };

struct execio

{

  int pipe_write;

  int pipe_read;

  pid_t child;

};

int execio_open(struct execio **rem, const char *progname, char *const argv[]);

/* 

   blocks, 

   returns 0 if len is 0. Otherwise, only returns 0 on error/EOF: use execio_state() to determine

*/

size_t execio_read(struct execio *eio, void *buf, size_t len);

size_t execio_write(struct execio *eio, void *buf, size_t len);

enum execio_state execio_state(struct execio *eio);

/* nonzero return on error */

int execio_close(struct execio *eio);

#endif