git.pond.sub.org Git - empserver/blob - src/client/expect.c

   1 /*
   2  *  Empire - A multi-player, client/server Internet based war game.
   3  *  Copyright (C) 1986-2007, Dave Pare, Jeff Bailey, Thomas Ruschak,
   4  *                           Ken Stevens, Steve McClure
   5  *
   6  *  This program is free software; you can redistribute it and/or modify
   7  *  it under the terms of the GNU General Public License as published by
   8  *  the Free Software Foundation; either version 2 of the License, or
   9  *  (at your option) any later version.
  10  *
  11  *  This program is distributed in the hope that it will be useful,
  12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  *  GNU General Public License for more details.
  15  *
  16  *  You should have received a copy of the GNU General Public License
  17  *  along with this program; if not, write to the Free Software
  18  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  *
  20  *  ---
  21  *
  22  *  See files README, COPYING and CREDITS in the root of the source
  23  *  tree for related information and legal notices.  It is expected
  24  *  that future projects/authors will amend these files as needed.
  25  *
  26  *  ---
  27  *
  28  *  expect.c: Read from the socket, expecting to see a particular code.
  29  *
  30  *  Known contributors to this file:
  31  *      Steve McClure, 1998
  32  */
  33
  34 #include <config.h>
  35
  36 #include <ctype.h>
  37 #include <errno.h>
  38 #include <stdio.h>
  39 #include <stdlib.h>
  40 #include <string.h>
  41 #ifndef _WIN32
  42 #include <sys/types.h>
  43 #include <sys/socket.h>
  44 #include <unistd.h>
  45 #endif
  46 #include "misc.h"
  47
  48 #ifdef _WIN32
  49 #define recv(sock, buffer, buf_size, flags) \
  50         w32_recv((sock), (buffer), (buf_size), (flags))
  51 #define read(sock, buffer, buf_size) \
  52         w32_recv((sock), (buffer), (buf_size), 0)
  53 #define write(sock, buffer, buf_size) \
  54         w32_send((sock), (buffer), (buf_size), 0)
  55 #endif
  56
  57 int
  58 recvline(int s, char *buf)
  59 {
  60     int size;
  61     char *p;
  62     int n;
  63     int newline;
  64     char *ptr;
  65     int cc;
  66
  67     size = 1024;
  68     (void)alarm(30);
  69     ptr = buf;
  70     n = recv(s, ptr, size, MSG_PEEK);
  71     if (n <= 0) {
  72         perror("recv");
  73         return 0;
  74     }
  75     size -= n;
  76     buf[n] = '\0';
  77     if ((p = strchr(ptr, '\n')) == NULL) {
  78         do {
  79             cc = read(s, ptr, n);
  80             if (cc < 0) {
  81                 perror("expect: read");
  82                 return 0;
  83             }
  84             if (cc != n) {
  85                 fprintf(stderr, "expect: short read (%d not %d)\n", cc, n);
  86                 return 0;
  87             }
  88             ptr += n;
  89             if ((n = recv(s, ptr, size, MSG_PEEK)) <= 0) {
  90                 perror("recv");
  91                 return 0;
  92             }
  93             size -= n;
  94             ptr[n] = '\0';
  95         } while ((p = strchr(ptr, '\n')) == 0);
  96         newline = 1 + p - buf;
  97         *p = 0;
  98     } else
  99         newline = 1 + p - ptr;
 100     cc = read(s, buf, newline);
 101     if (cc < 0) {
 102         perror("expect: read #2");
 103         return 0;
 104     }
 105     if (cc != newline) {
 106         fprintf(stderr, "expect: short read #2 (%d not %d)\n",
 107                 cc, newline);
 108         return 0;
 109     }
 110     buf[newline] = '\0';
 111     (void)alarm(0);
 112     if (!isxdigit(buf[0]) || buf[1] != ' ') {
 113         fprintf(stderr, "Malformed line %s\n", buf);
 114         return 0;
 115     }
 116     return strtol(buf, NULL, 16);
 117 }
 118
 119 int
 120 expect(int s, int match, char *buf)
 121 {
 122     int code = recvline(s, buf);
 123     return code == match;
 124 }
 125
 126 void
 127 sendcmd(int s, char *cmd, char *arg)
 128 {
 129     char buf[128];
 130     char *p;
 131     ssize_t n;
 132     int len;
 133
 134     len = snprintf(buf, sizeof(buf), "%s %s\n",
 135                    cmd, arg != NULL ? arg : "");
 136     if (len >= (int)sizeof(buf)) {
 137         fprintf(stderr, "%s too long\n", cmd);
 138         exit(1);
 139     }
 140     p = buf;
 141     while (len > 0) {
 142         n = write(s, buf, len);
 143         if (n < 0) {
 144             if (errno != EINTR) {
 145                 perror("sendcmd: write");
 146                 exit(1);
 147             }
 148             n = 0;
 149         }
 150         p += n;
 151         len -= n;
 152     }
 153 }