void io_send_queue(io_t * io, const char format[], ...) {
va_list arg_list;
char string[StringSize];
int len;
ASSERT(io_is_ok(io));
ASSERT(format!=NULL);
ASSERT(io->out_fd>=0);
// format
va_start(arg_list,format);
vsprintf(string,format,arg_list);
va_end(arg_list);
// append string to buffer
len = strlen(string);
if (io->out_size + len > BufferSize-2) my_fatal("io_send_queue(): buffer overflow\n");
memcpy(&io->out_buffer[io->out_size],string,len);
io->out_size += len;
ASSERT(io->out_size>=0&&io->out_size<=BufferSize-2);
}
bool engine_is_ok(const engine_t * engine) {
if (engine == NULL) return false;
if (!io_is_ok(engine->io)) return false;
return true;
}
bool io_get_line(io_t * io, char string[], int size) {
int src, dst;
int c;
ASSERT(io_is_ok(io));
ASSERT(string!=NULL);
ASSERT(size>=256);
src = 0;
dst = 0;
while (true) {
// test for end of buffer
if (src >= io->in_size) {
if (io->in_eof) {
my_log("< %s EOF\n",io->name);
return false;
} else {
my_fatal("io_get_line(): no EOL in buffer\n");
}
}
// test for end of string
if (dst >= size) my_fatal("io_get_line(): buffer overflow\n");
// copy the next character
c = io->in_buffer[src++];
if (c == LF) { // LF => line complete
string[dst] = '\0';
break;
} else if (c != CR) { // skip CRs
string[dst++] = c;
}
}
// shift the buffer
ASSERT(src>0);
io->in_size -= src;
ASSERT(io->in_size>=0);
if (io->in_size > 0) memmove(&io->in_buffer[0],&io->in_buffer[src],io->in_size);
// return
my_log("< %s %s\n",io->name,string);
return true;
}
bool io_line_ready(const io_t * io) {
ASSERT(io_is_ok(io));
if (io->in_eof) return true;
if (memchr(io->in_buffer,LF,io->in_size) != NULL) return true; // buffer contains LF
return false;
}
void io_init(io_t * io) {
ASSERT(io!=NULL);
io->in_eof = false;
io->in_size = 0;
io->out_size = 0;
ASSERT(io_is_ok(io));
}
void io_close(io_t * io) {
ASSERT(io_is_ok(io));
ASSERT(io->out_fd>=0);
my_log("> %s EOF\n",io->name);
if (close(io->out_fd) == -1) {
my_fatal("io_close(): close(): %s\n",strerror(errno));
}
io->out_fd = -1;
}
void io_send(io_t * io, const char format[], ...) {
va_list arg_list;
char string[StringSize];
int len;
ASSERT(io_is_ok(io));
ASSERT(format!=NULL);
ASSERT(io->out_fd>=0);
// format
va_start(arg_list,format);
vsprintf(string,format,arg_list);
va_end(arg_list);
// append string to buffer
len = strlen(string);
if (io->out_size + len > BufferSize-2) my_fatal("io_send(): buffer overflow\n");
memcpy(&io->out_buffer[io->out_size],string,len);
io->out_size += len;
ASSERT(io->out_size>=0&&io->out_size<=BufferSize-2);
// log
io->out_buffer[io->out_size] = '\0';
my_log("> %s %s\n",io->name,io->out_buffer);
// append EOL to buffer
if (UseCR) io->out_buffer[io->out_size++] = CR;
io->out_buffer[io->out_size++] = LF;
ASSERT(io->out_size>=0&&io->out_size<=BufferSize);
// flush buffer
if (UseDebug) my_log("POLYGLOT writing %d byte%s to %s\n",io->out_size,(io->out_size>1)?"s":"",io->name);
my_write(io->out_fd,io->out_buffer,io->out_size);
io->out_size = 0;
}
void io_get_update(io_t * io) {
int pos, size;
int n;
ASSERT(io_is_ok(io));
ASSERT(io->in_fd>=0);
ASSERT(!io->in_eof);
// init
pos = io->in_size;
size = BufferSize - pos;
if (size <= 0) my_fatal("io_get_update(): buffer overflow\n");
// read as many data as possible
n = my_read(io->in_fd,&io->in_buffer[pos],size);
if (UseDebug) my_log("POLYGLOT read %d byte%s from %s\n",n,(n>1)?"s":"",io->name);
if (n > 0) { // at least one character was read
// update buffer size
ASSERT(n>=1&&n<=size);
io->in_size += n;
ASSERT(io->in_size>=0&&io->in_size<=BufferSize);
} else { // EOF
ASSERT(n==0);
io->in_eof = true;
}
}
