testcode
This commit is contained in:
Jean-Francois Dockes
parent
4b1440b2b1
commit
bc455f365e
1 changed files with 304 additions and 0 deletions
304
src/windows/trexecmd.cpp
Normal file
304
src/windows/trexecmd.cpp
Normal file
|
|
@ -0,0 +1,304 @@
|
|||
#include "execmd.h"
|
||||
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <unistd.h>
|
||||
#include <string.h>
|
||||
#include <signal.h>
|
||||
|
||||
#include <string>
|
||||
#include <iostream>
|
||||
#include <sstream>
|
||||
#include <vector>
|
||||
|
||||
#include "debuglog.h"
|
||||
#include "cancelcheck.h"
|
||||
#include "execmd.h"
|
||||
#include "smallut.h"
|
||||
|
||||
using namespace std;
|
||||
|
||||
|
||||
// Testing the rclexecm protocol outside of recoll. Here we use the
|
||||
// rcldoc.py filter, you can try with rclaudio too, adjust the file arg
|
||||
// accordingly
|
||||
bool exercise_mhexecm(const string& cmdstr, const string& mimetype,
|
||||
vector<string>& files)
|
||||
{
|
||||
ExecCmd cmd;
|
||||
|
||||
vector<string> myparams;
|
||||
|
||||
if (cmd.startExec(cmdstr, myparams, 1, 1) < 0) {
|
||||
cerr << "startExec " << cmdstr << " failed. Missing command?\n";
|
||||
return false;
|
||||
}
|
||||
|
||||
for (vector<string>::const_iterator it = files.begin();
|
||||
it != files.end(); it++) {
|
||||
// Build request message
|
||||
ostringstream obuf;
|
||||
obuf << "Filename: " << (*it).length() << "\n" << (*it);
|
||||
obuf << "Mimetype: " << mimetype.length() << "\n" << mimetype;
|
||||
// Bogus parameter should be skipped by filter
|
||||
obuf << "BogusParam: " << string("bogus").length() << "\n" << "bogus";
|
||||
obuf << "\n";
|
||||
cerr << "SENDING: [" << obuf.str() << "]\n";
|
||||
// Send it
|
||||
if (cmd.send(obuf.str()) < 0) {
|
||||
// The real code calls zapchild here, but we don't need it as
|
||||
// this will be handled by ~ExecCmd
|
||||
//cmd.zapChild();
|
||||
cerr << "send error\n";
|
||||
return false;
|
||||
}
|
||||
|
||||
// Read answer
|
||||
for (int loop=0;;loop++) {
|
||||
string name, data;
|
||||
|
||||
// Code from mh_execm.cpp: readDataElement
|
||||
string ibuf;
|
||||
// Read name and length
|
||||
if (cmd.getline(ibuf) <= 0) {
|
||||
cerr << "getline error\n";
|
||||
return false;
|
||||
}
|
||||
// Empty line (end of message)
|
||||
if (!ibuf.compare("\n")) {
|
||||
cerr << "Got empty line\n";
|
||||
name.clear();
|
||||
break;
|
||||
}
|
||||
|
||||
// Filters will sometimes abort before entering the real
|
||||
// protocol, ie if a module can't be loaded. Check the
|
||||
// special filter error first word:
|
||||
if (ibuf.find("RECFILTERROR ") == 0) {
|
||||
cerr << "Got RECFILTERROR\n";
|
||||
return false;
|
||||
}
|
||||
|
||||
// We're expecting something like Name: len\n
|
||||
vector<string> tokens;
|
||||
stringToTokens(ibuf, tokens);
|
||||
if (tokens.size() != 2) {
|
||||
cerr << "bad line in filter output: [" << ibuf << "]\n";
|
||||
return false;
|
||||
}
|
||||
vector<string>::iterator it = tokens.begin();
|
||||
name = *it++;
|
||||
string& slen = *it;
|
||||
int len;
|
||||
if (sscanf(slen.c_str(), "%d", &len) != 1) {
|
||||
cerr << "bad line in filter output (no len): [" <<
|
||||
ibuf << "]\n";
|
||||
return false;
|
||||
}
|
||||
// Read element data
|
||||
data.erase();
|
||||
if (len > 0 && cmd.receive(data, len) != len) {
|
||||
cerr << "MHExecMultiple: expected " << len <<
|
||||
" bytes of data, got " << data.length() << endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
// Empty element: end of message
|
||||
if (name.empty())
|
||||
break;
|
||||
cerr << "Got name: [" << name << "] data [" << data << "]\n";
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
static char *thisprog;
|
||||
static char usage [] =
|
||||
"trexecmd [-c -r -i -o] cmd [arg1 arg2 ...]\n"
|
||||
" -c : test cancellation (ie: trexecmd -c sleep 1000)\n"
|
||||
" -r : run reexec. Must be separate option.\n"
|
||||
" -i : command takes input\n"
|
||||
" -o : command produces output\n"
|
||||
" If -i is set, we send /etc/group contents to whatever command is run\n"
|
||||
" If -o is set, we print whatever comes out\n"
|
||||
"trexecmd -m <filter> <mimetype> <file> [file ...]: test execm:\n"
|
||||
" <filter> should be the path to an execm filter\n"
|
||||
" <mimetype> the type of the file parameters\n"
|
||||
"trexecmd -w cmd : do the 'which' thing\n"
|
||||
;
|
||||
|
||||
static void Usage(void)
|
||||
{
|
||||
fprintf(stderr, "%s: usage:\n%s", thisprog, usage);
|
||||
exit(1);
|
||||
}
|
||||
|
||||
static int op_flags;
|
||||
#define OPT_MOINS 0x1
|
||||
#define OPT_i 0x4
|
||||
#define OPT_w 0x8
|
||||
#define OPT_c 0x10
|
||||
#define OPT_r 0x20
|
||||
#define OPT_m 0x40
|
||||
#define OPT_o 0x80
|
||||
|
||||
// Data sink for data coming out of the command. We also use it to set
|
||||
// a cancellation after a moment.
|
||||
class MEAdv : public ExecCmdAdvise {
|
||||
public:
|
||||
void newData(int cnt) {
|
||||
if (op_flags & OPT_c) {
|
||||
static int callcnt;
|
||||
if (callcnt++ == 10) {
|
||||
// Just sets the cancellation flag
|
||||
CancelCheck::instance().setCancel();
|
||||
// Would be called from somewhere else and throws an
|
||||
// exception. We call it here for simplicity
|
||||
CancelCheck::instance().checkCancel();
|
||||
}
|
||||
}
|
||||
cerr << "newData(" << cnt << ")" << endl;
|
||||
}
|
||||
};
|
||||
|
||||
// Data provider, used if the -i flag is set
|
||||
class MEPv : public ExecCmdProvide {
|
||||
public:
|
||||
FILE *m_fp;
|
||||
string *m_input;
|
||||
MEPv(string *i)
|
||||
: m_input(i)
|
||||
{
|
||||
m_fp = fopen("/etc/group", "r");
|
||||
}
|
||||
~MEPv() {
|
||||
if (m_fp)
|
||||
fclose(m_fp);
|
||||
}
|
||||
void newData() {
|
||||
char line[1024];
|
||||
if (m_fp && fgets(line, 1024, m_fp)) {
|
||||
m_input->assign((const char *)line);
|
||||
} else {
|
||||
m_input->erase();
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
|
||||
|
||||
ReExec reexec;
|
||||
int main(int argc, char *argv[])
|
||||
{
|
||||
#if 0
|
||||
reexec.init(argc, argv);
|
||||
|
||||
if (0) {
|
||||
// Disabled: For testing reexec arg handling
|
||||
vector<string> newargs;
|
||||
newargs.push_back("newarg");
|
||||
newargs.push_back("newarg1");
|
||||
newargs.push_back("newarg2");
|
||||
newargs.push_back("newarg3");
|
||||
newargs.push_back("newarg4");
|
||||
reexec.insertArgs(newargs, 2);
|
||||
}
|
||||
#endif
|
||||
|
||||
thisprog = argv[0];
|
||||
argc--; argv++;
|
||||
|
||||
while (argc > 0 && **argv == '-') {
|
||||
(*argv)++;
|
||||
if (!(**argv))
|
||||
/* Cas du "adb - core" */
|
||||
Usage();
|
||||
while (**argv)
|
||||
switch (*(*argv)++) {
|
||||
case 'c': op_flags |= OPT_c; break;
|
||||
case 'r': op_flags |= OPT_r; break;
|
||||
case 'w': op_flags |= OPT_w; break;
|
||||
case 'm': op_flags |= OPT_m; break;
|
||||
case 'i': op_flags |= OPT_i; break;
|
||||
case 'o': op_flags |= OPT_o; break;
|
||||
default: Usage(); break;
|
||||
}
|
||||
b1: argc--; argv++;
|
||||
}
|
||||
|
||||
if (argc < 1)
|
||||
Usage();
|
||||
|
||||
string arg1 = *argv++; argc--;
|
||||
vector<string> l;
|
||||
while (argc > 0) {
|
||||
l.push_back(*argv++); argc--;
|
||||
}
|
||||
|
||||
DebugLog::getdbl()->setloglevel(DEBDEB1);
|
||||
DebugLog::setfilename("stderr");
|
||||
#if 0
|
||||
signal(SIGPIPE, SIG_IGN);
|
||||
#endif
|
||||
|
||||
if (op_flags & OPT_w) {
|
||||
// Test "which" method
|
||||
string path;
|
||||
if (ExecCmd::which(arg1, path)) {
|
||||
cout << path << endl;
|
||||
return 0;
|
||||
}
|
||||
return 1;
|
||||
} else if (op_flags & OPT_m) {
|
||||
if (l.size() < 2)
|
||||
Usage();
|
||||
string mimetype = l[0];
|
||||
l.erase(l.begin());
|
||||
return exercise_mhexecm(arg1, mimetype, l) ? 0 : 1;
|
||||
} else {
|
||||
// Default: execute command line arguments
|
||||
ExecCmd mexec;
|
||||
|
||||
// Set callback to be called whenever there is new data
|
||||
// available and at a periodic interval, to check for
|
||||
// cancellation
|
||||
MEAdv adv;
|
||||
mexec.setAdvise(&adv);
|
||||
mexec.setTimeout(5);
|
||||
|
||||
// Stderr output goes there
|
||||
mexec.setStderr("/tmp/trexecStderr");
|
||||
|
||||
// A few environment variables. Check with trexecmd env
|
||||
mexec.putenv("TESTVARIABLE1=TESTVALUE1");
|
||||
mexec.putenv("TESTVARIABLE2=TESTVALUE2");
|
||||
mexec.putenv("TESTVARIABLE3=TESTVALUE3");
|
||||
|
||||
string input, output;
|
||||
MEPv pv(&input);
|
||||
|
||||
string *ip = 0;
|
||||
if (op_flags & OPT_i) {
|
||||
ip = &input;
|
||||
mexec.setProvide(&pv);
|
||||
}
|
||||
string *op = 0;
|
||||
if (op_flags & OPT_o) {
|
||||
op = &output;
|
||||
}
|
||||
|
||||
int status = -1;
|
||||
try {
|
||||
status = mexec.doexec(arg1, l, ip, op);
|
||||
} catch (CancelExcept) {
|
||||
cerr << "CANCELLED" << endl;
|
||||
}
|
||||
|
||||
fprintf(stderr, "Status: 0x%x\n", status);
|
||||
if (op_flags & OPT_o) {
|
||||
cout << output;
|
||||
}
|
||||
exit (status >> 8);
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue