- Kai Germaschewski: ymfpci cleanups and resource leak fixes
[davej-history.git] / net / core / scm.c
blob0bd1b40047de577bb9aa4dddbf9291d51d5cc61c
1 /* scm.c - Socket level control messages processing.
3 * Author: Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
4 * Alignment and value checking mods by Craig Metz
6 * This program is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU General Public License
8 * as published by the Free Software Foundation; either version
9 * 2 of the License, or (at your option) any later version.
12 #include <linux/signal.h>
13 #include <linux/errno.h>
14 #include <linux/sched.h>
15 #include <linux/mm.h>
16 #include <linux/kernel.h>
17 #include <linux/major.h>
18 #include <linux/stat.h>
19 #include <linux/socket.h>
20 #include <linux/file.h>
21 #include <linux/fcntl.h>
22 #include <linux/net.h>
23 #include <linux/interrupt.h>
24 #include <linux/netdevice.h>
26 #include <asm/system.h>
27 #include <asm/uaccess.h>
29 #include <linux/inet.h>
30 #include <net/ip.h>
31 #include <net/protocol.h>
32 #include <net/tcp.h>
33 #include <net/udp.h>
34 #include <linux/skbuff.h>
35 #include <net/sock.h>
36 #include <net/scm.h>
40 * Only allow a user to send credentials, that they could set with
41 * setu(g)id.
44 static __inline__ int scm_check_creds(struct ucred *creds)
46 if ((creds->pid == current->pid || capable(CAP_SYS_ADMIN)) &&
47 ((creds->uid == current->uid || creds->uid == current->euid ||
48 creds->uid == current->suid) || capable(CAP_SETUID)) &&
49 ((creds->gid == current->gid || creds->gid == current->egid ||
50 creds->gid == current->sgid) || capable(CAP_SETGID))) {
51 return 0;
53 return -EPERM;
56 static int scm_fp_copy(struct cmsghdr *cmsg, struct scm_fp_list **fplp)
58 int *fdp = (int*)CMSG_DATA(cmsg);
59 struct scm_fp_list *fpl = *fplp;
60 struct file **fpp;
61 int i, num;
63 num = (cmsg->cmsg_len - CMSG_ALIGN(sizeof(struct cmsghdr)))/sizeof(int);
65 if (num <= 0)
66 return 0;
68 if (num > SCM_MAX_FD)
69 return -EINVAL;
71 if (!fpl)
73 fpl = kmalloc(sizeof(struct scm_fp_list), GFP_KERNEL);
74 if (!fpl)
75 return -ENOMEM;
76 *fplp = fpl;
77 fpl->count = 0;
79 fpp = &fpl->fp[fpl->count];
81 if (fpl->count + num > SCM_MAX_FD)
82 return -EINVAL;
85 * Verify the descriptors and increment the usage count.
88 for (i=0; i< num; i++)
90 int fd = fdp[i];
91 struct file *file;
93 if (fd < 0 || !(file = fget(fd)))
94 return -EBADF;
95 *fpp++ = file;
96 fpl->count++;
98 return num;
101 void __scm_destroy(struct scm_cookie *scm)
103 struct scm_fp_list *fpl = scm->fp;
104 int i;
106 if (fpl) {
107 scm->fp = NULL;
108 for (i=fpl->count-1; i>=0; i--)
109 fput(fpl->fp[i]);
110 kfree(fpl);
114 int __scm_send(struct socket *sock, struct msghdr *msg, struct scm_cookie *p)
116 struct cmsghdr *cmsg;
117 int err;
119 for (cmsg = CMSG_FIRSTHDR(msg); cmsg; cmsg = CMSG_NXTHDR(msg, cmsg))
121 err = -EINVAL;
123 /* Verify that cmsg_len is at least sizeof(struct cmsghdr) */
124 /* The first check was omitted in <= 2.2.5. The reasoning was
125 that parser checks cmsg_len in any case, so that
126 additional check would be work duplication.
127 But if cmsg_level is not SOL_SOCKET, we do not check
128 for too short ancillary data object at all! Oops.
129 OK, let's add it...
131 if (cmsg->cmsg_len < sizeof(struct cmsghdr) ||
132 (unsigned long)(((char*)cmsg - (char*)msg->msg_control)
133 + cmsg->cmsg_len) > msg->msg_controllen)
134 goto error;
136 if (cmsg->cmsg_level != SOL_SOCKET)
137 continue;
139 switch (cmsg->cmsg_type)
141 case SCM_RIGHTS:
142 err=scm_fp_copy(cmsg, &p->fp);
143 if (err<0)
144 goto error;
145 break;
146 case SCM_CREDENTIALS:
147 if (cmsg->cmsg_len != CMSG_LEN(sizeof(struct ucred)))
148 goto error;
149 memcpy(&p->creds, CMSG_DATA(cmsg), sizeof(struct ucred));
150 err = scm_check_creds(&p->creds);
151 if (err)
152 goto error;
153 break;
154 default:
155 goto error;
159 if (p->fp && !p->fp->count)
161 kfree(p->fp);
162 p->fp = NULL;
164 return 0;
166 error:
167 scm_destroy(p);
168 return err;
171 int put_cmsg(struct msghdr * msg, int level, int type, int len, void *data)
173 struct cmsghdr *cm = (struct cmsghdr*)msg->msg_control;
174 struct cmsghdr cmhdr;
175 int cmlen = CMSG_LEN(len);
176 int err;
178 if (cm==NULL || msg->msg_controllen < sizeof(*cm)) {
179 msg->msg_flags |= MSG_CTRUNC;
180 return 0; /* XXX: return error? check spec. */
182 if (msg->msg_controllen < cmlen) {
183 msg->msg_flags |= MSG_CTRUNC;
184 cmlen = msg->msg_controllen;
186 cmhdr.cmsg_level = level;
187 cmhdr.cmsg_type = type;
188 cmhdr.cmsg_len = cmlen;
190 err = -EFAULT;
191 if (copy_to_user(cm, &cmhdr, sizeof cmhdr))
192 goto out;
193 if (copy_to_user(CMSG_DATA(cm), data, cmlen - sizeof(struct cmsghdr)))
194 goto out;
195 cmlen = CMSG_SPACE(len);
196 msg->msg_control += cmlen;
197 msg->msg_controllen -= cmlen;
198 err = 0;
199 out:
200 return err;
203 void scm_detach_fds(struct msghdr *msg, struct scm_cookie *scm)
205 struct cmsghdr *cm = (struct cmsghdr*)msg->msg_control;
207 int fdmax = 0;
208 int fdnum = scm->fp->count;
209 struct file **fp = scm->fp->fp;
210 int *cmfptr;
211 int err = 0, i;
213 if (msg->msg_controllen > sizeof(struct cmsghdr))
214 fdmax = ((msg->msg_controllen - sizeof(struct cmsghdr))
215 / sizeof(int));
217 if (fdnum < fdmax)
218 fdmax = fdnum;
220 for (i=0, cmfptr=(int*)CMSG_DATA(cm); i<fdmax; i++, cmfptr++)
222 int new_fd;
223 err = get_unused_fd();
224 if (err < 0)
225 break;
226 new_fd = err;
227 err = put_user(new_fd, cmfptr);
228 if (err) {
229 put_unused_fd(new_fd);
230 break;
232 /* Bump the usage count and install the file. */
233 get_file(fp[i]);
234 fd_install(new_fd, fp[i]);
237 if (i > 0)
239 int cmlen = CMSG_LEN(i*sizeof(int));
240 if (!err)
241 err = put_user(SOL_SOCKET, &cm->cmsg_level);
242 if (!err)
243 err = put_user(SCM_RIGHTS, &cm->cmsg_type);
244 if (!err)
245 err = put_user(cmlen, &cm->cmsg_len);
246 if (!err) {
247 cmlen = CMSG_SPACE(i*sizeof(int));
248 msg->msg_control += cmlen;
249 msg->msg_controllen -= cmlen;
252 if (i < fdnum || (fdnum && fdmax <= 0))
253 msg->msg_flags |= MSG_CTRUNC;
256 * All of the files that fit in the message have had their
257 * usage counts incremented, so we just free the list.
259 __scm_destroy(scm);
262 struct scm_fp_list *scm_fp_dup(struct scm_fp_list *fpl)
264 struct scm_fp_list *new_fpl;
265 int i;
267 if (!fpl)
268 return NULL;
270 new_fpl = kmalloc(sizeof(*fpl), GFP_KERNEL);
271 if (new_fpl) {
272 for (i=fpl->count-1; i>=0; i--)
273 get_file(fpl->fp[i]);
274 memcpy(new_fpl, fpl, sizeof(*fpl));
276 return new_fpl;