]> git.pond.sub.org Git - empserver/blob - src/client/host.c
dde057cd08672e0c0e73950bc0d333dd7a1ad2f6
[empserver] / src / client / host.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  *  host.c: make stream connection to empire
29  *
30  *  Known contributors to this file:
31  *     Dave Pare, 1989
32  *     Steve McClure, 1998
33  *     Markus Armbruster, 2005-2009
34  */
35
36 #include <config.h>
37
38 #include <ctype.h>
39 #include <errno.h>
40 #include <stdio.h>
41 #include <stdlib.h>
42 #include <string.h>
43 #include <sys/types.h>
44 #include <sys/socket.h>
45 #include <netinet/in.h>
46 #include <arpa/inet.h>
47 #include <netdb.h>
48 #include <unistd.h>
49 #include "misc.h"
50
51 /* Portability cruft, should become unnecessary eventually */
52 #ifndef AI_ADDRCONFIG
53 #define AI_ADDRCONFIG 0
54 #endif
55
56 #ifdef HAVE_GETADDRINFO
57 /*
58  * Inspired by example code from W. Richard Stevens: UNIX Network
59  * Programming, Vol. 1
60  */
61
62 int
63 tcp_connect(char *host, char *serv)
64 {
65     int sockfd, n;
66     struct addrinfo hints, *res, *ressave;
67
68     memset(&hints, 0, sizeof(struct addrinfo));
69     hints.ai_flags = AI_ADDRCONFIG;
70     hints.ai_family = AF_UNSPEC;
71     hints.ai_socktype = SOCK_STREAM;
72
73     if ((n = getaddrinfo(host, serv, &hints, &res)) != 0) {
74         fprintf(stderr, "Can't connect to %s:%s: %s\n",
75                 host, serv, gai_strerror(n));
76         exit(1);
77     }
78     ressave = res;
79
80     do {
81         sockfd = socket(res->ai_family, res->ai_socktype, res->ai_protocol);
82         if (sockfd < 0)
83             continue;           /* ignore this one */
84
85         if (connect(sockfd, res->ai_addr, res->ai_addrlen) == 0)
86             break;              /* success */
87         close(sockfd);          /* ignore this one */
88     } while ((res = res->ai_next) != NULL);
89
90     if (res == NULL) {          /* errno set from final connect() */
91         fprintf(stderr, "Can't connect to %s:%s: %s\n",
92                 host, serv, strerror(errno));
93         exit(1);
94     }
95
96     freeaddrinfo(ressave);
97
98     return sockfd;
99 }
100
101 #else  /* !HAVE_GETADDRINFO */
102
103 static int
104 hostaddr(char *name, struct sockaddr_in *addr)
105 {
106     struct hostent *hp;
107
108     if (name == NULL || *name == 0)
109         return 0;
110     if (isdigit(*name)) {
111         addr->sin_addr.s_addr = inet_addr(name);
112     } else {
113         hp = gethostbyname(name);
114         if (hp == NULL)
115             return 0;
116         memcpy(&addr->sin_addr, hp->h_addr, sizeof(addr->sin_addr));
117     }
118     return 1;
119 }
120
121 static int
122 hostport(char *name, struct sockaddr_in *addr)
123 {
124     struct servent *sp;
125
126     if (name == NULL || *name == 0)
127         return 0;
128     if (isdigit(*name)) {
129         addr->sin_port = htons(atoi(name));
130     } else {
131         sp = getservbyname(name, "tcp");
132         if (sp == NULL)
133             return 0;
134         addr->sin_port = sp->s_port;
135     }
136     return 1;
137 }
138
139 static int
140 hostconnect(struct sockaddr_in *addr)
141 {
142     /* FIXME should attempt connect to all addresses of multi-homed host, not just 1st */
143     int s;
144
145     s = socket(AF_INET, SOCK_STREAM, 0);
146     if (s < 0) {
147         return -1;
148     }
149     addr->sin_family = AF_INET;
150     if (connect(s, (struct sockaddr *)addr, sizeof(*addr)) < 0) {
151         (void)close(s);
152         return -1;
153     }
154     return s;
155 }
156
157 int
158 tcp_connect(char *host, char *serv)
159 {
160     struct sockaddr_in sin;
161     int sock;
162
163     if (!hostport(serv, &sin)) {
164         fprintf(stderr, "Can't resolve Empire port %s\n", serv);
165         exit(1);
166     }
167     if (!hostaddr(host, &sin)) {
168         fprintf(stderr, "Can't resolve Empire host %s\n", host);
169         exit(1);
170     }
171     if ((sock = hostconnect(&sin)) < 0) {
172         fprintf(stderr, "Can't connect to %s:%s: %s\n",
173                 serv, host, strerror(errno));
174         exit(1);
175     }
176     return sock;
177 }
178 #endif