- patch from Denis Vlasenko to add and use bb_xchdir()

[oweals/busybox.git] / console-tools / deallocvt.c

diff --git a/console-tools/deallocvt.c b/console-tools/deallocvt.c
index 1e13ff69cef561b310453195fcb12c4b51a60d68..853b2b48975f4387677660d8b986b51720fdebd1 100644 (file)
--- a/console-tools/deallocvt.c
+++ b/console-tools/deallocvt.c
@@ -1,50 +1,56 @@
+/* vi: set sw=4 ts=4: */
 /*
- * disalloc.c - aeb - 940501 - Disallocate virtual terminal(s)
- * Renamed deallocvt.
+ * Disallocate virtual terminal(s)
+ *
+ * Copyright (C) 2003 by Tito Ragusa <farmatito@tiscali.it>
+ * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  */
-#include "internal.h"
+
+/* no options, no getopt */
+
 #include <stdlib.h>
+#include <stdio.h>
 #include <fcntl.h>
 #include <sys/types.h>
 #include <sys/ioctl.h>
-#include <linux/vt.h>
-#include <stdio.h>
+#include "busybox.h"
 
-extern int getfd(void);
-char *progname;
+/* From <linux/vt.h> */
+enum { VT_DISALLOCATE = 0x5608 }; /* free memory associated to vt */
 
-int
-deallocvt_main(int argc, char *argv[]) {
-    int fd, num, i;
+int deallocvt_main(int argc, char *argv[])
+{
+       /* num = 0 deallocate all unused consoles */
+       int num = 0;
 
-    if ( ( argc != 2) || (**(argv+1) == '-' ) ) {
-       usage ("deallocvt N\n\nDeallocate unused virtual terminal /dev/ttyN\n");
-    }
-
-    progname = argv[0];
-
-    fd = get_console_fd("/dev/console");
-
-    if (argc == 1) {
-       /* deallocate all unused consoles */
-       if (ioctl(fd,VT_DISALLOCATE,0)) {
-           perror("VT_DISALLOCATE");
-           exit(1);
+       switch (argc) {
+               case 2:
+                       if ((num = bb_xgetlarg(argv[1], 10, 0, INT_MAX)) == 0) {
+                               bb_error_msg_and_die("0: illegal VT number");
+                       }
+               /* Fallthrough */
+               case 1:
+                       break;
+               default:
+                       bb_show_usage();
        }
-    } else
-    for (i = 1; i < argc; i++) {
-       num = atoi(argv[i]);
-       if (num == 0)
-           fprintf(stderr, "%s: 0: illegal VT number\n", progname);
-       else if (num == 1)
-           fprintf(stderr, "%s: VT 1 cannot be deallocated\n", progname);
-       else
-       if (ioctl(fd,VT_DISALLOCATE,num)) {
-           perror("VT_DISALLOCATE");
-           fprintf(stderr, "%s: could not deallocate console %d\n",
-                   progname, num);
-           exit(1);
+
+       if (-1 == ioctl(get_console_fd(), VT_DISALLOCATE, num)) {
+               bb_perror_msg_and_die("VT_DISALLOCATE");
        }
-    }
-    exit(0);
+       return EXIT_SUCCESS;
 }