2 * Command line oriented Moira users tool.
4 * zacheiss@mit.edu, September 2001
8 * Copyright (C) 2000, 2001 by the Massachusetts Institute of Technology.
9 * For copying and distribution information, please see the file
13 #include <mit-copyright.h>
15 #include <moira_site.h>
33 struct string_list *next;
42 /* argument parsing macro */
43 #define argis(a, b) (!strcmp(*arg + 1, a) || !strcmp(*arg + 1, b))
45 /* flags from command line */
46 int info_flag, update_flag, create_flag, deact_flag, reg_flag;
47 int list_res_flag, update_res_flag, unformatted_flag, verbose, noauth;
49 struct owner_type *sponsor;
50 struct string_list *reservation_add_queue, *reservation_remove_queue;
52 char *username, *whoami;
54 char *newlogin, *uid, *shell, *winshell, *last, *first, *middle, *u_status;
55 char *clearid, *class, *comment, *secure, *winhomedir, *winprofiledir, *expiration;
57 struct owner_type *parse_member(char *s);
59 static char *states[] = {
62 "Half Registered (2)",
64 "Not registerable (4)",
65 "Enrolled/Registerable (5)",
66 "Enrolled/Not Registerable (6)",
68 "Registerable, Kerberos only (8)",
69 "Active, Kerberos only (9)"
72 static char *UserState(int state)
74 static char buf[BUFSIZ];
76 if (state < 0 || state >= US_END)
78 sprintf(buf, "Unknown (%d)", state);
84 void usage(char **argv);
85 int save_query_info(int argc, char **argv, void *hint);
86 int show_reservations(int argc, char **argv, void *hint);
87 void show_user_info(char **argv);
88 void show_user_info_unformatted(char **argv);
89 struct string_list *add_to_string_list(struct string_list *old_list, char *s);
90 int wrap_mr_query(char *handle, int argc, char **argv,
91 int (*callback)(int, char **, void *), void *callarg);
92 void print_query(char *query_name, int argc, char **argv);
94 int main(int argc, char **argv)
100 /* clear all flags & lists */
101 info_flag = update_flag = create_flag = deact_flag = reg_flag = 0;
102 list_res_flag = update_res_flag = unformatted_flag = verbose = noauth = 0;
103 newlogin = uid = shell = winshell = last = first = middle = NULL;
104 u_status = clearid = class = comment = secure = NULL;
105 winhomedir = winprofiledir = expiration = NULL;
106 reservation_add_queue = reservation_remove_queue = NULL;
111 while (++arg - argv < argc)
115 if (argis("i", "info"))
117 else if (argis("C", "create"))
119 else if (argis("D", "deact"))
121 else if (argis("r", "register"))
123 else if (argis("R", "rename")) {
124 if (arg - argv < argc - 1) {
131 else if (argis("U", "uid")) {
132 if (arg - argv < argc - 1) {
139 else if (argis("s", "shell")) {
140 if (arg - argv < argc - 1) {
147 else if (argis("w", "winshell")) {
148 if (arg - argv < argc - 1) {
155 else if (argis("L", "last")) {
156 if (arg - argv < argc - 1) {
163 else if (argis("F", "first")) {
164 if (arg - argv < argc - 1) {
171 else if (argis("M", "middle")) {
172 if (arg - argv < argc - 1) {
179 else if (argis("S", "status")) {
180 if (arg - argv < argc - 1) {
187 len = strlen(u_status);
188 for (i = 0; i < len; i++) {
189 if (!isdigit(u_status[i])) {
190 printf("Error: status code %s is not numeric.\n", u_status);
197 else if (argis("I", "mitid")) {
198 if (arg - argv < argc - 1) {
205 else if (argis("cl", "class")) {
206 if (arg - argv < argc - 1) {
213 else if (argis("c", "comment")) {
214 if (arg - argv < argc - 1) {
221 else if (argis("6", "secure")) {
222 if (arg - argv < argc - 1) {
229 else if (argis("wh", "winhomedir")) {
230 if (arg - argv < argc - 1) {
237 else if (argis("wp", "winprofiledir")) {
238 if (arg - argv < argc - 1) {
241 winprofiledir = *arg;
245 else if (argis("sp", "sponsor")) {
246 if (arg - argv < argc - 1) {
249 sponsor = parse_member(*arg);
253 else if (argis("e", "expiration")) {
254 if (arg - argv < argc - 1) {
261 else if (argis("ar", "addreservation")) {
262 if (arg - argv < argc - 1) {
264 reservation_add_queue = add_to_string_list(reservation_add_queue,
270 else if (argis("dr", "deletereservation")) {
271 if (arg - argv < argc - 1) {
273 reservation_remove_queue = add_to_string_list(reservation_remove_queue, *arg);
278 else if (argis("lr", "listreservation"))
280 else if (argis("u", "unformatted"))
282 else if (argis("n", "noauth"))
284 else if (argis("v", "verbose"))
286 else if (argis("db", "database"))
288 if (arg - argv < argc - 1)
299 else if (username == NULL)
304 if (username == NULL && !create_flag)
307 /* default to info_flag if nothing else was specified */
308 if(!(info_flag || update_flag || create_flag || \
309 deact_flag || reg_flag || list_res_flag || \
315 status = mrcl_connect(server, "stanley", 12, !noauth);
316 if (status == MRCL_AUTH_ERROR)
318 com_err(whoami, 0, "Try the -noauth flag if you don't "
319 "need authentication.");
324 /* create if needed */
330 for (cnt = 0; cnt < 20; cnt++) {
335 argv[U_NAME] = username;
337 argv[U_NAME] = UNIQUE_LOGIN;
341 argv[U_UID] = UNIQUE_UID;
343 argv[U_SHELL] = shell;
345 argv[U_SHELL] = "/bin/athena/bash";
347 argv[U_WINCONSOLESHELL] = winshell;
349 argv[U_WINCONSOLESHELL] = "cmd";
353 argv[U_FIRST] = first;
355 argv[U_MIDDLE] = middle;
357 argv[U_STATE] = u_status;
361 argv[U_MITID] = clearid;
363 argv[U_CLASS] = class;
365 argv[U_COMMENT] = comment;
366 /* Signature field always is the empty string */
367 argv[U_SIGNATURE] = "";
369 argv[U_SECURE] = secure;
371 argv[U_SECURE] = "0";
373 argv[U_WINHOMEDIR] = winhomedir;
375 argv[U_WINHOMEDIR] = "[DFS]";
377 argv[U_WINPROFILEDIR] = winprofiledir;
379 argv[U_WINPROFILEDIR] = "[DFS]";
381 argv[U_EXPIRATION] = expiration;
384 argv[U_SPONSOR_NAME] = sponsor->name;
385 switch (sponsor->type)
389 argv[U_SPONSOR_TYPE] = "USER";
390 status = wrap_mr_query("add_user_account", 18, argv, NULL, NULL);
391 if (sponsor->type != M_ANY || status != MR_USER)
395 argv[U_SPONSOR_TYPE] = "LIST";
396 status = wrap_mr_query("add_user_account", 18, argv, NULL, NULL);
400 argv[U_SPONSOR_TYPE] = "KERBEROS";
401 status = mrcl_validate_kerberos_member(argv[U_SPONSOR_NAME],
402 &argv[U_SPONSOR_NAME]);
403 if (mrcl_get_message())
404 mrcl_com_err(whoami);
405 if (status == MRCL_REJECT)
407 status = wrap_mr_query("add_user_account", 18, argv, NULL, NULL);
411 argv[U_SPONSOR_TYPE] = "NONE";
412 status = wrap_mr_query("add_user_account", 18, argv, NULL, NULL);
418 argv[U_SPONSOR_TYPE] = "NONE";
419 argv[U_SPONSOR_NAME] = "NONE";
421 status = wrap_mr_query("add_user_account", 18, argv, NULL, NULL);
426 com_err(whoami, status, "while adding user account.");
430 else if (update_flag)
438 status = wrap_mr_query("get_user_account_by_login", 1, args,
439 save_query_info, old_argv);
442 com_err(whoami, status, "while getting user information.");
446 argv[1] = old_argv[0];
447 argv[2] = old_argv[1];
448 argv[3] = old_argv[2];
449 argv[4] = old_argv[3];
450 argv[5] = old_argv[4];
451 argv[6] = old_argv[5];
452 argv[7] = old_argv[6];
453 argv[8] = old_argv[7];
454 argv[9] = old_argv[8];
455 argv[10] = old_argv[9];
456 argv[11] = old_argv[10];
457 argv[12] = old_argv[11];
458 argv[13] = old_argv[12];
459 argv[14] = old_argv[13];
460 argv[15] = old_argv[14];
461 argv[16] = old_argv[15];
462 argv[17] = old_argv[16];
463 argv[18] = old_argv[17];
491 argv[14] = winhomedir;
493 argv[15] = winprofiledir;
495 argv[18] = expiration;
498 argv[17] = sponsor->name;
499 switch (sponsor->type)
504 status = wrap_mr_query("update_user_account", 19, argv, NULL,
506 if (sponsor->type != M_ANY || status != MR_USER)
511 status = wrap_mr_query("update_user_account", 19, argv, NULL,
516 argv[16] = "KERBEROS";
517 status = mrcl_validate_kerberos_member(argv[17], &argv[17]);
518 if (mrcl_get_message())
519 mrcl_com_err(whoami);
520 if (status == MRCL_REJECT)
522 status = wrap_mr_query("update_user_account", 19, argv, NULL,
528 status = wrap_mr_query("update_user_account", 19, argv, NULL,
534 status = wrap_mr_query("update_user_account", 19, argv, NULL, NULL);
537 com_err(whoami, status, "while updating user.");
542 /* Deactivate a user, and the matching list and filesystem if they exist */
552 status = wrap_mr_query("update_user_status", 2, args, NULL, NULL);
555 com_err(whoami, status, "while deactivating user.");
559 status = wrap_mr_query("get_list_info", 1, args, save_query_info, argv);
560 if (status == MR_SUCCESS)
562 for (i = 15; i > 0; i--)
563 argv[i + 1] = argv[i];
565 argv[L_ACTIVE + 1] = "0";
567 status = wrap_mr_query("update_list", 16, argv, NULL, NULL);
570 com_err(whoami, status, "while updating list, "
571 "not deactivating list or filesystem.");
575 else if (status && status != MR_NO_MATCH)
577 com_err(whoami, status, "while retrieving list information.");
581 status = wrap_mr_query("get_filesys_by_label", 1, args, save_query_info,
583 if (status == MR_SUCCESS)
585 for (i = 11; i > 0; i--)
586 argv[i + 1] = argv[i];
588 argv[FS_TYPE + 1] = "ERR";
589 argv[FS_COMMENTS + 1] = "Locker disabled; call 3-1325 for help";
591 status = wrap_mr_query("update_filesys", 12, argv, NULL, NULL);
594 com_err(whoami, status, "while updating filesystem, "
595 "not deactivating filesystem.");
599 else if (status && status != MR_NO_MATCH)
601 com_err(whoami, status, "while retrieving filesystem information.");
606 /* Display user info */
613 status = wrap_mr_query("get_user_account_by_login", 1, args,
614 save_query_info, argv);
617 com_err(whoami, status, "while getting user information.");
620 if (unformatted_flag)
621 show_user_info_unformatted(argv);
623 show_user_info(argv);
626 /* register a user */
633 status = wrap_mr_query("get_user_account_by_login", 1, args,
634 save_query_info, argv);
637 com_err(whoami, status, "while looking up uid.");
641 args[0] = argv[U_UID];
645 status = wrap_mr_query("register_user", 3, args, NULL, NULL);
648 com_err(whoami, status, "while registering user.");
653 /* list user reservations */
659 status = wrap_mr_query("get_user_reservations", 1, args,
660 show_reservations, NULL);
662 if (status != MR_NO_MATCH) {
663 com_err(whoami, status, "while getting user reservations.");
668 /* add user reservations */
669 if (reservation_add_queue)
671 struct string_list *q = reservation_add_queue;
675 char *reservation = q->string;
679 args[1] = reservation;
680 status = wrap_mr_query("add_user_reservation", 2, args, NULL, NULL);
683 com_err(whoami, status, "while adding user reservation.");
691 /* delete user reservations */
692 if (reservation_remove_queue)
694 struct string_list *q = reservation_remove_queue;
698 char *reservation = q->string;
702 args[1] = reservation;
703 status = wrap_mr_query("delete_user_reservation", 2, args, NULL,
707 com_err(whoami, status, "while deleting user reservation.");
719 int save_query_info(int argc, char **argv, void *hint)
724 for(i = 0; i < argc; i++)
725 nargv[i] = strdup(argv[i]);
730 int show_reservations(int argc, char **argv, void *hint)
732 printf("Reservation: %s\n", argv[0]);
737 struct string_list *add_to_string_list(struct string_list *old_list, char *s) {
738 struct string_list *new_list;
740 new_list = (struct string_list *)malloc(sizeof(struct string_list *));
741 new_list->next = old_list;
742 new_list->string = s;
747 int wrap_mr_query(char *handle, int argc, char **argv,
748 int (*callback)(int, char **, void *), void *callarg) {
750 print_query(handle, argc, argv);
752 return mr_query(handle, argc, argv, callback, callarg);
755 void print_query(char *query_name, int argc, char **argv) {
758 printf("qy %s", query_name);
759 for(cnt = 0; cnt < argc; cnt++)
760 printf(" <%s>", argv[cnt]);
764 void show_user_info(char **argv)
769 sprintf(tbuf, "%s, %s %s", argv[U_LAST], argv[U_FIRST], argv[U_MIDDLE]);
770 printf("Login name: %-20s Full name: %s\n", argv[U_NAME], tbuf);
771 printf("User id: %-23s Login shell: %-10s\n", argv[U_UID], argv[U_SHELL]);
772 printf("Class: %-25s Windows Console Shell: %-10s\n", argv[U_CLASS],
773 argv[U_WINCONSOLESHELL]);
774 sprintf(tbuf, "%s %s", argv[U_SPONSOR_TYPE],
775 strcmp(argv[U_SPONSOR_TYPE], "NONE") ? argv[U_SPONSOR_NAME] : "");
776 printf("Sponsor: %-23s Expiration date: %s\n", tbuf, argv[U_EXPIRATION]);
777 printf("Account is: %-20s MIT ID number: %s\n",
778 UserState(atoi(argv[U_STATE])), argv[U_MITID]);
779 printf("Windows Home Directory: %s\n", argv[U_WINHOMEDIR]);
780 printf("Windows Profile Directory: %s\n", argv[U_WINPROFILEDIR]);
781 status = atoi(argv[U_STATE]);
782 if (status == 0 || status == 2)
784 printf("User %s secure Account Coupon to register\n",
785 atoi(argv[U_SECURE]) ? "needs" : "does not need");
787 printf("Comments: %s\n", argv[U_COMMENT]);
788 printf("Created by %s on %s.\n", argv[U_CREATOR], argv[U_CREATED]);
789 printf("Last mod by %s at %s with %s.\n", argv[U_MODBY], argv[U_MODTIME],
793 void show_user_info_unformatted(char **argv)
798 printf("Login name: %s\n", argv[U_NAME]);
799 printf("Full name: %s, %s %s\n", argv[U_LAST], argv[U_FIRST],
801 printf("User id: %s\n", argv[U_UID]);
802 printf("Class: %s\n", argv[U_CLASS]);
803 sprintf(tbuf, "%s %s", argv[U_SPONSOR_TYPE],
804 strcmp(argv[U_SPONSOR_TYPE], "NONE") ? argv[U_SPONSOR_NAME] : "");
805 printf("Sponsor: %s\n", tbuf);
806 printf("Expiration date: %s\n", argv[U_EXPIRATION]);
807 printf("Login shell: %s\n", argv[U_SHELL]);
808 printf("Windows Console Shell: %s\n", argv[U_WINCONSOLESHELL]);
809 printf("Account is: %s\n", UserState(atoi(argv[U_STATE])));
810 printf("MIT ID number: %s\n", argv[U_MITID]);
811 printf("Windows Home Directory: %s\n", argv[U_WINHOMEDIR]);
812 printf("Windows Profile Directory: %s\n", argv[U_WINPROFILEDIR]);
813 status = atoi(argv[U_STATE]);
814 if (status == 0 || status == 2)
815 printf("Secure: %s secure Account Coupon to register\n",
816 atoi(argv[U_SECURE]) ? "Needs" : "Does not need");
817 printf("Comments: %s\n", argv[U_COMMENT]);
818 printf("Created by: %s\n", argv[U_CREATOR]);
819 printf("Created on: %s\n", argv[U_CREATED]);
820 printf("Last mod by: %s\n", argv[U_MODBY]);
821 printf("Last mod on: %s\n", argv[U_MODTIME]);
822 printf("Last mod with: %s\n", argv[U_MODWITH]);
825 void usage(char **argv)
827 #define USAGE_OPTIONS_FORMAT " %-39s%s\n"
828 fprintf(stderr, "Usage: %s username [options]\n", argv[0]);
829 fprintf(stderr, "Options are\n");
830 fprintf(stderr, " %-39s\n", "-i | -info");
831 fprintf(stderr, USAGE_OPTIONS_FORMAT, "-C | -create",
833 fprintf(stderr, USAGE_OPTIONS_FORMAT, "-r | -register",
834 "-R | -rename newname");
835 fprintf(stderr, USAGE_OPTIONS_FORMAT, "-U | -uid uid",
836 "-s | -shell shell");
837 fprintf(stderr, USAGE_OPTIONS_FORMAT, "-S | -status status",
838 "-w | -winshell winshell");
839 fprintf(stderr, USAGE_OPTIONS_FORMAT, "-F | -first firstname",
840 "-L | -last lastname");
841 fprintf(stderr, USAGE_OPTIONS_FORMAT, "-M | -middle middlename",
842 "-I | -mitid mitid");
843 fprintf(stderr, USAGE_OPTIONS_FORMAT, "-cl | -class class",
844 "-c | -comment comment");
845 fprintf(stderr, USAGE_OPTIONS_FORMAT, "-6 | -secure 0|1",
846 "-lr | -listreservation");
847 fprintf(stderr, USAGE_OPTIONS_FORMAT, "-ar | -addreservation reservation",
848 "-dr | -deletereservation reservation");
849 fprintf(stderr, USAGE_OPTIONS_FORMAT, "-wh | -winhomedir winhomedir",
850 "-wp | -winprofiledir winprofiledir");
851 fprintf(stderr, USAGE_OPTIONS_FORMAT, "-sp | -sponsor sponsor",
852 "-e | -expiration expiration date");
853 fprintf(stderr, USAGE_OPTIONS_FORMAT, "-u | -unformatted",
855 fprintf(stderr, USAGE_OPTIONS_FORMAT, "-v | -verbose",
856 "-db | -database host[:port]");
861 /* Parse a line of input, fetching a member. NULL is returned if a member
862 * is not found. ';' is a comment character.
864 struct owner_type *parse_member(char *s)
866 struct owner_type *m;
869 while (*s && isspace(*s))
872 while (*p && *p != '\n' && *p != ';')
874 if (isprint(*p) && !isspace(*p))
881 if (p == s || strlen(s) == 0)
884 if (!(m = malloc(sizeof(struct owner_type))))
887 if ((p = strchr(s, ':')))
891 if (!strcasecmp("user", s))
893 else if (!strcasecmp("list", s))
895 else if (!strcasecmp("kerberos", s))
896 m->type = M_KERBEROS;
897 else if (!strcasecmp("none", s))
905 m->name = strdup(m->name);
910 m->type = strcasecmp(s, "none") ? M_ANY : M_NONE;