1 | #include <sys/socket.h>
|
---|
2 | #include <netinet/in.h>
|
---|
3 | #include <netdb.h>
|
---|
4 | #include <ctype.h>
|
---|
5 | #include <string.h>
|
---|
6 | #include <fcntl.h>
|
---|
7 | #include <errno.h>
|
---|
8 | #include "lookup.h"
|
---|
9 | #include "stdio_impl.h"
|
---|
10 |
|
---|
11 | int __lookup_serv(struct service buf[static MAXSERVS], const char *name, int proto, int socktype, int flags)
|
---|
12 | {
|
---|
13 | char line[128];
|
---|
14 | int cnt = 0;
|
---|
15 | char *p, *z = "";
|
---|
16 | unsigned long port = 0;
|
---|
17 |
|
---|
18 | switch (socktype) {
|
---|
19 | case SOCK_STREAM:
|
---|
20 | switch (proto) {
|
---|
21 | case 0:
|
---|
22 | proto = IPPROTO_TCP;
|
---|
23 | case IPPROTO_TCP:
|
---|
24 | break;
|
---|
25 | default:
|
---|
26 | return EAI_SERVICE;
|
---|
27 | }
|
---|
28 | break;
|
---|
29 | case SOCK_DGRAM:
|
---|
30 | switch (proto) {
|
---|
31 | case 0:
|
---|
32 | proto = IPPROTO_UDP;
|
---|
33 | case IPPROTO_UDP:
|
---|
34 | break;
|
---|
35 | default:
|
---|
36 | return EAI_SERVICE;
|
---|
37 | }
|
---|
38 | case 0:
|
---|
39 | break;
|
---|
40 | default:
|
---|
41 | if (name) return EAI_SERVICE;
|
---|
42 | buf[0].port = 0;
|
---|
43 | buf[0].proto = proto;
|
---|
44 | buf[0].socktype = socktype;
|
---|
45 | return 1;
|
---|
46 | }
|
---|
47 |
|
---|
48 | if (name) {
|
---|
49 | if (!*name) return EAI_SERVICE;
|
---|
50 | port = strtoul(name, &z, 10);
|
---|
51 | }
|
---|
52 | if (!*z) {
|
---|
53 | if (port > 65535) return EAI_SERVICE;
|
---|
54 | if (proto != IPPROTO_UDP) {
|
---|
55 | buf[cnt].port = port;
|
---|
56 | buf[cnt].socktype = SOCK_STREAM;
|
---|
57 | buf[cnt++].proto = IPPROTO_TCP;
|
---|
58 | }
|
---|
59 | if (proto != IPPROTO_TCP) {
|
---|
60 | buf[cnt].port = port;
|
---|
61 | buf[cnt].socktype = SOCK_DGRAM;
|
---|
62 | buf[cnt++].proto = IPPROTO_UDP;
|
---|
63 | }
|
---|
64 | return cnt;
|
---|
65 | }
|
---|
66 |
|
---|
67 | if (flags & AI_NUMERICSERV) return EAI_SERVICE;
|
---|
68 |
|
---|
69 | size_t l = strlen(name);
|
---|
70 |
|
---|
71 | unsigned char _buf[1032];
|
---|
72 | FILE _f, *f = __fopen_rb_ca("/etc/services", &_f, _buf, sizeof _buf);
|
---|
73 | if (!f) switch (errno) {
|
---|
74 | case ENOENT:
|
---|
75 | case ENOTDIR:
|
---|
76 | case EACCES:
|
---|
77 | return EAI_SERVICE;
|
---|
78 | default:
|
---|
79 | return EAI_SYSTEM;
|
---|
80 | }
|
---|
81 |
|
---|
82 | while (fgets(line, sizeof line, f) && cnt < MAXSERVS) {
|
---|
83 | if ((p=strchr(line, '#'))) *p++='\n', *p=0;
|
---|
84 |
|
---|
85 | /* Find service name */
|
---|
86 | for(p=line; (p=strstr(p, name)); p++) {
|
---|
87 | if (p>line && !isspace(p[-1])) continue;
|
---|
88 | if (p[l] && !isspace(p[l])) continue;
|
---|
89 | break;
|
---|
90 | }
|
---|
91 | if (!p) continue;
|
---|
92 |
|
---|
93 | /* Skip past canonical name at beginning of line */
|
---|
94 | for (p=line; *p && !isspace(*p); p++);
|
---|
95 |
|
---|
96 | port = strtoul(p, &z, 10);
|
---|
97 | if (port > 65535 || z==p) continue;
|
---|
98 | if (!strncmp(z, "/udp", 4)) {
|
---|
99 | if (proto == IPPROTO_TCP) continue;
|
---|
100 | buf[cnt].port = port;
|
---|
101 | buf[cnt].socktype = SOCK_DGRAM;
|
---|
102 | buf[cnt++].proto = IPPROTO_UDP;
|
---|
103 | }
|
---|
104 | if (!strncmp(z, "/tcp", 4)) {
|
---|
105 | if (proto == IPPROTO_UDP) continue;
|
---|
106 | buf[cnt].port = port;
|
---|
107 | buf[cnt].socktype = SOCK_STREAM;
|
---|
108 | buf[cnt++].proto = IPPROTO_TCP;
|
---|
109 | }
|
---|
110 | }
|
---|
111 | __fclose_ca(f);
|
---|
112 | return cnt > 0 ? cnt : EAI_SERVICE;
|
---|
113 | }
|
---|