log_proxy.c 3.42 KB
Newer Older
Jeroen Vreeken's avatar
Jeroen Vreeken committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
/*
	Copyright Jeroen Vreeken (pe1rxq@amsat.org), 2008
	Copyright Stichting C.A. Muller Radioastronomiestation, 2008

	This program is free software: you can redistribute it and/or modify
	it under the terms of the GNU General Public License as published by
	the Free Software Foundation, either version 3 of the License, or
	(at your option) any later version.

	This program is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
	GNU General Public License for more details.

	You should have received a copy of the GNU General Public License
	along with this program.  If not, see <http://www.gnu.org/licenses/>.

 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/ioctl.h>
#include <unistd.h>
#include <errno.h>
#include <signal.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

#include "tcp_listen.h"
#include "tcp_connect.h"

#include "dt_port_numbers.h"
Jeroen Vreeken's avatar
Jeroen Vreeken committed
35
#include "dt_host.h"
Jeroen Vreeken's avatar
Jeroen Vreeken committed
36
37
38
39
40
41
42

char *logfile = "controller.log";

int main (int argc, char **argv)
{
	int fd_listen;
	int fd_log;
Jeroen Vreeken's avatar
Jeroen Vreeken committed
43
	FILE *f_log;
Jeroen Vreeken's avatar
Jeroen Vreeken committed
44
45
46
47
	int *fd_clients = NULL;
	int nr_clients = 0;
	int fd_logfile = -1;
	
Jeroen Vreeken's avatar
Jeroen Vreeken committed
48
49
	if (argc != 1 && argc != 2) {
		printf("Usage: %s <logfile>\n", argv[0]);
Jeroen Vreeken's avatar
Jeroen Vreeken committed
50
		printf("\n");
Jeroen Vreeken's avatar
Jeroen Vreeken committed
51
52
		printf("e.g.: %s\n", argv[0]);
		printf("or:   %s /var/log/controller.log\n", argv[0]);
Jeroen Vreeken's avatar
Jeroen Vreeken committed
53
54
		return 1;
	}
Jeroen Vreeken's avatar
Jeroen Vreeken committed
55
56
	if (argc == 2) {
		logfile = argv[1];
Jeroen Vreeken's avatar
Jeroen Vreeken committed
57
58
59
60
61
62
63
64
65
66
67
68
	}

	fd_listen = tcp_listen(CONSOLE_LOG_PORT, 0, 100);
	if (fd_listen < 0) {
		printf("Could not open listen port\n");
		return 1;
	}
	
	signal(SIGPIPE, SIG_IGN);
	
	ioctl(fd_listen, FIONBIO, &(int){ 1 });

Jeroen Vreeken's avatar
Jeroen Vreeken committed
69
	fd_log = tcp_connect(dt_host_controller(), CTRL_LOG_PORT);
Jeroen Vreeken's avatar
Jeroen Vreeken committed
70
	f_log = fdopen(fd_log, "r");
Jeroen Vreeken's avatar
Jeroen Vreeken committed
71
72
73
	
	while (1) {
		int rb;
Jeroen Vreeken's avatar
Jeroen Vreeken committed
74
		char buffer[1000];
Jeroen Vreeken's avatar
Jeroen Vreeken committed
75
76
		int fdnew;
		
Jeroen Vreeken's avatar
Jeroen Vreeken committed
77
78
79
80
		if (fd_log < 0) {
			while (fd_log < 0) {
				sleep(1);
				printf("Trying to connect to server.\n");
Jeroen Vreeken's avatar
Jeroen Vreeken committed
81
				fd_log = tcp_connect(dt_host_controller(), CTRL_LOG_PORT);
Jeroen Vreeken's avatar
Jeroen Vreeken committed
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
			}
			rb = sprintf(buffer, 
			    "Established connection to controller\n");
			printf(buffer);
			f_log = fdopen(fd_log, "r");
			printf("fd_log: %d f_log %p\n", fd_log, f_log);
		} else {
			char *s;
			
			printf("Going to wait for data\n");
			s = fgets(buffer, sizeof(buffer), f_log);
			if (!s)
				rb = -1;
			else
				rb = strlen(buffer);
			printf("rb: %d\n", rb);
			
			
			if (rb < 0) {
				fclose(f_log);
				
				fd_log = -1;
				rb = sprintf(buffer, 
				    "Lost connection to controller\n");
				printf(buffer);
			}
Jeroen Vreeken's avatar
Jeroen Vreeken committed
108
		}
Jeroen Vreeken's avatar
Jeroen Vreeken committed
109
			
Jeroen Vreeken's avatar
Jeroen Vreeken committed
110
111
112
113
114
115
116
117
		if (fd_logfile < 0) {
			fd_logfile = open(logfile, 
			    O_WRONLY | O_APPEND | O_CREAT,
			    S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH);
		}
		if (fd_logfile >= 0) {
			int ret;
			
Jeroen Vreeken's avatar
Jeroen Vreeken committed
118
			ret = write(fd_logfile, buffer, rb);
Jeroen Vreeken's avatar
Jeroen Vreeken committed
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
			if (ret != 1) {
				close(fd_logfile);
				fd_logfile = -1;
			}
		}

		fdnew = tcp_accept(fd_listen);
		if (fdnew >= 0) {
			nr_clients++;
			fd_clients = realloc(fd_clients, sizeof(int)*nr_clients);
			fd_clients[nr_clients-1] = fdnew;
			ioctl(fdnew, FIONBIO, &(int){ 1 });
			printf("New client. (now at %d)\n", nr_clients);
		}
		
		if (rb > 0) {
			int i;
			for (i = 0; i < nr_clients; i++) {
				int ret;
				
Jeroen Vreeken's avatar
Jeroen Vreeken committed
139
140
				ret = write(fd_clients[i], buffer, rb);
				if (ret != rb) {
Jeroen Vreeken's avatar
Jeroen Vreeken committed
141
142
143
144
145
146
147
148
149
150
151
152
153
154
					int j;
					close(fd_clients[i]);
					for (j = i + 1; j < nr_clients; j++) {
						fd_clients[j-1] = fd_clients[j];
					}
					nr_clients--;
					i--;
					printf("Lost a client. (now at %d)\n",
					    nr_clients);
				}
			}
		}
	}
}