]> git.pond.sub.org Git - empserver/blobdiff - src/client/host.c
Update copyright notice
[empserver] / src / client / host.c
index ab3da133ca0bc10ee9ca08ed42be5a5c07c201f9..68df7e21f8d43c71efdfcc105dede145687916d4 100644 (file)
@@ -1,6 +1,6 @@
 /*
  *  Empire - A multi-player, client/server Internet based war game.
- *  Copyright (C) 1986-2000, Dave Pare, Jeff Bailey, Thomas Ruschak,
+ *  Copyright (C) 1986-2009, Dave Pare, Jeff Bailey, Thomas Ruschak,
  *                           Ken Stevens, Steve McClure
  *
  *  This program is free software; you can redistribute it and/or modify
  *
  *  ---
  *
- *  See the "LEGAL", "LICENSE", "CREDITS" and "README" files for all the
- *  related information and legal notices. It is expected that any future
- *  projects/authors will amend these files as needed.
+ *  See files README, COPYING and CREDITS in the root of the source
+ *  tree for related information and legal notices.  It is expected
+ *  that future projects/authors will amend these files as needed.
  *
  *  ---
  *
  *  host.c: make stream connection to empire
- * 
+ *
  *  Known contributors to this file:
  *     Dave Pare, 1989
  *     Steve McClure, 1998
+ *     Markus Armbruster, 2005
  */
 
+#include <config.h>
+
+#include <ctype.h>
+#include <errno.h>
 #include <stdio.h>
-#include <sys/types.h>
+#include <stdlib.h>
+#include <string.h>
 #ifndef _WIN32
+#include <sys/types.h>
 #include <sys/socket.h>
 #include <netinet/in.h>
+#include <arpa/inet.h>
 #include <netdb.h>
 #include <unistd.h>
 #else
-#include <winsock.h>
+#define close(fd) w32_close_socket((fd))
 #endif
-#include <ctype.h>
 #include "misc.h"
 
+#ifdef HAVE_GETADDRINFO
+/*
+ * Inspired by example code from W. Richard Stevens: UNIX Network
+ * Programming, Vol. 1
+ */
+
 int
-hostaddr(name, addr)
-       s_char  *name;
-       struct  sockaddr_in *addr;
+tcp_connect(char *host, char *serv)
 {
-#ifndef _WIN32
-       extern  u_long inet_addr();
-#endif
-       struct  hostent *hp;
-
-       if (name == 0 || *name == 0)
-               return 0;
-       if (isdigit(*name)) {
-               addr->sin_addr.s_addr = inet_addr(name);
-       } else {
-               hp = gethostbyname(name);
-               if (hp == NULL) {
-                       fprintf(stderr, "%s: No such host\n", name);
-                       return 0;
-               }
-               bcopy(hp->h_addr, (s_char *)&addr->sin_addr,
-                       sizeof(addr->sin_addr));
-#ifdef _WIN32
-               printf("Trying to connect to '%s'\n", inet_ntoa(addr->sin_addr));
-               fflush(stdout);
-#endif
-       }
-       return 1;
+    int sockfd, n;
+    struct addrinfo hints, *res, *ressave;
+
+    memset(&hints, 0, sizeof(struct addrinfo));
+    hints.ai_family = AF_UNSPEC;
+    hints.ai_socktype = SOCK_STREAM;
+
+    if ((n = getaddrinfo(host, serv, &hints, &res)) != 0) {
+       fprintf(stderr, "Can't connect to %s:%s: %s\n",
+               host, serv, gai_strerror(n));
+       exit(1);
+    }
+    ressave = res;
+
+    do {
+       sockfd = socket(res->ai_family, res->ai_socktype, res->ai_protocol);
+       if (sockfd < 0)
+           continue;           /* ignore this one */
+
+       if (connect(sockfd, res->ai_addr, res->ai_addrlen) == 0)
+           break;              /* success */
+       close(sockfd);          /* ignore this one */
+    } while ((res = res->ai_next) != NULL);
+
+    if (res == NULL) {         /* errno set from final connect() */
+       fprintf(stderr, "Can't connect to %s:%s: %s\n",
+               host, serv, strerror(errno));
+       exit(1);
+    }
+
+    freeaddrinfo(ressave);
+
+    return sockfd;
 }
 
-int
-hostport(name, addr)
-       s_char  *name;
-       struct  sockaddr_in *addr;
+#else  /* !HAVE_GETADDRINFO */
+
+static int
+hostaddr(char *name, struct sockaddr_in *addr)
 {
-       struct  servent *sp;
-#ifndef _WIN32
-       int atoi();
-#endif
+    struct hostent *hp;
 
-       if (name == 0 || *name == 0)
-               return 0;
-       if (isdigit(*name)) {
-#ifndef _WIN32
-               addr->sin_port = htons(atoi(name));
-#else
-               addr->sin_port = atoi(name);
-               addr->sin_port = htons(addr->sin_port);
-#endif
-       } else {
-               sp = getservbyname(name, "tcp");
-               if (sp == NULL)
-                       return 0;
-               addr->sin_port = sp->s_port;
-       }
-       return 1;
+    if (name == NULL || *name == 0)
+       return 0;
+    if (isdigit(*name)) {
+       addr->sin_addr.s_addr = inet_addr(name);
+    } else {
+       hp = gethostbyname(name);
+       if (hp == NULL)
+           return 0;
+       memcpy(&addr->sin_addr, hp->h_addr, sizeof(addr->sin_addr));
+    }
+    return 1;
+}
+
+static int
+hostport(char *name, struct sockaddr_in *addr)
+{
+    struct servent *sp;
+
+    if (name == NULL || *name == 0)
+       return 0;
+    if (isdigit(*name)) {
+       addr->sin_port = htons(atoi(name));
+    } else {
+       sp = getservbyname(name, "tcp");
+       if (sp == NULL)
+           return 0;
+       addr->sin_port = sp->s_port;
+    }
+    return 1;
+}
+
+static int
+hostconnect(struct sockaddr_in *addr)
+{
+    /* FIXME should attempt connect to all addresses of multi-homed host, not just 1st */
+    int s;
+
+    s = socket(AF_INET, SOCK_STREAM, 0);
+    if (s < 0) {
+       return -1;
+    }
+    addr->sin_family = AF_INET;
+    if (connect(s, (struct sockaddr *)addr, sizeof(*addr)) < 0) {
+       (void)close(s);
+       return -1;
+    }
+    return s;
 }
 
 int
-hostconnect(addr)
-       struct  sockaddr_in *addr;
+tcp_connect(char *host, char *serv)
 {
-       int     s;
+    struct sockaddr_in sin;
+    int sock;
 
-       s = socket(AF_INET, SOCK_STREAM, 0);
-       if (s < 0) {
-#ifdef _WIN32
-               errno = WSAGetLastError();
-#endif
-               perror("socket");
-               return -1;
-       }
-       addr->sin_family = AF_INET;
-       if (connect(s, (struct sockaddr *)addr, sizeof(*addr)) < 0) {
-#ifdef _WIN32
-               errno = WSAGetLastError();
-#endif
-               perror("connect");
-#ifdef _WIN32
-               printf("Check that your EMPIREHOST and EMPIREPORT are correct.\n");
-#endif
-               (void) close(s);
-               return -1;
-       }
-       return s;
+    if (!hostport(serv, &sin)) {
+       fprintf(stderr, "Can't resolve Empire port %s\n", serv);
+       exit(1);
+    }
+    if (!hostaddr(host, &sin)) {
+       fprintf(stderr, "Can't resolve Empire host %s\n", host);
+       exit(1);
+    }
+    if ((sock = hostconnect(&sin)) < 0) {
+       fprintf(stderr, "Can't connect to %s:%s: %s\n",
+               serv, host, strerror(errno));
+       exit(1);
+    }
+    return sock;
 }
+#endif