]> code.delx.au - pulseaudio/blob - polyp/socket-util.c
Apply Joe Marcus Clarke's FreeBSD patches
[pulseaudio] / polyp / socket-util.c
1 /* $Id$ */
2
3 /***
4 This file is part of polypaudio.
5
6 polypaudio is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published
8 by the Free Software Foundation; either version 2 of the License,
9 or (at your option) any later version.
10
11 polypaudio is distributed in the hope that it will be useful, but
12 WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 General Public License for more details.
15
16 You should have received a copy of the GNU General Public License
17 along with polypaudio; if not, write to the Free Software
18 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
19 USA.
20 ***/
21
22 #ifdef HAVE_CONFIG_H
23 #include <config.h>
24 #endif
25
26 #include <stdarg.h>
27 #include <stdlib.h>
28 #include <signal.h>
29 #include <errno.h>
30 #include <assert.h>
31 #include <string.h>
32 #include <stdio.h>
33 #include <sys/types.h>
34 #include <sys/un.h>
35 #include <netinet/in.h>
36 #include <fcntl.h>
37 #include <unistd.h>
38 #include <netinet/in_systm.h>
39 #include <netinet/tcp.h>
40 #include <netinet/ip.h>
41 #include <sys/socket.h>
42 #include <sys/stat.h>
43 #include <netdb.h>
44
45 #include "socket-util.h"
46 #include "util.h"
47 #include "xmalloc.h"
48 #include "log.h"
49
50 void pa_socket_peer_to_string(int fd, char *c, size_t l) {
51 struct stat st;
52
53 assert(c && l && fd >= 0);
54
55 if (fstat(fd, &st) < 0) {
56 snprintf(c, l, "Invalid client fd");
57 return;
58 }
59
60 if (S_ISSOCK(st.st_mode)) {
61 union {
62 struct sockaddr sa;
63 struct sockaddr_in in;
64 struct sockaddr_un un;
65 } sa;
66 socklen_t sa_len = sizeof(sa);
67
68 if (getpeername(fd, &sa.sa, &sa_len) >= 0) {
69
70 if (sa.sa.sa_family == AF_INET) {
71 uint32_t ip = ntohl(sa.in.sin_addr.s_addr);
72
73 snprintf(c, l, "TCP/IP client from %i.%i.%i.%i:%u",
74 ip >> 24,
75 (ip >> 16) & 0xFF,
76 (ip >> 8) & 0xFF,
77 ip & 0xFF,
78 ntohs(sa.in.sin_port));
79 return;
80 } else if (sa.sa.sa_family == AF_LOCAL) {
81 snprintf(c, l, "UNIX socket client");
82 return;
83 }
84
85 }
86 snprintf(c, l, "Unknown network client");
87 return;
88 } else if (S_ISCHR(st.st_mode) && (fd == 0 || fd == 1)) {
89 snprintf(c, l, "STDIN/STDOUT client");
90 return;
91 }
92
93 snprintf(c, l, "Unknown client");
94 }
95
96 int pa_socket_low_delay(int fd) {
97 int priority;
98 assert(fd >= 0);
99
100 #ifdef SO_PRIORITY
101 priority = 7;
102 if (setsockopt(fd, SOL_SOCKET, SO_PRIORITY, &priority, sizeof(priority)) < 0)
103 return -1;
104 #endif
105
106 return 0;
107 }
108
109 int pa_socket_tcp_low_delay(int fd) {
110 int ret, tos, on;
111
112 assert(fd >= 0);
113
114 ret = pa_socket_low_delay(fd);
115
116 on = 1;
117 #if defined(SOL_TCP) || defined(IPPROTO_TCP)
118 #if defined(SOL_TCP)
119 if (setsockopt(fd, SOL_TCP, TCP_NODELAY, &on, sizeof(on)) < 0)
120 #else
121 if (setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &on, sizeof(on)) < 0)
122 #endif
123 ret = -1;
124 #endif
125
126 #if defined(IPTOS_LOWDELAY) && defined(IP_TOS) && (defined(SOL_IP) || \
127 defined(IPPROTO_IP))
128 tos = IPTOS_LOWDELAY;
129 #ifdef SOL_IP
130 if (setsockopt(fd, SOL_IP, IP_TOS, &tos, sizeof(tos)) < 0)
131 #else
132 if (setsockopt(fd, IPPROTO_IP, IP_TOS, &tos, sizeof(tos)) < 0)
133 #endif
134 ret = -1;
135 #endif
136
137 return ret;
138
139 }
140
141 int pa_socket_set_rcvbuf(int fd, size_t l) {
142 assert(fd >= 0);
143
144 /* if (setsockopt(fd, SOL_SOCKET, SO_RCVBUF, &l, sizeof(l)) < 0) { */
145 /* pa_log(__FILE__": SO_RCVBUF: %s\n", strerror(errno)); */
146 /* return -1; */
147 /* } */
148
149 return 0;
150 }
151
152 int pa_socket_set_sndbuf(int fd, size_t l) {
153 assert(fd >= 0);
154
155 /* if (setsockopt(fd, SOL_SOCKET, SO_SNDBUF, &l, sizeof(l)) < 0) { */
156 /* pa_log(__FILE__": SO_SNDBUF: %s\n", strerror(errno)); */
157 /* return -1; */
158 /* } */
159
160 return 0;
161 }
162
163 int pa_unix_socket_is_stale(const char *fn) {
164 struct sockaddr_un sa;
165 int fd = -1, ret = -1;
166
167 if ((fd = socket(PF_LOCAL, SOCK_STREAM, 0)) < 0) {
168 pa_log(__FILE__": socket(): %s\n", strerror(errno));
169 goto finish;
170 }
171
172 sa.sun_family = AF_LOCAL;
173 strncpy(sa.sun_path, fn, sizeof(sa.sun_path)-1);
174 sa.sun_path[sizeof(sa.sun_path) - 1] = 0;
175
176 if (connect(fd, (struct sockaddr*) &sa, sizeof(sa)) < 0) {
177 if (errno == ECONNREFUSED)
178 ret = 1;
179 } else
180 ret = 0;
181
182 finish:
183 if (fd >= 0)
184 close(fd);
185
186 return ret;
187 }
188
189 int pa_unix_socket_remove_stale(const char *fn) {
190 int r;
191
192 if ((r = pa_unix_socket_is_stale(fn)) < 0)
193 return errno != ENOENT ? -1 : 0;
194
195 if (!r)
196 return 0;
197
198 /* Yes, here is a race condition. But who cares? */
199 if (unlink(fn) < 0)
200 return -1;
201
202 return 0;
203 }
204
205 int pa_unix_socket_make_secure_dir(const char *fn) {
206 int ret = -1;
207 char *slash, *dir = pa_xstrdup(fn);
208
209 if (!(slash = strrchr(dir, '/')))
210 goto finish;
211 *slash = 0;
212
213 if (pa_make_secure_dir(dir) < 0)
214 goto finish;
215
216 ret = 0;
217
218 finish:
219 pa_xfree(dir);
220 return ret;
221 }
222
223 int pa_unix_socket_remove_secure_dir(const char *fn) {
224 int ret = -1;
225 char *slash, *dir = pa_xstrdup(fn);
226
227 if (!(slash = strrchr(dir, '/')))
228 goto finish;
229 *slash = 0;
230
231 if (rmdir(dir) < 0)
232 goto finish;
233
234 ret = 0;
235
236 finish:
237 pa_xfree(dir);
238 return ret;
239 }
240
241 struct sockaddr *pa_resolve_server(const char *server, size_t *len, uint16_t nport) {
242 struct sockaddr *sa;
243 struct addrinfo hints, *result = NULL;
244 char *port, host[256], tmp[16];
245 assert(server && len);
246
247 snprintf(host, sizeof(host), "%s", server);
248 host[strcspn(host, ":")] = 0;
249
250 if ((port = strrchr(server, ':')))
251 port++;
252
253 if (!port)
254 snprintf(port = tmp, sizeof(tmp), "%u", nport);
255
256 memset(&hints, 0, sizeof(hints));
257 hints.ai_family = PF_UNSPEC;
258 hints.ai_socktype = SOCK_STREAM;
259 hints.ai_protocol = 0;
260
261 if (getaddrinfo(host, port, &hints, &result) != 0)
262 return NULL;
263 assert(result);
264
265 sa = pa_xmalloc(*len = result->ai_addrlen);
266 memcpy(sa, result->ai_addr, *len);
267
268 freeaddrinfo(result);
269
270 return sa;
271 }
272