04383f98f4d5ff9258109941ac19cfe387907778
[linux-2.6.git] / arch / um / drivers / mconsole_user.c
1 /*
2  * Copyright (C) 2001 Lennert Buytenhek (buytenh@gnu.org)
3  * Copyright (C) 2001 - 2003 Jeff Dike (jdike@addtoit.com)
4  * Licensed under the GPL
5  */
6
7 #include <stdio.h>
8 #include <stdlib.h>
9 #include <errno.h>
10 #include <signal.h>
11 #include <sys/socket.h>
12 #include <sys/types.h>
13 #include <sys/uio.h>
14 #include <sys/un.h>
15 #include <unistd.h>
16 #include "user.h"
17 #include "mconsole.h"
18 #include "umid.h"
19
20 static struct mconsole_command commands[] = {
21         { "version", mconsole_version, MCONSOLE_INTR },
22         { "halt", mconsole_halt, MCONSOLE_PROC },
23         { "reboot", mconsole_reboot, MCONSOLE_PROC },
24         { "config", mconsole_config, MCONSOLE_PROC },
25         { "remove", mconsole_remove, MCONSOLE_PROC },
26         { "sysrq", mconsole_sysrq, MCONSOLE_PROC },
27         { "help", mconsole_help, MCONSOLE_INTR },
28         { "cad", mconsole_cad, MCONSOLE_INTR },
29         { "stop", mconsole_stop, MCONSOLE_PROC },
30         { "go", mconsole_go, MCONSOLE_INTR },
31         { "log", mconsole_log, MCONSOLE_INTR },
32         { "proc", mconsole_proc, MCONSOLE_PROC },
33         { "stack", mconsole_stack, MCONSOLE_INTR },
34 };
35
36 /* Initialized in mconsole_init, which is an initcall */
37 char mconsole_socket_name[256];
38
39 int mconsole_reply_v0(struct mc_request *req, char *reply)
40 {
41         struct iovec iov;
42         struct msghdr msg;
43
44         iov.iov_base = reply;
45         iov.iov_len = strlen(reply);
46
47         msg.msg_name = &(req->origin);
48         msg.msg_namelen = req->originlen;
49         msg.msg_iov = &iov;
50         msg.msg_iovlen = 1;
51         msg.msg_control = NULL;
52         msg.msg_controllen = 0;
53         msg.msg_flags = 0;
54
55         return sendmsg(req->originating_fd, &msg, 0);
56 }
57
58 static struct mconsole_command *mconsole_parse(struct mc_request *req)
59 {
60         struct mconsole_command *cmd;
61         int i;
62
63         for(i=0;i<sizeof(commands)/sizeof(commands[0]);i++){
64                 cmd = &commands[i];
65                 if(!strncmp(req->request.data, cmd->command, 
66                             strlen(cmd->command))){
67                         return(cmd);
68                 }
69         }
70         return(NULL);
71 }
72
73 #define MIN(a,b) ((a)<(b) ? (a):(b))
74
75 #define STRINGX(x) #x
76 #define STRING(x) STRINGX(x)
77
78 int mconsole_get_request(int fd, struct mc_request *req)
79 {
80         int len;
81
82         req->originlen = sizeof(req->origin);
83         req->len = recvfrom(fd, &req->request, sizeof(req->request), 0,
84                             (struct sockaddr *) req->origin, &req->originlen);
85         if (req->len < 0)
86                 return 0;
87
88         req->originating_fd = fd;
89
90         if(req->request.magic != MCONSOLE_MAGIC){
91                 /* Unversioned request */
92                 len = MIN(sizeof(req->request.data) - 1, 
93                           strlen((char *) &req->request));
94                 memmove(req->request.data, &req->request, len);
95                 req->request.data[len] = '\0';
96
97                 req->request.magic = MCONSOLE_MAGIC;
98                 req->request.version = 0;
99                 req->request.len = len;
100
101                 mconsole_reply_v0(req, "ERR Version 0 mconsole clients are "
102                                   "not supported by this driver");
103                 return(0);
104         }
105
106         if(req->request.len >= MCONSOLE_MAX_DATA){
107                 mconsole_reply(req, "Request too large", 1, 0);
108                 return(0);
109         }
110         if(req->request.version != MCONSOLE_VERSION){
111                 mconsole_reply(req, "This driver only supports version " 
112                                STRING(MCONSOLE_VERSION) " clients", 1, 0);
113         }
114         
115         req->request.data[req->request.len] = '\0';
116         req->cmd = mconsole_parse(req);
117         if(req->cmd == NULL){
118                 mconsole_reply(req, "Unknown command", 1, 0);
119                 return(0);
120         }
121
122         return(1);
123 }
124
125 int mconsole_reply(struct mc_request *req, char *str, int err, int more)
126 {
127         struct mconsole_reply reply;
128         int total, len, n;
129
130         total = strlen(str);
131         do {
132                 reply.err = err;
133
134                 /* err can only be true on the first packet */
135                 err = 0;
136
137                 len = MIN(total, MCONSOLE_MAX_DATA - 1);
138
139                 if(len == total) reply.more = more;
140                 else reply.more = 1;
141
142                 memcpy(reply.data, str, len);
143                 reply.data[len] = '\0';
144                 total -= len;
145                 str += len;
146                 reply.len = len + 1;
147
148                 len = sizeof(reply) + reply.len - sizeof(reply.data);
149
150                 n = sendto(req->originating_fd, &reply, len, 0,
151                            (struct sockaddr *) req->origin, req->originlen);
152
153                 if(n < 0) return(-errno);
154         } while(total > 0);
155         return(0);
156 }
157
158 int mconsole_unlink_socket(void)
159 {
160         unlink(mconsole_socket_name);
161         return 0;
162 }
163
164 static int notify_sock = -1;
165
166 int mconsole_notify(char *sock_name, int type, const void *data, int len)
167 {
168         struct sockaddr_un target;
169         struct mconsole_notify packet;
170         int n, err = 0;
171
172         lock_notify();
173         if(notify_sock < 0){
174                 notify_sock = socket(PF_UNIX, SOCK_DGRAM, 0);
175                 if(notify_sock < 0){
176                         err = -errno;
177                         printk("mconsole_notify - socket failed, errno = %d\n",
178                                err);
179                 }
180         }
181         unlock_notify();
182         
183         if(err)
184                 return(err);
185
186         target.sun_family = AF_UNIX;
187         strcpy(target.sun_path, sock_name);
188
189         packet.magic = MCONSOLE_MAGIC;
190         packet.version = MCONSOLE_VERSION;
191         packet.type = type;
192         len = (len > sizeof(packet.data)) ? sizeof(packet.data) : len;
193         packet.len = len;
194         memcpy(packet.data, data, len);
195
196         err = 0;
197         len = sizeof(packet) + packet.len - sizeof(packet.data);
198         n = sendto(notify_sock, &packet, len, 0, (struct sockaddr *) &target, 
199                    sizeof(target));
200         if(n < 0){
201                 err = -errno;
202                 printk("mconsole_notify - sendto failed, errno = %d\n", errno);
203         }
204         return(err);
205 }
206
207 /*
208  * Overrides for Emacs so that we follow Linus's tabbing style.
209  * Emacs will notice this stuff at the end of the file and automatically
210  * adjust the settings for this buffer only.  This must remain at the end
211  * of the file.
212  * ---------------------------------------------------------------------------
213  * Local variables:
214  * c-file-style: "linux"
215  * End:
216  */