/*
* Empire - A multi-player, client/server Internet based war game.
- * Copyright (C) 1986-2008, Dave Pare, Jeff Bailey, Thomas Ruschak,
+ * Copyright (C) 1986-2010, Dave Pare, Jeff Bailey, Thomas Ruschak,
* Ken Stevens, Steve McClure
*
* This program is free software; you can redistribute it and/or modify
* Known contributors to this file:
* Doug Hay, 1998
* Steve McClure, 1998
+ * Ron Koenderink, 2009
*/
/*
#include <unistd.h>
#include "empio.h"
#include "empthread.h"
+#include "file.h"
#include "ioqueue.h"
#include "misc.h"
#include "queue.h"
-#include "server.h"
struct iop {
int fd;
struct ioqueue *output;
int flags;
int bufsize;
+ int last_out;
struct timeval input_timeout;
};
struct iop *
io_open(int fd, int flags, int bufsize, struct timeval timeout)
{
+ int fdfl;
struct iop *iop;
- flags = flags & (IO_READ | IO_WRITE | IO_NBLOCK | IO_NEWSOCK);
+ flags = flags & (IO_READ | IO_WRITE);
if ((flags & (IO_READ | IO_WRITE)) == 0)
return NULL;
+
+ fdfl = fcntl(fd, F_GETFL, 0);
+ if (fdfl < 0)
+ return NULL;
+ fdfl |= O_NONBLOCK;
+ if (fcntl(fd, F_SETFL, fdfl) < 0)
+ return NULL;
+
iop = malloc(sizeof(struct iop));
if (!iop)
return NULL;
iop->fd = fd;
- iop->input = 0;
- iop->output = 0;
- iop->flags = 0;
- iop->input_timeout = timeout;
+ iop->input = NULL;
+ iop->output = NULL;
+ iop->flags = flags;
+ iop->last_out = 0;
iop->bufsize = bufsize;
- if ((flags & IO_READ) && (flags & IO_NEWSOCK) == 0)
+ iop->input_timeout = timeout;
+ if (flags & IO_READ)
iop->input = ioq_create(bufsize);
- if ((flags & IO_WRITE) && (flags & IO_NEWSOCK) == 0)
+ if (flags & IO_WRITE)
iop->output = ioq_create(bufsize);
- if (flags & IO_NBLOCK)
- io_noblocking(iop, 1); /* FIXME check success */
- iop->flags = flags;
return iop;
}
io_close(struct iop *iop)
{
- if (iop->input != 0)
+ if (iop->input)
ioq_destroy(iop->input);
- if (iop->output != 0)
+ if (iop->output)
ioq_destroy(iop->output);
(void)close(iop->fd);
free(iop);
}
/*
- * Return number of bytes read on success, zero on timeout or EOF, -1
- * on error, with errno set appropriately. In particular, return -1
- * with errno set to EAGAIN or EWOULDBLOCK when no data is available
- * for non-blocking input (WAITFORINPUT false). Use io_eof() to
- * distinguish timeout from EOF.
+ * Return number of bytes read on success, zero on timeout, early
+ * wakeup or EOF, -1 on error, with errno set appropriately. In
+ * particular, return -1 with errno set to EAGAIN or EWOULDBLOCK when
+ * no data is available for non-blocking input (WAITFORINPUT false).
+ * Use io_eof() to distinguish timeout and early wakeup from EOF.
*/
int
io_input(struct iop *iop, int waitforinput)
return ioq_qsize(iop->output);
}
+/*
+ * Write output queued in IOP.
+ * If WAIT, writing may put the thread to sleep.
+ * Return number of bytes written on success, -1 on error.
+ * In particular, return zero when nothing was written because the
+ * queue was empty, or because the write slept and got woken up (only
+ * if WAIT), or because the write refused to sleep (only if !WAIT).
+ */
int
-io_output(struct iop *iop, int waitforoutput)
+io_output(struct iop *iop, int wait)
{
struct iovec iov[16];
- int cc;
- int n;
- int remain;
+ int n, res, cc;
+
+ if (wait)
+ ef_make_stale();
- /* If there is no output waiting. */
- if (!io_outputwaiting(iop))
+ if (!ioq_qsize(iop->output))
return 0;
- /* If the iop is not write enabled. */
if ((iop->flags & IO_WRITE) == 0)
return -1;
- /* If the io is marked as in error... */
if (iop->flags & IO_ERROR)
return -1;
- /* make the iov point to the data in the queue. */
- /* I.E., each of the elements in the queue. */
- /* returns the number of elements in the iov. */
n = ioq_makeiov(iop->output, iov, IO_BUFSIZE);
- if (n <= 0) {
- return 0;
- }
-
- /* wait for the file to be output ready. */
- if (waitforoutput != IO_NOWAIT) {
- /* This waits for the file to be ready for writing, */
- /* and lets other threads run. */
- empth_select(iop->fd, EMPTH_FD_WRITE, NULL);
+ if (wait) {
+ res = empth_select(iop->fd, EMPTH_FD_WRITE, NULL);
+ if (res == 0)
+ return 0;
+ if (res < 0) {
+ iop->flags |= IO_ERROR;
+ return -1;
+ }
}
- /* Do the actual write. */
cc = writev(iop->fd, iov, n);
-
- /* if it failed.... */
if (cc < 0) {
- /* Hmm, it would block. file is opened noblock, soooooo.. */
- if (errno == EAGAIN || errno == EWOULDBLOCK) {
- /* If there are remaining bytes, set the IO as remaining.. */
- remain = ioq_qsize(iop->output);
- return remain;
- }
+ if (errno == EAGAIN || errno == EWOULDBLOCK)
+ return 0;
iop->flags |= IO_ERROR;
return -1;
}
- /* If no bytes were written, something happened.. Like an EOF. */
- if (cc == 0) {
- iop->flags |= IO_EOF;
- return 0;
- }
-
- /* Remove the number of written bytes from the queue. */
ioq_dequeue(iop->output, cc);
-
+ iop->last_out = ioq_qsize(iop->output);
return cc;
}
+/*
+ * Write output queued in IOP if enough have been enqueued.
+ * Write if at least one buffer has been filled since the last write.
+ * If WAIT, writing may put the thread to sleep.
+ * Return number of bytes written on success, -1 on error.
+ * In particular, return zero when nothing was written because the
+ * queue was not long, or the write slept and got woken up (only if
+ * WAIT), or the write refused to sleep (only if !WAIT).
+ */
+int
+io_output_if_queue_long(struct iop *iop, int wait)
+{
+ int len = ioq_qsize(iop->output);
+
+ if (CANT_HAPPEN(iop->last_out > len))
+ iop->last_out = 0;
+ if (len - iop->last_out < iop->bufsize) {
+ if (wait)
+ ef_make_stale();
+ return 0;
+ }
+ return io_output(iop, wait);
+}
+
+
int
io_peek(struct iop *iop, char *buf, int nbytes)
{
}
int
-io_write(struct iop *iop, char *buf, int nbytes, int doWait)
+io_write(struct iop *iop, char *buf, int nbytes)
{
- int len;
-
if ((iop->flags & IO_WRITE) == 0)
return -1;
ioq_append(iop->output, buf, nbytes);
- len = ioq_qsize(iop->output);
- if (len > iop->bufsize) {
- if (doWait) {
- io_output_all(iop);
- } else {
- /* only try a write every BUFSIZE characters */
- if (((len - nbytes) % iop->bufsize) < (len % iop->bufsize))
- io_output(iop, IO_NOWAIT);
- }
- }
return nbytes;
}
-int
-io_output_all(struct iop *iop)
-{
- int n;
-
- /*
- * Mustn't block a player thread while update is pending, or else
- * a malicous player could delay the update indefinitely
- */
- while ((n = io_output(iop, IO_NOWAIT)) > 0 && !play_wrlock_wanted)
- empth_select(iop->fd, EMPTH_FD_WRITE, NULL);
-
- return n;
-}
-
int
io_gets(struct iop *iop, char *buf, int nbytes)
{
if (flags & IO_WRITE) {
shutdown(iop->fd, 1);
ioq_drain(iop->output);
+ iop->last_out = 0;
}
return 0;
}
-int
-io_noblocking(struct iop *iop, int value)
-{
- int flags;
-
- flags = fcntl(iop->fd, F_GETFL, 0);
- if (flags < 0)
- return -1;
- if (value == 0)
- flags &= ~O_NONBLOCK;
- else
- flags |= O_NONBLOCK;
- if (fcntl(iop->fd, F_SETFL, flags) < 0)
- return -1;
- if (value == 0)
- iop->flags &= ~IO_NBLOCK;
- else
- iop->flags |= IO_NBLOCK;
- return 0;
-}
-
int
io_error(struct iop *iop)
{