New upstream release:
[debian/iodine.git] / src / common.c
1 /* Copyright (c) 2006-2007 Bjorn Andersson <flex@kryo.se>, Erik Ekman <yarrick@kryo.se>
2  * Copyright (c) 2007 Albert Lee <trisk@acm.jhu.edu>.
3  *
4  * Permission to use, copy, modify, and distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  *
8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15  */
16
17 #include <arpa/inet.h>
18 #include <arpa/nameser.h>
19 #include <netinet/in.h>
20 #ifdef DARWIN
21 #include <arpa/nameser8_compat.h>
22 #endif
23 #include <time.h>
24 #include <err.h>
25 #include <sys/types.h>
26 #include <sys/param.h>
27 #include <sys/stat.h>
28 #include <stdio.h>
29 #include <stdint.h>
30 #include <stdlib.h>
31 #include <stdint.h>
32 #include <unistd.h>
33 #include <string.h>
34 #include <ctype.h>
35 #include <fcntl.h>
36 #include <termios.h>
37
38 #include "common.h"
39
40 /* daemon(3) exists only in 4.4BSD or later, and in GNU libc */
41 #if !(defined(BSD) && (BSD >= 199306)) && !defined(__GLIBC__)
42 static int daemon(int nochdir, int noclose)
43 {
44         int fd, i;
45  
46         switch (fork()) {
47                 case 0:
48                         break;
49                 case -1:
50                         return -1;
51                 default:
52                         _exit(0);
53         }
54  
55         if (!nochdir) {
56                 chdir("/");
57         }
58  
59         if (setsid() < 0) {
60                 return -1;
61         }
62         
63         if (!noclose) {
64                 if ((fd = open("/dev/null", O_RDWR)) >= 0) {
65                         for (i = 0; i < 3; i++) {
66                                 dup2(fd, i);
67                         }
68                         if (fd > 2) {
69                                 close(fd);
70                         }
71                 }
72         }
73         return 0;
74 }
75 #endif
76
77 int 
78 open_dns(int localport, in_addr_t listen_ip) 
79 {
80         struct sockaddr_in addr;
81         int flag;
82         int fd;
83
84         memset(&addr, 0, sizeof(addr));
85         addr.sin_family = AF_INET;
86         addr.sin_port = htons(localport);
87         /* listen_ip already in network byte order from inet_addr, or 0 */
88         addr.sin_addr.s_addr = listen_ip; 
89
90         if ((fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)) < 0)
91                 err(1, "socket");
92
93         flag = 1;
94 #ifdef SO_REUSEPORT
95         setsockopt(fd, SOL_SOCKET, SO_REUSEPORT, &flag, sizeof(flag));
96 #endif
97         setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &flag, sizeof(flag));
98
99         if(bind(fd, (struct sockaddr*)&addr, sizeof(addr)) < 0) 
100                 err(1, "bind");
101
102         printf("Opened UDP socket\n");
103
104         return fd;
105 }
106
107 void
108 close_dns(int fd)
109 {
110         close(fd);
111 }
112
113 void
114 do_chroot(char *newroot)
115 {
116         if (chroot(newroot) != 0 || chdir("/") != 0)
117                 err(1, "%s", newroot);
118
119         seteuid(geteuid());
120         setuid(getuid());
121 }
122
123 void
124 do_detach()
125 {
126         printf("Detaching from terminal...\n");
127         daemon(0, 0);
128         umask(0);
129         alarm(0);
130 }
131
132 void
133 read_password(char *buf, size_t len)
134 {
135         struct termios old;
136         struct termios tp;
137         char pwd[80];
138
139         tcgetattr(0, &tp);
140         old = tp;
141         
142         tp.c_lflag &= (~ECHO);
143         tcsetattr(0, TCSANOW, &tp);
144
145         printf("Enter password: ");
146         fflush(stdout);
147         scanf("%79s", pwd);
148         printf("\n");
149
150         tcsetattr(0, TCSANOW, &old);    
151
152         strncpy(buf, pwd, len);
153         buf[len-1] = '\0';
154 }
155
156 int
157 check_topdomain(char *str)
158 {
159        int i;
160
161        if(str[0] == '.') /* special case */
162                return 1;
163
164        for( i = 0; i < strlen(str); i++) {
165                if( isalpha(str[i]) || isdigit(str[i]) || str[i] == '-' || str[i] == '.' )
166                        continue;
167                else 
168                        return 1;
169        }
170        return 0;
171 }