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-2009, 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  *      Markus Armbruster, 2007
  33  */
  34
  35 #include <config.h>
  36
  37 #include <ctype.h>
  38 #include <errno.h>
  39 #include <stdio.h>
  40 #include <stdlib.h>
  41 #include <string.h>
  42 #ifdef _WIN32
  43 #include "w32types.h"
  44 #else
  45 #include <sys/types.h>
  46 #include <sys/socket.h>
  47 #include <unistd.h>
  48 #endif
  49 #include "misc.h"
  50 #include "proto.h"
  51
  52 #ifdef _WIN32
  53 #define read(sock, buffer, buf_size) \
  54         w32_recv((sock), (buffer), (buf_size), 0)
  55 #define write(sock, buffer, buf_size) \
  56         w32_send((sock), (buffer), (buf_size), 0)
  57 #endif
  58
  59 int
  60 recvline(int s, char *buf)
  61 {
  62     int sz = 1024;
  63     char *bp;
  64     char ch;
  65     ssize_t n;
  66
  67     bp = buf;
  68     for (;;) {
  69         n = read(s, &ch, 1);
  70         if (n < 0) {
  71             if (errno != EINTR) {
  72                 perror("read");
  73                 exit(1);
  74             }
  75             continue;
  76         }
  77         if (n == 0)
  78             return -1;
  79         if (ch == '\n')
  80             break;
  81         if (bp < buf + sz - 2)
  82             *bp++ = ch;
  83         /* else silently truncate */
  84     }
  85
  86     *bp++ = ch;
  87     *bp = 0;
  88     return parseid(buf);
  89 }
  90
  91 int
  92 parseid(char *line)
  93 {
  94     char *end;
  95     long id;
  96
  97     id = strtol(line, &end, 36);
  98     if (end == line || *end != ' ') {
  99         fprintf(stderr, "Malformed id in line %s", line);
 100         id = -1;
 101     }
 102     if (id > C_LAST)
 103         id = -1;
 104     return id;
 105 }
 106
 107 int
 108 expect(int s, int match, char *buf)
 109 {
 110     return recvline(s, buf) == match;
 111 }
 112
 113 void
 114 sendcmd(int s, char *cmd, char *arg)
 115 {
 116     char buf[128];
 117     char *p;
 118     ssize_t n;
 119     int len;
 120
 121     len = snprintf(buf, sizeof(buf), "%s %s\n",
 122                    cmd, arg != NULL ? arg : "");
 123     if (len >= (int)sizeof(buf)) {
 124         fprintf(stderr, "%s too long\n", cmd);
 125         exit(1);
 126     }
 127     p = buf;
 128     while (len > 0) {
 129         n = write(s, buf, len);
 130         if (n < 0) {
 131             if (errno != EINTR) {
 132                 perror("sendcmd: write");
 133                 exit(1);
 134             }
 135             n = 0;
 136         }
 137         p += n;
 138         len -= n;
 139     }
 140 }