src/achown.c

   1 /* Chown-advanced command -- for the Midnight Commander
   2    Copyright (C) 1994, 1995 Radek Doulik
   3
   4    This program is free software; you can redistribute it and/or modify
   5    it under the terms of the GNU General Public License as published by
   6    the Free Software Foundation; either version 2 of the License, or
   7    (at your option) any later version.
   8
   9    This program is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12    GNU General Public License for more details.
  13
  14    You should have received a copy of the GNU General Public License
  15    along with this program; if not, write to the Free Software
  16    Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  17  */
  18
  19 #include <config.h>
  20 /* Needed for the extern declarations of integer parameters */
  21 #include <sys/types.h>
  22 #include <sys/stat.h>
  23 #ifdef HAVE_UNISTD_H
  24 #   include <unistd.h>
  25 #endif
  26 #include <string.h>
  27 #include <stdio.h>
  28 #include <errno.h>      /* For errno on SunOS systems         */
  29
  30 #include "global.h"
  31 #include "tty.h"
  32 #include "win.h"
  33 #include "color.h"
  34 #include "dlg.h"
  35 #include "widget.h"
  36 #include "dialog.h"             /* For do_refresh() */
  37 #include "wtools.h"             /* For init_box_colors() */
  38 #include "key.h"                /* XCTRL and ALT macros */
  39
  40 #include "dir.h"
  41 #include "panel.h"              /* Needed for the externs */
  42 #include "chmod.h"
  43 #include "main.h"
  44 #include "achown.h"
  45
  46 #define BX              5
  47 #define BY              6
  48
  49 #define TX              50
  50 #define TY              2
  51
  52 #define BUTTONS         9
  53
  54 #define B_SETALL        B_USER
  55 #define B_SKIP          B_USER + 1
  56
  57 #define B_OWN           B_USER + 3
  58 #define B_GRP           B_USER + 4
  59 #define B_OTH           B_USER + 5
  60 #define B_OUSER         B_USER + 6
  61 #define B_OGROUP        B_USER + 7
  62
  63 static struct Dlg_head *ch_dlg;
  64
  65 static struct {
  66     int ret_cmd, flags, y, x;
  67     char *text;
  68 } chown_advanced_but [BUTTONS] = {
  69     { B_CANCEL, NORMAL_BUTTON, 4, 53, N_("&Cancel") },
  70     { B_ENTER,  DEFPUSH_BUTTON,4, 40, N_("&Set") },
  71     { B_SKIP,   NORMAL_BUTTON, 4, 23, N_("S&kip") },
  72     { B_SETALL, NORMAL_BUTTON, 4, 0, N_("Set &all")},
  73     { B_ENTER,  NARROW_BUTTON, 0, 47, "               "},
  74     { B_ENTER,  NARROW_BUTTON, 0, 29, "               "},
  75     { B_ENTER,  NARROW_BUTTON, 0, 19, "   "},
  76     { B_ENTER,  NARROW_BUTTON, 0, 11, "   "},
  77     { B_ENTER,  NARROW_BUTTON, 0, 3, "   "},
  78 };
  79
  80 static WButton *b_att[3];       /* permission */
  81 static WButton *b_user, *b_group;       /* owner */
  82
  83 static int files_on_begin;      /* Number of files at startup */
  84 static int flag_pos;
  85 static int x_toggle;
  86 static char ch_flags[11];
  87 static const char ch_perm[] = "rwx";
  88 static umode_t ch_cmode;
  89 static struct stat *sf_stat;
  90 static int need_update;
  91 static int end_chown;
  92 static int current_file;
  93 static int single_set;
  94 static char *fname;
  95
  96 static void get_ownership (void)
  97 {                               /* set buttons  - ownership */
  98     char *name_t;
  99
 100     name_t = name_trunc (get_owner (sf_stat->st_uid), 15);
 101     memset (b_user->text, ' ', 15);
 102     strncpy (b_user->text, name_t, strlen (name_t));
 103     name_t = name_trunc (get_group (sf_stat->st_gid), 15);
 104     memset (b_group->text, ' ', 15);
 105     strncpy (b_group->text, name_t, strlen (name_t));
 106 }
 107
 108
 109 static int inc_flag_pos (int f_pos)
 110 {
 111     if (flag_pos == 10) {
 112         flag_pos = 0;
 113         return 0;
 114     }
 115     flag_pos++;
 116     if (!(flag_pos % 3) || f_pos > 2)
 117         return 0;
 118     return 1;
 119 }
 120
 121 static int dec_flag_pos (int f_pos)
 122 {
 123     if (!flag_pos) {
 124         flag_pos = 10;
 125         return 0;
 126     }
 127     flag_pos--;
 128     if (!((flag_pos + 1) % 3) || f_pos > 2)
 129         return 0;
 130     return 1;
 131 }
 132
 133 static void set_perm_by_flags (char *s, int f_p)
 134 {
 135     int i;
 136
 137     for (i = 0; i < 3; i++) {
 138         if (ch_flags[f_p + i] == '+')
 139             s[i] = ch_perm[i];
 140         else if (ch_flags[f_p + i] == '-')
 141             s[i] = '-';
 142         else
 143             s[i] = (ch_cmode & (1 << (8 - f_p - i))) ? ch_perm[i] : '-';
 144     }
 145 }
 146
 147 static umode_t get_perm (char *s, int base)
 148 {
 149     umode_t m;
 150
 151     m = 0;
 152     m |= (s [0] == '-') ? 0 :
 153         ((s[0] == '+') ? (1 << (base + 2)) : (1 << (base + 2)) & ch_cmode);
 154
 155     m |= (s [1] == '-') ? 0 :
 156         ((s[1] == '+') ? (1 << (base + 1)) : (1 << (base + 1)) & ch_cmode);
 157
 158     m |= (s [2] == '-') ? 0 :
 159         ((s[2] == '+') ? (1 << base) : (1 << base) & ch_cmode);
 160
 161     return m;
 162 }
 163
 164 static umode_t get_mode (void)
 165 {
 166     umode_t m;
 167
 168     m = ch_cmode ^ (ch_cmode & 0777);
 169     m |= get_perm (ch_flags, 6);
 170     m |= get_perm (ch_flags + 3, 3);
 171     m |= get_perm (ch_flags + 6, 0);
 172
 173     return m;
 174 }
 175
 176 static void print_flags (void)
 177 {
 178     int i;
 179
 180     attrset (COLOR_NORMAL);
 181
 182     for (i = 0; i < 3; i++){
 183         dlg_move (ch_dlg, BY+1, 9+i);
 184         addch (ch_flags [i]);
 185     }
 186
 187     for (i = 0; i < 3; i++){
 188         dlg_move (ch_dlg, BY + 1, 17 + i);
 189         addch (ch_flags [i+3]);
 190     }
 191
 192     for (i = 0; i < 3; i++){
 193         dlg_move (ch_dlg, BY + 1, 25 + i);
 194         addch (ch_flags [i+6]);
 195     }
 196
 197     set_perm_by_flags (b_att[0]->text, 0);
 198     set_perm_by_flags (b_att[1]->text, 3);
 199     set_perm_by_flags (b_att[2]->text, 6);
 200
 201     for (i = 0; i < 15; i++){
 202         dlg_move (ch_dlg, BY+1, 35+i);
 203         addch (ch_flags[9]);
 204     }
 205     for (i = 0; i < 15; i++){
 206         dlg_move (ch_dlg, BY + 1, 53 + i);
 207         addch (ch_flags[10]);
 208     }
 209 }
 210
 211 static void update_mode (Dlg_head * h)
 212 {
 213     print_flags ();
 214     attrset (COLOR_NORMAL);
 215     dlg_move (h, BY + 2, 9);
 216     printw ("%12o", get_mode ());
 217     send_message (h->current->widget, WIDGET_FOCUS, 0);
 218 }
 219
 220 static int l_call (void *data)
 221 {
 222     return 1;
 223 }
 224
 225 static int chl_callback (Dlg_head * h, int Par, int Msg)
 226 {
 227     switch (Msg) {
 228     case DLG_DRAW:
 229         common_dialog_repaint (h);
 230         break;
 231
 232     case DLG_KEY:
 233         switch (Par) {
 234         case KEY_LEFT:
 235         case KEY_RIGHT:
 236             h->ret_value = Par;
 237             dlg_stop (h);
 238         }
 239     }
 240     return 0;
 241 }
 242
 243 static void
 244 do_enter_key (Dlg_head * h, int f_pos)
 245 {
 246     Dlg_head *chl_dlg;
 247     WListbox *chl_list;
 248     struct passwd *chl_pass;
 249     struct group *chl_grp;
 250     WLEntry *fe;
 251     int lxx, lyy, chl_end, b_pos;
 252     int is_owner;
 253     char *title;
 254
 255     do {
 256         is_owner = (f_pos == 3);
 257         title = is_owner ? _("owner") : _("group");
 258
 259         lxx = (COLS - 74) / 2 + (is_owner ? 35 : 53);
 260         lyy = (LINES - 13) / 2;
 261         chl_end = 0;
 262
 263         chl_dlg =
 264             create_dlg (lyy, lxx, 13, 17, dialog_colors, chl_callback,
 265                         "[Advanced Chown]", title, DLG_COMPACT);
 266
 267         /* get new listboxes */
 268         chl_list = listbox_new (1, 1, 15, 11, 0, l_call, NULL);
 269
 270         listbox_add_item (chl_list, 0, 0, "<Unknown>", NULL);
 271
 272         if (is_owner) {
 273             /* get and put user names in the listbox */
 274             setpwent ();
 275             while ((chl_pass = getpwent ()))
 276                 listbox_add_item (chl_list, 0, 0, chl_pass->pw_name, NULL);
 277             endpwent ();
 278             fe = listbox_search_text (chl_list,
 279                                       get_owner (sf_stat->st_uid));
 280         } else {
 281             /* get and put group names in the listbox */
 282             setgrent ();
 283             while ((chl_grp = getgrent ())) {
 284                 listbox_add_item (chl_list, 0, 0, chl_grp->gr_name, NULL);
 285             }
 286             endgrent ();
 287             fe = listbox_search_text (chl_list,
 288                                       get_group (sf_stat->st_gid));
 289         }
 290
 291         if (fe)
 292             listbox_select_entry (chl_list, fe);
 293
 294         b_pos = chl_list->pos;
 295         add_widget (chl_dlg, chl_list);
 296
 297         run_dlg (chl_dlg);
 298
 299         if (b_pos != chl_list->pos) {
 300             int ok = 0;
 301             if (is_owner) {
 302                 chl_pass = getpwnam (chl_list->current->text);
 303                 if (chl_pass) {
 304                     ok = 1;
 305                     sf_stat->st_uid = chl_pass->pw_uid;
 306                 }
 307             } else {
 308                 chl_grp = getgrnam (chl_list->current->text);
 309                 if (chl_grp) {
 310                     sf_stat->st_gid = chl_grp->gr_gid;
 311                     ok = 1;
 312                 }
 313             }
 314             if (ok) {
 315                 ch_flags[f_pos + 6] = '+';
 316                 get_ownership ();
 317             }
 318             dlg_focus (h);
 319             if (ok)
 320                 print_flags ();
 321         }
 322         if (chl_dlg->ret_value == KEY_LEFT) {
 323             if (!is_owner)
 324                 chl_end = 1;
 325             dlg_one_up (ch_dlg);
 326             f_pos--;
 327         } else if (chl_dlg->ret_value == KEY_RIGHT) {
 328             if (is_owner)
 329                 chl_end = 1;
 330             dlg_one_down (ch_dlg);
 331             f_pos++;
 332         }
 333         /* Here we used to redraw the window */
 334         destroy_dlg (chl_dlg);
 335     } while (chl_end);
 336 }
 337
 338 static void chown_refresh (void)
 339 {
 340     common_dialog_repaint (ch_dlg);
 341
 342     attrset (COLOR_NORMAL);
 343
 344     dlg_move (ch_dlg, BY - 1, 8);
 345     addstr (_("owner"));
 346     dlg_move (ch_dlg, BY - 1, 16);
 347     addstr (_("group"));
 348     dlg_move (ch_dlg, BY - 1, 24);
 349     addstr (_("other"));
 350
 351     dlg_move (ch_dlg, BY - 1, 35);
 352     addstr (_("owner"));
 353     dlg_move (ch_dlg, BY - 1, 53);
 354     addstr (_("group"));
 355
 356     dlg_move (ch_dlg, 3, 4);
 357     addstr (_("On"));
 358     dlg_move (ch_dlg, BY + 1, 4);
 359     addstr (_("Flag"));
 360     dlg_move (ch_dlg, BY + 2, 4);
 361     addstr (_("Mode"));
 362
 363     if (!single_set){
 364         dlg_move (ch_dlg, 3, 54);
 365         printw (_("%6d of %d"), files_on_begin - (cpanel->marked) + 1,
 366                    files_on_begin);
 367     }
 368
 369     print_flags ();
 370 }
 371
 372 static void chown_info_update (void)
 373 {
 374     /* display file info */
 375     attrset (COLOR_NORMAL);
 376
 377     /* name && mode */
 378     dlg_move (ch_dlg, 3, 8);
 379     printw ("%s", name_trunc (fname, 45));
 380     dlg_move (ch_dlg, BY + 2, 9);
 381     printw ("%12o", get_mode ());
 382
 383     /* permissions */
 384     set_perm_by_flags (b_att[0]->text, 0);
 385     set_perm_by_flags (b_att[1]->text, 3);
 386     set_perm_by_flags (b_att[2]->text, 6);
 387 }
 388
 389 static void b_setpos (int f_pos) {
 390         b_att[0]->hotpos=-1;
 391         b_att[1]->hotpos=-1;
 392         b_att[2]->hotpos=-1;
 393         b_att[f_pos]->hotpos = (flag_pos % 3);
 394 }
 395
 396 static int advanced_chown_callback (Dlg_head * h, int Par, int Msg)
 397 {
 398     int i = 0, f_pos = BUTTONS - h->current->dlg_id - single_set - 1;
 399
 400     switch (Msg) {
 401     case DLG_DRAW:
 402         chown_refresh ();
 403         chown_info_update ();
 404         return 1;
 405
 406     case DLG_POST_KEY:
 407         if (f_pos < 3)
 408                 b_setpos (f_pos);
 409         break;
 410
 411     case DLG_FOCUS:
 412         if (f_pos < 3) {
 413             if ((flag_pos / 3) != f_pos)
 414                 flag_pos = f_pos * 3;
 415             b_setpos (f_pos);
 416         } else if (f_pos < 5)
 417             flag_pos = f_pos + 6;
 418         break;
 419
 420     case DLG_KEY:
 421         switch (Par) {
 422
 423         case XCTRL('b'):
 424         case KEY_LEFT:
 425             if (f_pos < 5)
 426                 return (dec_flag_pos (f_pos));
 427             break;
 428
 429         case XCTRL('f'):
 430         case KEY_RIGHT:
 431             if (f_pos < 5)
 432                 return (inc_flag_pos (f_pos));
 433             break;
 434
 435         case ' ':
 436             if (f_pos < 3)
 437                 return 1;
 438             break;
 439
 440         case '\n':
 441         case KEY_ENTER:
 442             if (f_pos <= 2 || f_pos >= 5)
 443                 break;
 444             do_enter_key (h, f_pos);
 445             return 1;
 446
 447         case ALT ('x'):
 448             i++;
 449
 450         case ALT ('w'):
 451             i++;
 452
 453         case ALT ('r'):
 454             Par = i + 3;
 455             for (i = 0; i < 3; i++)
 456                 ch_flags[i * 3 + Par - 3] = (x_toggle & (1 << Par)) ? '-' : '+';
 457             x_toggle ^= (1 << Par);
 458             update_mode (h);
 459             dlg_broadcast_msg (h, WIDGET_DRAW, 0);
 460             send_message (h->current->widget, WIDGET_FOCUS, 0);
 461             break;
 462
 463         case XCTRL ('x'):
 464             i++;
 465
 466         case XCTRL ('w'):
 467             i++;
 468
 469         case XCTRL ('r'):
 470             Par = i;
 471             for (i = 0; i < 3; i++)
 472                 ch_flags[i * 3 + Par] = (x_toggle & (1 << Par)) ? '-' : '+';
 473             x_toggle ^= (1 << Par);
 474             update_mode (h);
 475             dlg_broadcast_msg (h, WIDGET_DRAW, 0);
 476             send_message (h->current->widget, WIDGET_FOCUS, 0);
 477             break;
 478
 479         case 'x':
 480             i++;
 481
 482         case 'w':
 483             i++;
 484
 485         case 'r':
 486             if (f_pos > 2)
 487                 break;
 488             flag_pos = f_pos * 3 + i;   /* (strchr(ch_perm,Par)-ch_perm); */
 489             if (((WButton *) h->current->widget)->text[(flag_pos % 3)] == '-')
 490                 ch_flags[flag_pos] = '+';
 491             else
 492                 ch_flags[flag_pos] = '-';
 493             update_mode (h);
 494             break;
 495
 496         case '4':
 497             i++;
 498
 499         case '2':
 500             i++;
 501
 502         case '1':
 503             if (f_pos > 2)
 504                 break;
 505             flag_pos = i + f_pos * 3;
 506             ch_flags[flag_pos] = '=';
 507             update_mode (h);
 508             break;
 509
 510         case '-':
 511             if (f_pos > 2)
 512                 break;
 513
 514         case '*':
 515             if (Par == '*')
 516                 Par = '=';
 517
 518         case '=':
 519         case '+':
 520             if (f_pos > 4)
 521                 break;
 522             ch_flags[flag_pos] = Par;
 523             update_mode (h);
 524             advanced_chown_callback (h, KEY_RIGHT, DLG_KEY);
 525             if (flag_pos>8 || !(flag_pos%3)) dlg_one_down (h);
 526
 527             break;
 528         }
 529         return 0;
 530     }
 531     return 0;
 532 }
 533
 534 static void
 535 init_chown_advanced (void)
 536 {
 537     int i;
 538
 539     sf_stat = g_new (struct stat, 1);
 540     do_refresh ();
 541     end_chown = need_update = current_file = 0;
 542     single_set = (cpanel->marked < 2) ? 2 : 0;
 543     memset (ch_flags, '=', 11);
 544     flag_pos = 0;
 545     x_toggle = 070;
 546
 547     ch_dlg =
 548         create_dlg (0, 0, 13, 74, dialog_colors, advanced_chown_callback,
 549                     "[Advanced Chown]", _(" Chown advanced command "),
 550                     DLG_CENTER);
 551
 552 #define XTRACT(i) BY+chown_advanced_but[i].y, BX+chown_advanced_but[i].x, \
 553         chown_advanced_but[i].ret_cmd, chown_advanced_but[i].flags, _(chown_advanced_but[i].text), \
 554         0, 0, NULL
 555
 556     for (i = 0; i < BUTTONS - 5; i++)
 557         if (!single_set || i < 2)
 558             add_widget (ch_dlg, button_new (XTRACT (i)));
 559
 560     b_att[0] = button_new (XTRACT (8));
 561     b_att[1] = button_new (XTRACT (7));
 562     b_att[2] = button_new (XTRACT (6));
 563     b_user = button_new (XTRACT (5));
 564     b_group = button_new (XTRACT (4));
 565
 566     add_widget (ch_dlg, b_group);
 567     add_widget (ch_dlg, b_user);
 568     add_widget (ch_dlg, b_att[2]);
 569     add_widget (ch_dlg, b_att[1]);
 570     add_widget (ch_dlg, b_att[0]);
 571 }
 572
 573 static void
 574 chown_advanced_done (void)
 575 {
 576     g_free (sf_stat);
 577     if (need_update)
 578         update_panels (UP_OPTIMIZE, UP_KEEPSEL);
 579     repaint_screen ();
 580 }
 581
 582 #if 0
 583 static inline void do_chown (uid_t u, gid_t g)
 584 {
 585     chown (cpanel->dir.list[current_file].fname, u, g);
 586     file_mark (cpanel, current_file, 0);
 587 }
 588 #endif
 589
 590 static char *next_file (void)
 591 {
 592     while (!cpanel->dir.list[current_file].f.marked)
 593         current_file++;
 594
 595     return cpanel->dir.list[current_file].fname;
 596 }
 597
 598 static void apply_advanced_chowns (struct stat *sf)
 599 {
 600     char *fname;
 601     gid_t a_gid = sf->st_gid;
 602     uid_t a_uid = sf->st_uid;
 603
 604     fname = cpanel->dir.list[current_file].fname;
 605     need_update = end_chown = 1;
 606     if (mc_chmod (fname, get_mode ()) == -1)
 607         message (1, MSG_ERROR, _(" Cannot chmod \"%s\" \n %s "),
 608                  fname, unix_error_string (errno));
 609     /* call mc_chown only, if mc_chmod didn't fail */
 610     else if (mc_chown (fname, (ch_flags[9] == '+') ? sf->st_uid : -1,
 611                        (ch_flags[10] == '+') ? sf->st_gid : -1) == -1)
 612         message (1, MSG_ERROR, _(" Cannot chown \"%s\" \n %s "),
 613                  fname, unix_error_string (errno));
 614     do_file_mark (cpanel, current_file, 0);
 615
 616     do {
 617         fname = next_file ();
 618
 619         if (!stat_file (fname, sf))
 620             break;
 621         ch_cmode = sf->st_mode;
 622         if (mc_chmod (fname, get_mode ()) == -1)
 623             message (1, MSG_ERROR, _(" Cannot chmod \"%s\" \n %s "),
 624                      fname, unix_error_string (errno));
 625         /* call mc_chown only, if mc_chmod didn't fail */
 626         else if (mc_chown (fname, (ch_flags[9] == '+') ? a_uid : -1, (ch_flags[10] == '+') ? a_gid : -1) == -1)
 627             message (1, MSG_ERROR, _(" Cannot chown \"%s\" \n %s "),
 628                      fname, unix_error_string (errno));
 629
 630         do_file_mark (cpanel, current_file, 0);
 631     } while (cpanel->marked);
 632 }
 633
 634 void
 635 chown_advanced_cmd (void)
 636 {
 637
 638     files_on_begin = cpanel->marked;
 639
 640     do {                        /* do while any files remaining */
 641         init_chown_advanced ();
 642
 643         if (cpanel->marked)
 644             fname = next_file ();       /* next marked file */
 645         else
 646             fname = selection (cpanel)->fname;  /* single file */
 647
 648         if (!stat_file (fname, sf_stat)){       /* get status of file */
 649             destroy_dlg (ch_dlg);
 650             break;
 651         }
 652         ch_cmode = sf_stat->st_mode;
 653
 654         chown_refresh ();
 655
 656         get_ownership ();
 657
 658         /* game can begin */
 659         run_dlg (ch_dlg);
 660
 661         switch (ch_dlg->ret_value) {
 662         case B_CANCEL:
 663             end_chown = 1;
 664             break;
 665
 666         case B_ENTER:
 667             need_update = 1;
 668             if (mc_chmod (fname, get_mode ()) == -1)
 669                 message (1, MSG_ERROR, _(" Cannot chmod \"%s\" \n %s "),
 670                          fname, unix_error_string (errno));
 671             /* call mc_chown only, if mc_chmod didn't fail */
 672             else if (mc_chown (fname, (ch_flags[9] == '+') ? sf_stat->st_uid : -1, (ch_flags[10] == '+') ? sf_stat->st_gid : -1) == -1)
 673                 message (1, MSG_ERROR, _(" Cannot chown \"%s\" \n %s "),
 674                          fname, unix_error_string (errno));
 675             break;
 676         case B_SETALL:
 677             apply_advanced_chowns (sf_stat);
 678             break;
 679
 680         case B_SKIP:
 681             break;
 682
 683         }
 684
 685         if (cpanel->marked && ch_dlg->ret_value != B_CANCEL) {
 686             do_file_mark (cpanel, current_file, 0);
 687             need_update = 1;
 688         }
 689         destroy_dlg (ch_dlg);
 690     } while (cpanel->marked && !end_chown);
 691
 692     chown_advanced_done ();
 693 }