1 /*- 2 * Copyright (c) 2002 Poul-Henning Kamp 3 * Copyright (c) 2002 Networks Associates Technology, Inc. 4 * All rights reserved. 5 * 6 * This software was developed for the FreeBSD Project by Poul-Henning Kamp 7 * and NAI Labs, the Security Research Division of Network Associates, Inc. 8 * under DARPA/SPAWAR contract N66001-01-C-8035 ("CBOSS"), as part of the 9 * DARPA CHATS research program. 10 * 11 * Redistribution and use in source and binary forms, with or without 12 * modification, are permitted provided that the following conditions 13 * are met: 14 * 1. Redistributions of source code must retain the above copyright 15 * notice, this list of conditions and the following disclaimer. 16 * 2. Redistributions in binary form must reproduce the above copyright 17 * notice, this list of conditions and the following disclaimer in the 18 * documentation and/or other materials provided with the distribution. 19 * 3. The names of the authors may not be used to endorse or promote 20 * products derived from this software without specific prior written 21 * permission. 22 * 23 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 24 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 25 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 26 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 27 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 28 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 29 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 30 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 31 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 32 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 33 * SUCH DAMAGE. 34 */ 35 36 #include <sys/cdefs.h> 37 __FBSDID("$FreeBSD$"); 38 39 #include "opt_geom.h" 40 41 #include <sys/param.h> 42 #include <sys/systm.h> 43 #include <sys/kernel.h> 44 #include <sys/sysctl.h> 45 #include <sys/bio.h> 46 #include <sys/conf.h> 47 #include <sys/disk.h> 48 #include <sys/malloc.h> 49 #include <sys/sysctl.h> 50 #include <sys/sbuf.h> 51 52 #include <sys/lock.h> 53 #include <sys/mutex.h> 54 55 #include <vm/vm.h> 56 #include <vm/vm_extern.h> 57 58 #include <geom/geom.h> 59 #include <geom/geom_int.h> 60 #define GCTL_TABLE 1 61 #include <geom/geom_ctl.h> 62 63 #include <machine/stdarg.h> 64 65 static d_ioctl_t g_ctl_ioctl; 66 67 static struct cdevsw g_ctl_cdevsw = { 68 .d_version = D_VERSION, 69 .d_flags = D_NEEDGIANT, 70 .d_ioctl = g_ctl_ioctl, 71 .d_name = "g_ctl", 72 }; 73 74 void 75 g_ctl_init(void) 76 { 77 78 make_dev(&g_ctl_cdevsw, 0, 79 UID_ROOT, GID_OPERATOR, 0640, PATH_GEOM_CTL); 80 KASSERT(GCTL_PARAM_RD == VM_PROT_READ, 81 ("GCTL_PARAM_RD != VM_PROT_READ")); 82 KASSERT(GCTL_PARAM_WR == VM_PROT_WRITE, 83 ("GCTL_PARAM_WR != VM_PROT_WRITE")); 84 } 85 86 /* 87 * Report an error back to the user in ascii format. Return whatever copyout 88 * returned, or EINVAL if it succeeded. 89 */ 90 int 91 gctl_error(struct gctl_req *req, const char *fmt, ...) 92 { 93 va_list ap; 94 95 if (req == NULL) 96 return (EINVAL); 97 98 /* We only record the first error */ 99 if (sbuf_done(req->serror)) { 100 if (!req->nerror) 101 req->nerror = EEXIST; 102 } 103 if (req->nerror) 104 return (req->nerror); 105 106 va_start(ap, fmt); 107 sbuf_vprintf(req->serror, fmt, ap); 108 va_end(ap); 109 sbuf_finish(req->serror); 110 if (g_debugflags & G_F_CTLDUMP) 111 printf("gctl %p error \"%s\"\n", req, sbuf_data(req->serror)); 112 return (0); 113 } 114 115 /* 116 * Allocate space and copyin() something. 117 * XXX: this should really be a standard function in the kernel. 118 */ 119 static void * 120 geom_alloc_copyin(struct gctl_req *req, void *uaddr, size_t len) 121 { 122 void *ptr; 123 124 ptr = g_malloc(len, M_WAITOK); 125 if (ptr == NULL) 126 req->nerror = ENOMEM; 127 else 128 req->nerror = copyin(uaddr, ptr, len); 129 if (!req->nerror) 130 return (ptr); 131 if (ptr != NULL) 132 g_free(ptr); 133 return (NULL); 134 } 135 136 static void 137 gctl_copyin(struct gctl_req *req) 138 { 139 int error, i; 140 struct gctl_req_arg *ap; 141 char *p; 142 143 ap = geom_alloc_copyin(req, req->arg, req->narg * sizeof(*ap)); 144 if (ap == NULL) { 145 req->nerror = ENOMEM; 146 req->arg = NULL; 147 return; 148 } 149 150 /* Nothing have been copyin()'ed yet */ 151 for (i = 0; i < req->narg; i++) { 152 ap[i].flag &= ~(GCTL_PARAM_NAMEKERNEL|GCTL_PARAM_VALUEKERNEL); 153 ap[i].flag &= ~GCTL_PARAM_CHANGED; 154 ap[i].kvalue = NULL; 155 } 156 157 error = 0; 158 for (i = 0; i < req->narg; i++) { 159 if (ap[i].nlen < 1 || ap[i].nlen > SPECNAMELEN) { 160 error = gctl_error(req, 161 "wrong param name length %d: %d", i, ap[i].nlen); 162 break; 163 } 164 p = geom_alloc_copyin(req, ap[i].name, ap[i].nlen); 165 if (p == NULL) 166 break; 167 if (p[ap[i].nlen - 1] != '\0') { 168 error = gctl_error(req, "unterminated param name"); 169 g_free(p); 170 break; 171 } 172 ap[i].name = p; 173 ap[i].flag |= GCTL_PARAM_NAMEKERNEL; 174 if (ap[i].len <= 0) { 175 error = gctl_error(req, "negative param length"); 176 break; 177 } 178 p = geom_alloc_copyin(req, ap[i].value, ap[i].len); 179 if (p == NULL) 180 break; 181 if ((ap[i].flag & GCTL_PARAM_ASCII) && 182 p[ap[i].len - 1] != '\0') { 183 error = gctl_error(req, "unterminated param value"); 184 g_free(p); 185 break; 186 } 187 ap[i].kvalue = p; 188 ap[i].flag |= GCTL_PARAM_VALUEKERNEL; 189 } 190 req->arg = ap; 191 return; 192 } 193 194 static void 195 gctl_copyout(struct gctl_req *req) 196 { 197 int error, i; 198 struct gctl_req_arg *ap; 199 200 if (req->nerror) 201 return; 202 error = 0; 203 ap = req->arg; 204 for (i = 0; i < req->narg; i++, ap++) { 205 if (!(ap->flag & GCTL_PARAM_CHANGED)) 206 continue; 207 error = copyout(ap->kvalue, ap->value, ap->len); 208 if (!error) 209 continue; 210 req->nerror = error; 211 return; 212 } 213 return; 214 } 215 216 static void 217 gctl_free(struct gctl_req *req) 218 { 219 int i; 220 221 if (req->arg == NULL) 222 return; 223 for (i = 0; i < req->narg; i++) { 224 if (req->arg[i].flag & GCTL_PARAM_NAMEKERNEL) 225 g_free(req->arg[i].name); 226 if ((req->arg[i].flag & GCTL_PARAM_VALUEKERNEL) && 227 req->arg[i].len > 0) 228 g_free(req->arg[i].kvalue); 229 } 230 g_free(req->arg); 231 sbuf_delete(req->serror); 232 } 233 234 static void 235 gctl_dump(struct gctl_req *req) 236 { 237 u_int i; 238 int j; 239 struct gctl_req_arg *ap; 240 241 printf("Dump of gctl request at %p:\n", req); 242 if (req->nerror > 0) { 243 printf(" nerror:\t%d\n", req->nerror); 244 if (sbuf_len(req->serror) > 0) 245 printf(" error:\t\"%s\"\n", sbuf_data(req->serror)); 246 } 247 for (i = 0; i < req->narg; i++) { 248 ap = &req->arg[i]; 249 if (!(ap->flag & GCTL_PARAM_NAMEKERNEL)) 250 printf(" param:\t%d@%p", ap->nlen, ap->name); 251 else 252 printf(" param:\t\"%s\"", ap->name); 253 printf(" [%s%s%d] = ", 254 ap->flag & GCTL_PARAM_RD ? "R" : "", 255 ap->flag & GCTL_PARAM_WR ? "W" : "", 256 ap->len); 257 if (!(ap->flag & GCTL_PARAM_VALUEKERNEL)) { 258 printf(" =@ %p", ap->value); 259 } else if (ap->flag & GCTL_PARAM_ASCII) { 260 printf("\"%s\"", (char *)ap->kvalue); 261 } else if (ap->len > 0) { 262 for (j = 0; j < ap->len && j < 512; j++) 263 printf(" %02x", ((u_char *)ap->kvalue)[j]); 264 } else { 265 printf(" = %p", ap->kvalue); 266 } 267 printf("\n"); 268 } 269 } 270 271 void 272 gctl_set_param(struct gctl_req *req, const char *param, void const *ptr, int len) 273 { 274 int i; 275 struct gctl_req_arg *ap; 276 277 for (i = 0; i < req->narg; i++) { 278 ap = &req->arg[i]; 279 if (strcmp(param, ap->name)) 280 continue; 281 if (!(ap->flag & GCTL_PARAM_WR)) { 282 gctl_error(req, "No write access %s argument", param); 283 return; 284 } 285 if (ap->len < len) { 286 gctl_error(req, "Wrong length %s argument", param); 287 return; 288 } 289 bcopy(ptr, ap->kvalue, len); 290 ap->flag |= GCTL_PARAM_CHANGED; 291 return; 292 } 293 gctl_error(req, "Missing %s argument", param); 294 return; 295 } 296 297 void * 298 gctl_get_param(struct gctl_req *req, const char *param, int *len) 299 { 300 int i; 301 void *p; 302 struct gctl_req_arg *ap; 303 304 for (i = 0; i < req->narg; i++) { 305 ap = &req->arg[i]; 306 if (strcmp(param, ap->name)) 307 continue; 308 if (!(ap->flag & GCTL_PARAM_RD)) 309 continue; 310 p = ap->kvalue; 311 if (len != NULL) 312 *len = ap->len; 313 return (p); 314 } 315 return (NULL); 316 } 317 318 char const * 319 gctl_get_asciiparam(struct gctl_req *req, const char *param) 320 { 321 int i; 322 char const *p; 323 struct gctl_req_arg *ap; 324 325 for (i = 0; i < req->narg; i++) { 326 ap = &req->arg[i]; 327 if (strcmp(param, ap->name)) 328 continue; 329 if (!(ap->flag & GCTL_PARAM_RD)) 330 continue; 331 p = ap->kvalue; 332 if (ap->len < 1) { 333 gctl_error(req, "No length argument (%s)", param); 334 return (NULL); 335 } 336 if (p[ap->len - 1] != '\0') { 337 gctl_error(req, "Unterminated argument (%s)", param); 338 return (NULL); 339 } 340 return (p); 341 } 342 return (NULL); 343 } 344 345 void * 346 gctl_get_paraml(struct gctl_req *req, const char *param, int len) 347 { 348 int i; 349 void *p; 350 351 p = gctl_get_param(req, param, &i); 352 if (p == NULL) 353 gctl_error(req, "Missing %s argument", param); 354 else if (i != len) { 355 p = NULL; 356 gctl_error(req, "Wrong length %s argument", param); 357 } 358 return (p); 359 } 360 361 struct g_class * 362 gctl_get_class(struct gctl_req *req, char const *arg) 363 { 364 char const *p; 365 struct g_class *cp; 366 367 p = gctl_get_asciiparam(req, arg); 368 if (p == NULL) 369 return (NULL); 370 LIST_FOREACH(cp, &g_classes, class) { 371 if (!strcmp(p, cp->name)) 372 return (cp); 373 } 374 return (NULL); 375 } 376 377 struct g_geom * 378 gctl_get_geom(struct gctl_req *req, struct g_class *mpr, char const *arg) 379 { 380 char const *p; 381 struct g_class *mp; 382 struct g_geom *gp; 383 384 p = gctl_get_asciiparam(req, arg); 385 if (p != NULL) { 386 LIST_FOREACH(mp, &g_classes, class) { 387 if (mpr != NULL && mpr != mp) 388 continue; 389 LIST_FOREACH(gp, &mp->geom, geom) { 390 if (!strcmp(p, gp->name)) 391 return (gp); 392 } 393 } 394 } 395 gctl_error(req, "Geom not found"); 396 return (NULL); 397 } 398 399 struct g_provider * 400 gctl_get_provider(struct gctl_req *req, char const *arg) 401 { 402 char const *p; 403 struct g_provider *pp; 404 405 p = gctl_get_asciiparam(req, arg); 406 if (p == NULL) 407 return (NULL); 408 pp = g_provider_by_name(p); 409 if (pp != NULL) 410 return (pp); 411 gctl_error(req, "Provider not found"); 412 return (NULL); 413 } 414 415 static void 416 g_ctl_req(void *arg, int flag __unused) 417 { 418 struct g_class *mp; 419 struct gctl_req *req; 420 char const *verb; 421 422 g_topology_assert(); 423 req = arg; 424 mp = gctl_get_class(req, "class"); 425 if (mp == NULL) { 426 gctl_error(req, "Class not found"); 427 return; 428 } 429 if (mp->ctlreq == NULL) { 430 gctl_error(req, "Class takes no requests"); 431 return; 432 } 433 verb = gctl_get_param(req, "verb", NULL); 434 if (verb == NULL) { 435 gctl_error(req, "Verb missing"); 436 return; 437 } 438 mp->ctlreq(req, mp, verb); 439 g_topology_assert(); 440 } 441 442 443 static int 444 g_ctl_ioctl_ctl(struct cdev *dev, u_long cmd, caddr_t data, int fflag, struct thread *td) 445 { 446 struct gctl_req *req; 447 int nerror; 448 449 req = (void *)data; 450 req->nerror = 0; 451 req->serror = sbuf_new(NULL, NULL, 0, SBUF_AUTOEXTEND); 452 /* It is an error if we cannot return an error text */ 453 if (req->lerror < 2) 454 return (EINVAL); 455 if (!useracc(req->error, req->lerror, VM_PROT_WRITE)) 456 return (EINVAL); 457 458 /* Check the version */ 459 if (req->version != GCTL_VERSION) 460 return (gctl_error(req, 461 "kernel and libgeom version mismatch.")); 462 463 /* Get things on board */ 464 gctl_copyin(req); 465 466 if (g_debugflags & G_F_CTLDUMP) 467 gctl_dump(req); 468 469 if (!req->nerror) { 470 g_waitfor_event(g_ctl_req, req, M_WAITOK, NULL); 471 gctl_copyout(req); 472 } 473 if (sbuf_done(req->serror)) { 474 req->nerror = copyout(sbuf_data(req->serror), req->error, 475 imin(req->lerror, sbuf_len(req->serror) + 1)); 476 } 477 478 nerror = req->nerror; 479 gctl_free(req); 480 return (nerror); 481 } 482 483 static int 484 g_ctl_ioctl(struct cdev *dev, u_long cmd, caddr_t data, int fflag, struct thread *td) 485 { 486 int error; 487 488 switch(cmd) { 489 case GEOM_CTL: 490 error = g_ctl_ioctl_ctl(dev, cmd, data, fflag, td); 491 break; 492 default: 493 error = ENOIOCTL; 494 break; 495 } 496 return (error); 497 498 } 499