Add a missing initbufbio() to fix a panic when vinum tries to issue a
[dragonfly.git] / usr.sbin / wormcontrol / wormcontrol.c
1 /* 
2  * Copyright (C) 1996
3  *   interface business GmbH
4  *   Tolkewitzer Strasse 49
5  *   D-01277 Dresden
6  *   F.R. Germany
7  *
8  * All rights reserved.
9  *
10  * Written by Joerg Wunsch <joerg_wunsch@interface-business.de>
11  *
12  * 
13  * Redistribution and use in source and binary forms, with or without
14  * modification, are permitted provided that the following conditions
15  * are met:
16  * 1. Redistributions of source code must retain the above copyright
17  *    notice, this list of conditions and the following disclaimer.
18  * 2. Redistributions in binary form must reproduce the above copyright
19  *    notice, this list of conditions and the following disclaimer in the
20  *    documentation and/or other materials provided with the distribution.
21  *
22  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY
23  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
25  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE
26  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
27  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT
28  * OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
29  * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
30  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
31  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
32  * USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
33  * DAMAGE.
34  *
35  * $FreeBSD: src/usr.sbin/wormcontrol/wormcontrol.c,v 1.11 1999/10/01 14:12:46 peter Exp $
36  * $DragonFly: src/usr.sbin/wormcontrol/Attic/wormcontrol.c,v 1.3 2005/12/05 01:23:23 swildner Exp $
37  */
38
39 #include <err.h>
40 #include <fcntl.h>
41 #include <stdio.h>
42 #include <stdlib.h>
43 #include <string.h>
44 #include <sysexits.h>
45 #include <unistd.h>
46
47 #include <sys/ioctl.h>
48 #include <sys/wormio.h>
49
50 static void
51 usage(void)
52 {
53         fprintf(stderr,"usage: wormcontrol [-f device] command [args]\n");
54         exit(EX_USAGE);
55 }
56
57 #define eq(a, b) (strcmp(a, b) == 0)
58
59 int
60 main(int argc, char **argv)
61 {
62         int fd, c, i;
63         int errs = 0;
64         const char *devname = "/dev/rworm0";
65
66         while ((c = getopt(argc, argv, "f:")) != -1)
67                 switch(c) {
68                 case 'f':
69                         devname = optarg;
70                         break;
71
72                 case '?':
73                 default:
74                         errs++;
75                 }
76         
77         argc -= optind;
78         argv += optind;
79
80         if (errs || argc < 1)
81                 usage();
82
83         if ((fd = open(devname, O_RDONLY | O_NONBLOCK, 0)) == -1)
84                 err(EX_NOINPUT, "open(%s)", devname);
85
86         if (eq(argv[0], "prepdisk")) {
87                 struct wormio_prepare_disk d;
88                 d.dummy = 0;
89                 d.speed = -1;
90                 for (i = 1; i < argc; i++) {
91                         if (eq(argv[i], "dummy"))
92                                 d.dummy = 1;
93                         else if (eq(argv[i], "single"))
94                                 d.speed = 1;
95                         else if (eq(argv[i], "double"))
96                                 d.speed = 2;
97                         else if (eq(argv[i], "quad"))
98                                 d.speed = 4;
99                         else if (eq(argv[i], "max"))
100                                 d.speed = 99;
101                         else
102                                 errx(EX_USAGE,
103                                      "wrong param for \"prepdisk\": %s",
104                                      argv[i]);
105                 }
106                 if (d.speed == -1)
107                         errx(EX_USAGE, "missing speed parameter");
108                 if (ioctl(fd, WORMIOCPREPDISK, &d) == -1)
109                         err(EX_IOERR, "ioctl(WORMIOCPREPDISK)");
110         }
111         else if (eq(argv[0], "track")) {
112                 struct wormio_prepare_track t;
113                 bzero(&t, sizeof (t));
114                 t.audio = -1;
115                 t.preemp = 0;
116                 for (i = 1; i < argc; i++) {
117                         if (eq(argv[i], "audio"))
118                                 t.audio = 1;
119                         else if (eq(argv[i], "data")) {
120                                 t.audio = 0;
121                                 t.track_type = BLOCK_MODE_1;
122                         } else if (eq(argv[i], "preemp"))
123                                 t.preemp = 1;
124                         else
125                                 errx(EX_USAGE,
126                                      "wrong param for \"track\": %s",
127                                      argv[i]);
128                 }
129                 if (t.audio == -1)
130                         errx(EX_USAGE, "missing track type parameter");
131                 if (t.audio == 0 && t.preemp == 1)
132                         errx(EX_USAGE, "\"preemp\" attempted on data track");
133                 if (ioctl(fd, WORMIOCPREPTRACK, &t) == -1)
134                         err(EX_IOERR, "ioctl(WORMIOCPREPTRACK)");
135         }
136         else if (eq(argv[0], "fixate")) {
137                 struct wormio_fixation f;
138                 f.toc_type = -1;
139                 f.onp = 0;
140                 for (i = 1; i < argc; i++) {
141                         if (eq(argv[i], "onp"))
142                                 f.onp = 1;
143                         else if (argv[i][0] >= '0' && argv[i][0] <= '4' &&
144                                  argv[i][1] == '\0')
145                                 f.toc_type = argv[i][0] - '0';
146                         else
147                                 errx(EX_USAGE,
148                                      "wrong param for \"fixate\": %s",
149                                      argv[i]);
150                 }
151                 if (f.toc_type == -1)
152                         errx(EX_USAGE, "missing TOC type parameter");
153                 if (ioctl(fd, WORMIOCFIXATION, &f) == -1)
154                         err(EX_IOERR, "ioctl(WORMIOFIXATION)");
155         }
156         else if (eq(argv[0], "blank")) {
157                 if (ioctl(fd, CDRIOCBLANK) == -1)
158                         err(EX_IOERR, "ioctl(CDRIOCBLANK)");
159         }
160         else if (eq(argv[0], "nextwriteable")) {
161                 int addr;
162                 if (ioctl(fd, CDRIOCNEXTWRITEABLEADDR, &addr) == -1)
163                         err(EX_IOERR, "ioctl(CDRIOCNEXTWRITEABLEADDR)");
164                 printf("%d\n", addr);
165         }
166         else
167                 errx(EX_USAGE, "unknown command: %s", argv[0]);
168
169         return EX_OK;
170 }