3350beba48c880c70a8dc7321793cd47dde46f38
[dragonfly.git] / sbin / mount_portal / pt_tcp.c
1 /*
2  * Copyright (c) 1992, 1993, 1994
3  *      The Regents of the University of California.  All rights reserved.
4  * All rights reserved.
5  *
6  * This code is derived from software donated to Berkeley by
7  * Jan-Simon Pendry.
8  *
9  * Redistribution and use in source and binary forms, with or without
10  * modification, are permitted provided that the following conditions
11  * are met:
12  * 1. Redistributions of source code must retain the above copyright
13  *    notice, this list of conditions and the following disclaimer.
14  * 2. Redistributions in binary form must reproduce the above copyright
15  *    notice, this list of conditions and the following disclaimer in the
16  *    documentation and/or other materials provided with the distribution.
17  * 3. All advertising materials mentioning features or use of this software
18  *    must display the following acknowledgement:
19  *      This product includes software developed by the University of
20  *      California, Berkeley and its contributors.
21  * 4. Neither the name of the University nor the names of its contributors
22  *    may be used to endorse or promote products derived from this software
23  *    without specific prior written permission.
24  *
25  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
26  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
27  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
28  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
29  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
30  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
31  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
32  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
33  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
34  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
35  * SUCH DAMAGE.
36  *
37  *      @(#)pt_tcp.c    8.5 (Berkeley) 4/28/95
38  *
39  * $FreeBSD: src/sbin/mount_portal/pt_tcp.c,v 1.9 1999/08/28 00:13:38 peter Exp $
40  * $DragonFly: src/sbin/mount_portal/pt_tcp.c,v 1.5 2005/11/06 12:36:40 swildner Exp $
41  */
42
43 #include <errno.h>
44 #include <stdlib.h>
45 #include <string.h>
46 #include <unistd.h>
47 #include <sys/types.h>
48 #include <sys/param.h>
49 #include <sys/syslog.h>
50 #include <sys/socket.h>
51 #include <netinet/in.h>
52 #include <arpa/inet.h>
53 #include <netdb.h>
54
55 #include "portald.h"
56
57 /*
58  * Key will be tcp/host/port[/"priv"]
59  * Create a TCP socket connected to the
60  * requested host and port.
61  * Some trailing suffix values have special meanings.
62  * An unrecognized suffix is an error.
63  */
64 int
65 portal_tcp(struct portal_cred *pcr, char *key, char **v, int kso, int *fdp)
66 {
67         char host[MAXHOSTNAMELEN];
68         char port[MAXHOSTNAMELEN];
69         char *p = key + (v[1] ? strlen(v[1]) : 0);
70         char *q;
71         struct hostent *hp;
72         struct servent *sp;
73         struct in_addr **ipp;
74         struct in_addr *ip[2];
75         struct in_addr ina;
76         u_short s_port;
77         int priv = 0;
78         struct sockaddr_in sain;
79
80         q = strchr(p, '/');
81         if (q == 0 || q - p >= sizeof(host))
82                 return (EINVAL);
83         *q = '\0';
84         strcpy(host, p);
85         p = q + 1;
86
87         q = strchr(p, '/');
88         if (q)
89                 *q = '\0';
90         if (strlen(p) >= sizeof(port))
91                 return (EINVAL);
92         strcpy(port, p);
93         if (q) {
94                 p = q + 1;
95                 if (strcmp(p, "priv") == 0) {
96                         if (pcr->pcr_uid == 0)
97                                 priv = 1;
98                         else
99                                 return (EPERM);
100                 } else {
101                         return (EINVAL);
102                 }
103         }
104
105         hp = gethostbyname(host);
106         if (hp != 0) {
107                 ipp = (struct in_addr **) hp->h_addr_list;
108         } else {
109                 ina.s_addr = inet_addr(host);
110                 if (ina.s_addr == INADDR_NONE)
111                         return (EINVAL);
112                 ip[0] = &ina;
113                 ip[1] = 0;
114                 ipp = ip;
115         }
116 #ifdef DEBUG
117         printf ("inet address for %s is %s\n", host, inet_ntoa(*ipp[0]));
118 #endif
119
120         sp = getservbyname(port, "tcp");
121         if (sp != NULL) {
122                 s_port = (u_short)sp->s_port;
123         } else {
124                 s_port = strtoul(port, &p, 0);
125                 if (s_port == 0 || *p != '\0')
126                         return (EINVAL);
127                 s_port = htons(s_port);
128         }
129 #ifdef DEBUG
130         printf ("port number for %s is %d\n", port, (int)ntohs(s_port));
131 #endif
132
133         memset(&sain, 0, sizeof(sain));
134         sain.sin_len = sizeof(sain);
135         sain.sin_family = AF_INET;
136         sain.sin_port = s_port;
137
138         while (ipp[0]) {
139                 int so;
140
141                 if (priv)
142                         so = rresvport(NULL);
143                 else
144                         so = socket(AF_INET, SOCK_STREAM, 0);
145                 if (so < 0) {
146                         syslog(LOG_ERR, "socket: %m");
147                         return (errno);
148                 }
149
150                 sain.sin_addr = *ipp[0];
151                 if (connect(so, (struct sockaddr *) &sain, sizeof(sain)) == 0) {
152                         *fdp = so;
153                         return (0);
154                 }
155                 close(so);
156
157                 ipp++;
158         }
159
160         return (errno);
161 }