t_redirection.cpp

Go to the documentation of this file.

/*****************************************************************************\
*                                                                             *
*  Name   : test_stdio_redirecter                                             *
*  Author : Chris Koeritz                                                     *
*                                                                             *
*  Purpose:                                                                   *
*                                                                             *
*    Tests out the stdio redirection class.                                   *
*                                                                             *
*******************************************************************************
* Copyright (c) 2005-$now By Author.  This program is free software; you can  *
* redistribute it and/or modify it under the terms of the GNU General Public  *
* License as published by the Free Software Foundation; either version 2 of   *
* the License or (at your option) any later version.  This is online at:      *
*     http://www.fsf.org/copyleft/gpl.html                                    *
* Please send any updates to: fred@gruntose.com                               *
\*****************************************************************************/

#include <basis/byte_array.h>
#include <basis/function.h>
#include <basis/istring.h>
#include <basis/log_base.h>
#include <basis/object_base.h>
#include <mechanisms/ithread.h>
#include <opsystem/command_line.h>
#include <opsystem/redirecter.h>
#include <data_struct/static_memory_gremlin.h>
#include <textual/byte_format.h>

#include <stdio.h>

const int READ_INTERVAL = 30;
  // how frequently our thread looks for data.

#undef LOG
#define LOG(to_print) CLASS_EMERGENCY_LOG(program_wide_logger(), to_print)
#undef BASE_LOG
#define BASE_LOG(to_print) program_wide_logger().log(to_print)

HOOPLE_STARTUP_CODE;


class text_grabber : public ithread
{
public:
  text_grabber(stdio_redirecter &examine)
    : ithread(READ_INTERVAL), _stdio(examine) {}

  virtual ~text_grabber() {}

  IMPLEMENT_CLASS_NAME("text_grabber");

  void perform_activity(void *) {
    FUNCDEF("perform_activity");
    byte_array out;
    outcome ret = _stdio.read(out);
    out += '\0';
    if (ret == stdio_redirecter::OKAY) {
      BASE_LOG(istring("out> ") + (char *)out.observe());
    } else if ( (ret != stdio_redirecter::OKAY)
        && (ret != stdio_redirecter::NONE_READY) ) {
      LOG(istring("received error on read: ") + common::outcome_name(ret));
    }

    ret = _stdio.read_stderr(out);
    out += '\0';
    if (ret == stdio_redirecter::OKAY) {
      BASE_LOG(istring("err> ") + (char *)out.observe());
    } else if ( (ret != stdio_redirecter::OKAY)
        && (ret != stdio_redirecter::NONE_READY) ) {
      LOG(istring("received error on read_stderr: ")
          + common::outcome_name(ret));
    }
  }

private:
  stdio_redirecter &_stdio;
};


#undef static_class_name
#define static_class_name() "test_stdio_redirecter"

int print_instructions()
{
  BASE_LOG("\
This program takes two arguments and runs a subprogram with redirected I/O.\n\
  --command cmdname\n\
      provides the program name to run.\n\
  --parms parmlist\n\
      provides the set of command line parameters to pass.\n\
The parmslist should be enclosed in quotes in order to contain spaces.\n\
Any embedded quotes should themselves be quoted as appropriate for the OS.\n");

  return 123;
}

int main(int argc, char *argv[])
{
  FUNCDEF("main");
  command_line cmds(argc, argv);
  istring cmd;
  istring parms;
  if (!cmds.get_value("command", cmd)) return print_instructions();
BASE_LOG(istring("got command: ") + cmd);
  if (!cmds.get_value("parms", parms)) return print_instructions();
BASE_LOG(istring("got parms: ") + parms);

  stdio_redirecter stooge(cmd, parms);
  if (stooge.health() != stdio_redirecter::OKAY) {
    LOG(istring("failed to create the redirecter process: ")
       + common::outcome_name(stooge.health()));
    return 140;
  }

  text_grabber stooge_thread(stooge);  // reads the out and err from app.
  stooge_thread.start(NIL);

  bool first_input = true;

  bool done = false;
  while (!done) {
    BASE_LOG("Enter text to send to process (QUIT to exit):");
    istring in('\0', 257);
    fgets(in.access(), 256, stdin);
    in.shrink();
    if ( (in.length() >= 4) && (in.substring(0, 3) == "QUIT") ) break;
    if (in.length()) {
      int count;
      if (first_input) {
        // password likes a CRLF attached.
        in.zap(in.end(), in.end());
        in += "\r\n";
      }
      outcome ret = stooge.write(byte_array(in.length(), (byte *)in.s()), count);
      if (ret != stdio_redirecter::OKAY) {
        LOG(istring("received error on write: ") + common::outcome_name(ret));
      }
    }
    
  }

  // get last outputs from the program before it goes away.
BASE_LOG("before last scavenge");
  stooge.close_input();  // make sure it knows we're done talking.
  portable::sleep_ms(400);
  stooge_thread.stop();
/*
  outcome ret = stooge.read(out);
  if (ret == stdio_redirecter::OKAY) {
    BASE_LOG(istring("out> ") + (char *)out.observe());
  } else if ( (ret != stdio_redirecter::OKAY) && (ret != stdio_redirecter::NONE_READY) ) {
    LOG(istring("received error on read: ") + common::outcome_name(ret));
  }
*/
BASE_LOG("after last scavenge");

}

#undef static_class_name

---