don't set page size by default
[tridge/junkcode.git] / sockspy-spnego.c
1 #define _GNU_SOURCE
2
3 #include <stdio.h>
4 #include <stdlib.h>
5 #include <string.h>
6 #include <time.h>
7 #include <errno.h>
8 #include <sys/socket.h>
9 #include <netdb.h>
10 #include <unistd.h>
11 #include <stdlib.h>
12 #include <fcntl.h>
13 #include <sys/types.h>
14 #include <sys/time.h>
15 #include <arpa/inet.h>
16 #include <netinet/in.h>
17
18 #define MAX(a,b) ((a)>(b)?(a):(b))
19
20 #define SEARCH "GSS-SPNEGO"
21 #define REPLACE "GSS-XXXXXX"
22
23 #define SEARCH2 "NTLMSSP"
24 #define REPLACE2 "NTLMXXX"
25
26 static void replace_str(char *buf, int n)
27 {
28         void *p;
29         p = memmem(buf, n, SEARCH, strlen(SEARCH));
30         if (p) {
31                 printf("Replaced %s with %s\n", SEARCH, REPLACE);
32                 memcpy(p, REPLACE, strlen(REPLACE));
33         }
34         p = memmem(buf, n, SEARCH2, strlen(SEARCH2));
35         if (p) {
36                 printf("Replaced %s with %s\n", SEARCH2, REPLACE2);
37                 memcpy(p, REPLACE2, strlen(REPLACE2));
38         }
39 }
40
41 /* open a socket to a tcp remote host with the specified port */
42 static int open_socket_out(const char *host, int port)
43 {
44         struct sockaddr_in sock_out;
45         int res;
46         struct hostent *hp;  
47         struct in_addr addr;
48
49         res = socket(PF_INET, SOCK_STREAM, 0);
50         if (res == -1) {
51                 return -1;
52         }
53
54         if (inet_pton(AF_INET, host, &addr) > 0) {
55                 memcpy(&sock_out.sin_addr, &addr, sizeof(addr));
56         } else {
57                 hp = gethostbyname(host);
58                 if (!hp) {
59                         fprintf(stderr,"unknown host %s\n", host);
60                         return -1;
61                 }
62                 memcpy(&sock_out.sin_addr, hp->h_addr, hp->h_length);
63         }
64
65         sock_out.sin_port = htons(port);
66         sock_out.sin_family = PF_INET;
67
68         if (connect(res,(struct sockaddr *)&sock_out,sizeof(sock_out)) != 0) {
69                 close(res);
70                 fprintf(stderr,"failed to connect to %s (%s)\n", 
71                         host, strerror(errno));
72                 return -1;
73         }
74
75         return res;
76 }
77
78
79 /*
80   open a socket of the specified type, port and address for incoming data
81 */
82 int open_socket_in(int port)
83 {
84         struct sockaddr_in sock;
85         int res;
86         int one=1;
87
88         memset(&sock,0,sizeof(sock));
89
90 #ifdef HAVE_SOCK_SIN_LEN
91         sock.sin_len = sizeof(sock);
92 #endif
93         sock.sin_port = htons(port);
94         sock.sin_family = AF_INET;
95
96         res = socket(AF_INET, SOCK_STREAM, 0);
97         if (res == -1) { 
98                 fprintf(stderr, "socket failed\n"); return -1; 
99                 return -1;
100         }
101
102         setsockopt(res,SOL_SOCKET,SO_REUSEADDR,(char *)&one,sizeof(one));
103
104         if (bind(res, (struct sockaddr *)&sock, sizeof(sock)) < 0) { 
105                 return(-1); 
106         }
107
108         return res;
109 }
110
111 /* write to a file descriptor, making sure we get all the data out or
112  * die trying */
113 static void write_all(int fd, unsigned char *s, size_t n)
114 {
115         while (n) {
116                 int r;
117                 r = write(fd, s, n);
118                 if (r <= 0) {
119                         exit(1);
120                 }
121                 s += r;
122                 n -= r;
123         }
124 }
125
126 static void main_loop(int sock1, int sock2)
127 {
128         unsigned char buf[1024];
129         int log1, log2, i=0;
130
131         do {
132                 char *fname1, *fname2;
133                 asprintf(&fname1, "sockspy-in.%d", i);
134                 asprintf(&fname2, "sockspy-out.%d", i);
135                 log1 = open(fname1, O_WRONLY|O_CREAT|O_EXCL, 0644);
136                 log2 = open(fname2, O_WRONLY|O_CREAT|O_EXCL, 0644);
137                 free(fname1);
138                 free(fname2);
139                 i++;
140         } while (i<1000 && (log1 == -1 || log2 == -1));
141
142         if (log1 == -1 || log2 == -1) {
143                 fprintf(stderr,"Failed to open log files\n");
144                 return;
145         }
146
147         while (1) {
148                 fd_set fds;
149                 int ret;
150
151                 FD_ZERO(&fds);
152                 FD_SET(sock1, &fds);
153                 FD_SET(sock2, &fds);
154
155                 ret = select(MAX(sock1, sock2)+1, &fds, NULL, NULL, NULL);
156                 if (ret == -1 && errno == EINTR) continue;
157                 if (ret <= 0) break;
158
159                 if (FD_ISSET(sock1, &fds)) {
160                         int n = read(sock1, buf, sizeof(buf));
161                         if (n <= 0) break;
162
163                         replace_str(buf, n);
164
165                         write_all(sock2, buf, n);
166                         write_all(log1, buf, n);
167                 }
168
169                 if (FD_ISSET(sock2, &fds)) {
170                         int n = read(sock2, buf, sizeof(buf));
171                         if (n <= 0) break;
172
173                         replace_str(buf, n);
174
175                         write_all(sock1, buf, n);
176                         write_all(log2, buf, n);
177                 }
178         }       
179 }
180
181 int main(int argc, char *argv[])
182 {
183         int listen_port, dest_port;
184         char *host;
185         int sock_in;
186         int sock_out;
187         int listen_fd;
188         struct sockaddr addr;
189         int in_addrlen = sizeof(addr);
190
191         if (argc < 4) {
192                 printf("Usage: sockspy <inport> <host> <port>\n");
193                 exit(1);
194         }
195
196         listen_port = atoi(argv[1]);
197         host = argv[2];
198         dest_port = atoi(argv[3]);
199
200         listen_fd = open_socket_in(listen_port);
201
202         if (listen_fd == -1) {
203                 fprintf(stderr,"listen on port %d failed - %s\n", 
204                         listen_port, strerror(errno));
205                 exit(1);
206         }
207
208         if (listen(listen_fd, 5) == -1) {
209                 fprintf(stderr,"listen failed\n");
210                 exit(1);
211         }
212
213         sock_in = accept(listen_fd,&addr,&in_addrlen);
214
215         if (sock_in == -1) {
216                 fprintf(stderr,"accept on port %d failed - %s\n", 
217                         listen_port, strerror(errno));
218                 exit(1);
219         }
220
221         close(listen_fd);
222
223         sock_out = open_socket_out(host, dest_port);
224         if (sock_out == -1) {
225                 exit(1);
226         }
227
228         main_loop(sock_in, sock_out);
229         return 0;
230 }