src/misc-progs/sambactrl.c

   1 #include <stdio.h>
   2 #include <string.h>
   3 #include <stdlib.h>
   4 #include <unistd.h>
   5 #include <sys/types.h>
   6 #include <fcntl.h>
   7 #include "setuid.h"
   8
   9 #define BUFFER_SIZE 1024
  10
  11 char command[BUFFER_SIZE];
  12
  13 int main(int argc, char *argv[]) {
  14         if (!(initsetuid()))
  15                 exit(1);
  16
  17         // Check what command is asked
  18         if (argc == 1) {
  19                 fprintf (stderr, "Missing smbctrl command!\n");
  20                 return 1;
  21
  22         } else if (strcmp(argv[1], "smbuserdisable") == 0) {
  23                 if (!is_valid_argument_alnum(argv[2])) {
  24                         fprintf(stderr, "Invalid username '%s'\n", argv[2]);
  25                         exit(2);
  26                 }
  27
  28                 snprintf(command, BUFFER_SIZE-1, "/usr/bin/smbpasswd -d %s >/dev/null", argv[2]);
  29                 safe_system(command);
  30
  31         } else if (strcmp(argv[1], "smbuserenable") == 0) {
  32                 if (!is_valid_argument_alnum(argv[2])) {
  33                         fprintf(stderr, "Invalid username '%s'\n", argv[2]);
  34                         exit(2);
  35                 }
  36
  37                 snprintf(command, BUFFER_SIZE-1, "/usr/bin/smbpasswd -e %s >/dev/null", argv[2]);
  38                 safe_system(command);
  39
  40         } else if (strcmp(argv[1], "smbuserdelete") == 0) {
  41                 if (!is_valid_argument_alnum(argv[2])) {
  42                         fprintf(stderr, "Invalid username '%s'\n", argv[2]);
  43                         exit(2);
  44                 }
  45
  46                 snprintf(command, BUFFER_SIZE-1, "/usr/bin/smbpasswd -x %s >/dev/null", argv[2]);
  47                 safe_system(command);
  48
  49                 snprintf(command, BUFFER_SIZE-1, "/usr/sbin/userdel %s >/dev/null", argv[2]);
  50                 safe_system(command);
  51
  52         } else if (strcmp(argv[1], "smbsafeconf") == 0) {
  53                 safe_system("/bin/cat /var/ipfire/samba/global /var/ipfire/samba/shares > /var/ipfire/samba/smb.conf");
  54
  55         } else if (strcmp(argv[1], "smbstop") == 0) {
  56                 safe_system("/etc/rc.d/init.d/samba stop >/dev/null");
  57                 safe_system("/usr/local/bin/sambactrl disable");
  58
  59         } else if (strcmp(argv[1], "smbstart") == 0) {
  60                 safe_system("/etc/rc.d/init.d/samba start >/dev/null");
  61                 safe_system("/usr/local/bin/sambactrl enable");
  62
  63         } else if (strcmp(argv[1], "smbrestart") == 0) {
  64                 safe_system("/etc/rc.d/init.d/samba restart >/dev/null");
  65
  66         } else if (strcmp(argv[1], "smbreload") == 0) {
  67                 safe_system("/etc/rc.d/init.d/samba reload >/dev/null");
  68
  69         } else if (strcmp(argv[1], "smbstatus") == 0) {
  70                 snprintf(command, BUFFER_SIZE-1, "/usr/bin/smbstatus 2>/dev/null");
  71                 safe_system(command);
  72
  73         } else if (strcmp(argv[1], "smbuseradd") == 0) {
  74                 if (!is_valid_argument_alnum(argv[2])) {
  75                         fprintf(stderr, "Invalid username '%s'\n", argv[2]);
  76                         exit(2);
  77                 }
  78
  79                 snprintf(command, BUFFER_SIZE-1, "/usr/sbin/groupadd sambauser >/dev/null");
  80                 safe_system(command);
  81
  82                 snprintf(command, BUFFER_SIZE-1, "/usr/sbin/useradd -c 'Samba User' -m -g sambauser -s /bin/false %s >/dev/null", argv[2]);
  83                 safe_system(command);
  84
  85                 run("/usr/sbin/samba-change-password", argv + 2);
  86
  87         } else if (strcmp(argv[1], "smbchangepw") == 0) {
  88                 if (!is_valid_argument_alnum(argv[2])) {
  89                         fprintf(stderr, "Invalid username '%s'\n", argv[2]);
  90                         exit(2);
  91                 }
  92
  93                 run("/usr/sbin/samba-change-password", argv + 2);
  94
  95         } else if (strcmp(argv[1], "readsmbpasswd") == 0) {
  96                 safe_system("/bin/chown root:nobody /var/ipfire/samba/private >/dev/null");
  97                 safe_system("/bin/chown root:nobody /var/ipfire/samba/private/smbpasswd >/dev/null");
  98                 safe_system("/bin/chmod 640 /var/ipfire/samba/private/smbpasswd >/dev/null");
  99                 safe_system("/bin/chmod 650 /var/ipfire/samba/private >/dev/null");
 100
 101         } else if (strcmp(argv[1], "locksmbpasswd") == 0) {
 102                 safe_system("/bin/chown root:root /var/ipfire/samba/private >/dev/null");
 103                 safe_system("/bin/chown root:root /var/ipfire/samba/private/smbpasswd >/dev/null");
 104                 safe_system("/bin/chmod 600 /var/ipfire/samba/private/smbpasswd >/dev/null");
 105                 safe_system("/bin/chmod 600 /var/ipfire/samba/private >/dev/null");
 106
 107         } else if (strcmp(argv[1], "enable") == 0) {
 108                 safe_system("touch /var/ipfire/samba/enable");
 109                 safe_system("ln -snf /etc/rc.d/init.d/samba /etc/rc.d/rc3.d/S45samba");
 110                 safe_system("ln -snf /etc/rc.d/init.d/samba /etc/rc.d/rc0.d/K48samba");
 111                 safe_system("ln -snf /etc/rc.d/init.d/samba /etc/rc.d/rc6.d/K48samba");
 112
 113         } else if (strcmp(argv[1], "disable") == 0) {
 114                 safe_system("unlink /var/ipfire/samba/enable");
 115                 safe_system("rm -rf /etc/rc.d/rc*.d/*samba");
 116
 117         } else if (strcmp(argv[1], "join") == 0) {
 118                 if (argc == 4) {
 119                         snprintf(command, BUFFER_SIZE - 1, "/usr/bin/net join -U \"%s%%%s\"",
 120                                 argv[2], argv[3]);
 121                         return safe_system(command);
 122                 } else {
 123                         fprintf(stderr, "Wrong number of arguments. Need username and password.\n");
 124                         return 1;
 125                 }
 126         }
 127
 128         return 0;
 129 }