1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65
|
/* cvm/module_command.c - CVM command module main routine
* Copyright (C)2006 Bruce Guenter <bruce@untroubled.org>
*
* 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 program 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*/
#include <unistd.h>
#include "module.h"
static int read_input(void)
{
size_t rd;
cvm_module_inbuflen = 0;
for (cvm_module_inbuflen = 0;
cvm_module_inbuflen <= BUFSIZE;
cvm_module_inbuflen += rd) {
if ((rd = read(0, cvm_module_inbuffer+cvm_module_inbuflen,
BUFSIZE-cvm_module_inbuflen)) == (unsigned)-1)
return CVME_IO;
if (rd == 0) break;
}
return 0;
}
static int write_output(void)
{
size_t wr;
char* ptr;
for (ptr = cvm_module_outbuffer;
cvm_module_outbuflen > 0;
cvm_module_outbuflen -= wr, ptr += wr) {
if ((wr = write(1, ptr, cvm_module_outbuflen)) == (unsigned)-1
|| wr == 0)
return CVME_IO;
}
return 0;
}
extern void usage(void);
int command_main(void)
{
int code;
int wcode;
if ((code = cvm_module_init()) != 0) return code;
if ((code = read_input()) != 0) { cvm_module_stop(); return code; }
code = cvm_module_handle_request();
cvm_module_fact_end(code);
if ((wcode = write_output()) != 0 && code == 0) code = wcode;
cvm_module_stop();
return code & CVME_MASK;
}
|