aa6a300fb7cee0d67074ac515ab9cf20e0f0bb69
[l2tpns.git] / nsctl.c
1 #include <stdio.h>
2 #include <arpa/inet.h>
3 #include <errno.h>
4 #include <fcntl.h>
5 #include <malloc.h>
6 #include <netdb.h>
7 #include <string.h>
8 #include <sys/time.h>
9 #include <sys/types.h>
10 #include <sys/socket.h>
11 #include <netinet/in.h>
12 #include <signal.h>
13 #include <stdarg.h>
14 #include <stdlib.h>
15 #include <unistd.h>
16 #include <time.h>
17 #include "control.h"
18
19 struct { char *command; int pkt_type; int params; } commands[] = {
20 { "load_plugin", PKT_LOAD_PLUGIN, 1 },
21 { "unload_plugin", PKT_UNLOAD_PLUGIN, 1 },
22 { "garden", PKT_GARDEN, 1 },
23 { "ungarden", PKT_UNGARDEN, 1 },
24 };
25
26 char *dest_host = NULL;
27 unsigned int dest_port = 1702;
28 int udpfd;
29
30 int main(int argc, char *argv[])
31 {
32 int len = 0;
33 int dest_ip = 0;
34 int pkt_type = 0;
35 char *packet = NULL;
36 int i;
37
38 setbuf(stdout, NULL);
39
40 if (argc < 3)
41 {
42 printf("Usage: %s <host> <command> [args...]\n", argv[0]);
43 return 1;
44 }
45
46 dest_host = strdup(argv[1]);
47
48 {
49 // Init socket
50 int on = 1;
51 struct sockaddr_in addr;
52
53 memset(&addr, 0, sizeof(addr));
54 addr.sin_family = AF_INET;
55 addr.sin_port = htons(1703);
56 udpfd = socket(AF_INET, SOCK_DGRAM, 17);
57 setsockopt(udpfd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));
58 if (bind(udpfd, (void *) &addr, sizeof(addr)) < 0)
59 {
60 perror("bind");
61 return(1);
62 }
63 }
64
65 {
66 struct hostent *h = gethostbyname(dest_host);
67 if (h) dest_ip = ntohl(*(unsigned int *)h->h_addr);
68 if (!dest_ip) dest_ip = ntohl(inet_addr(dest_host));
69 if (!dest_ip)
70 {
71 printf("Can't resolve \"%s\"\n", dest_host);
72 return 0;
73 }
74 }
75
76 if (!(packet = calloc(1400, 1)))
77 {
78 perror("calloc");
79 return(1);
80 }
81
82 srand(time(NULL));
83
84 // Deal with command & params
85 for (i = 0; i < (sizeof(commands) / sizeof(commands[0])); i++)
86 {
87 if (strcasecmp(commands[i].command, argv[2]) == 0)
88 {
89 int p;
90 pkt_type = commands[i].pkt_type;
91 len = new_packet(pkt_type, packet);
92 if (argc < (commands[i].params + 3))
93 {
94 printf("Not enough parameters for %s\n", argv[2]);
95 return 1;
96 }
97 for (p = 0; p < commands[i].params; p++)
98 {
99 strncpy((packet + len), argv[p + 3], 1400 - len - 1);
100 len += strlen(argv[p + 3]) + 1;
101 }
102 break;
103 }
104 }
105 if (!pkt_type)
106 {
107 printf("Unknown command\n");
108 return 1;
109 }
110
111 send_packet(udpfd, dest_ip, dest_port, packet, len);
112
113 {
114 int n;
115 fd_set r;
116 struct timeval timeout;
117
118 FD_ZERO(&r);
119 FD_SET(udpfd, &r);
120 timeout.tv_sec = 1;
121 timeout.tv_usec = 0;
122
123 n = select(udpfd + 1, &r, 0, 0, &timeout);
124 if (n <= 0)
125 {
126 printf("Timeout waiting for packet\n");
127 return 0;
128 }
129 }
130 if ((len = read_packet(udpfd, packet)))
131 {
132 printf("Received ");
133 dump_packet(packet, stdout);
134 }
135
136 return 0;
137 }
138