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