/* kproc.c
 * プロセス処理のサンプル
 */

#define USAGE "Usage: kproc <prog_name> <args>"

#include "apr_general.h"
#include "apr_file_io.h"
#include "apr_errno.h"
#include "apr_env.h"
#include "apr_strings.h"
#include "apr_thread_proc.h"

#include "mystab.h"

int apr_my_main (
  int ac, char **av
  , apr_file_t * astdin
  , apr_file_t * astdout
  , apr_file_t * astderr
  , apr_pool_t * pool
) {

  apr_status_t rv=APR_SUCCESS;
  
  char *prog_name = NULL;
  char **args = NULL;
  
  apr_proc_t proc;
  apr_pool_t *pool2 = NULL;
  
  apr_procattr_t *proc_attr = NULL;
  
  if (ac < 2) {
    apr_file_printf(astderr, "%s\n", USAGE);
    goto _FINALLY_;
  }

  /* 子プロセスが実行するコマンド */

  prog_name = av[1];

  /* 子プロセスの実行するコマンドへの引数配列の準備 */

  if (ac > 1) {
    int i;
    args = (char**)apr_palloc(pool, sizeof(char*)*(ac));
    for (i=0; i<ac-1; i++) { /* args[0] には av[1] が入る */
      args[i] = av[i+1];
    }
    args[i]=NULL; /* 最後はヌル */
    
  }

  /* 子プロセス用引数配列の確認 */

  if (args) {
    char **x=args;
    printf("#prog_name=[%s]\n", prog_name);
    for (; *x; x++) {
      printf("#args[%d]=[%s]\n", x-args, *x);
    }
  }
  
  /* 子プロセス用メモリプール生成 */

  rv=apr_pool_create(&pool2, NULL);
  if (APR_SUCCESS != rv) {
    goto _FINALLY_;
  }

  /* 子プロセスの属性情報生成 */

  rv=apr_procattr_create(&proc_attr, pool);
  if (APR_SUCCESS != rv) {
    goto _FINALLY_;
  }

  /* 子プロセスのコマンドタイプ設定 */

  rv=apr_procattr_cmdtype_set(proc_attr, APR_PROGRAM);
  if (APR_SUCCESS != rv) {
    goto _FINALLY_;
  }

  /* 子プロセスのパイプ準備 */

  rv=apr_procattr_io_set(proc_attr, APR_FULL_BLOCK, APR_FULL_BLOCK, APR_NO_PIPE);
  if (APR_SUCCESS != rv) {
    goto _FINALLY_;
  }

  /* 子プロセスの生成 */

  rv = apr_proc_create(&proc, prog_name, (const char*const*)args, NULL, proc_attr, pool2);
  if (APR_SUCCESS!=rv) {
    goto _FINALLY_;
  }
  
  /* 子プロセスへの送信 */
  
  if (proc.in) {
    apr_file_printf(proc.in, "This is a pen.\n");
    apr_file_printf(proc.in, "That is a pencil.\n");
    apr_file_close(proc.in);
  }
  
  /* 子プロセスからの受信 */

  if (proc.out) {
    #define BUF_SIZE 1024
    char buf[BUF_SIZE];
    while (1) {
      if(APR_EOF == apr_file_eof(proc.out)) {
        break;
      }
      if (APR_SUCCESS == apr_file_gets(buf, BUF_SIZE, proc.out)) {
        apr_file_printf(astdout, "[%s]\n", buf);
      }
    }
    apr_file_close(proc.out);
  }

  /* 子プロセスの終了待ち */

  {
    int exitcode=0;
    apr_exit_why_e why;
    char *tmp = "";
    rv = apr_proc_wait(&proc, &exitcode, &why, APR_WAIT);
    if (APR_CHILD_DONE != rv) {
      goto _FINALLY_;
    }
    
    rv = APR_SUCCESS;
    
    apr_file_printf(astdout, "exitcode=[%d], [", exitcode);
    switch (why) {
      case APR_PROC_EXIT:
      tmp = "process exited normally";
      break;
      case APR_PROC_SIGNAL:
      tmp = "process exited due to a signal";
      break;
      case APR_PROC_SIGNAL_CORE:
      tmp = "process exited and dumped a core file";
    }
    apr_file_printf(astdout, "%s]\n", tmp);
  }
  
  _FINALLY_:

  /* 子プロセス用メモリプールの解放 */
  
  if (pool2) {
    apr_pool_destroy(pool2);
    pool2 = NULL;
  }

  if (APR_SUCCESS != rv) {
    char error_buf[64];
    apr_file_printf(astderr, 
      "ERROR: %s\n", apr_strerror(rv, error_buf, sizeof(error_buf)));

    apr_file_printf(astderr, "failed!\n");
    return 1; /* 異常終了 */

  }

  apr_file_printf(astdout, "\ndone.\n");

  return 0; /* 正常終了 */
  

} /* end of main */