src/WINNT/afssvrmgr/set_restore.cpp

   1 /*
   2  * Copyright 2000, International Business Machines Corporation and others.
   3  * All Rights Reserved.
   4  *
   5  * This software has been released under the terms of the IBM Public
   6  * License.  For details, see the LICENSE file in the top-level source
   7  * directory or online at http://www.openafs.org/dl/license10.html
   8  */
   9
  10 #include <winsock2.h>
  11 #include <ws2tcpip.h>
  12
  13 extern "C" {
  14 #include <afs/param.h>
  15 #include <afs/stds.h>
  16 }
  17
  18 #include "svrmgr.h"
  19 #include "set_restore.h"
  20 #include "set_general.h"
  21 #include "propcache.h"
  22 #include "columns.h"
  23 #include "svr_window.h"
  24 #include "display.h"
  25
  26
  27 /*
  28  * PROTOTYPES _________________________________________________________________
  29  *
  30  */
  31
  32 BOOL CALLBACK Filesets_Restore_DlgProc (HWND hDlg, UINT msg, WPARAM wp, LPARAM lp);
  33 void Filesets_Restore_OnInitDialog (HWND hDlg, LPSET_RESTORE_PARAMS psrp);
  34 void Filesets_Restore_OnEndTask_LookupFileset (HWND hDlg, LPTASKPACKET ptp, LPSET_RESTORE_PARAMS psrp);
  35 void Filesets_Restore_OnEndTask_EnumServers (HWND hDlg, LPSET_RESTORE_PARAMS psrp);
  36 void Filesets_Restore_OnEndTask_EnumAggregates (HWND hDlg, LPSET_RESTORE_PARAMS psrp);
  37 void Filesets_Restore_OnSelectServer (HWND hDlg, LPSET_RESTORE_PARAMS psrp);
  38 void Filesets_Restore_OnSetName (HWND hDlg, LPSET_RESTORE_PARAMS psrp);
  39 void Filesets_Restore_OnFileName (HWND hDlg, LPSET_RESTORE_PARAMS psrp);
  40 void Filesets_Restore_OnAggregate (HWND hDlg, LPSET_RESTORE_PARAMS psrp);
  41 void Filesets_Restore_OnBrowse (HWND hDlg, LPSET_RESTORE_PARAMS psrp);
  42 void Filesets_Restore_EnableOK (HWND hDlg, LPSET_RESTORE_PARAMS psrp);
  43
  44
  45 /*
  46  * ROUTINES ___________________________________________________________________
  47  *
  48  */
  49
  50 void Filesets_Restore (LPIDENT lpiParent)
  51 {
  52    LPSET_RESTORE_PARAMS psrp = New (SET_RESTORE_PARAMS);
  53    psrp->lpi = lpiParent;
  54    psrp->szFileset[0] = TEXT('\0');
  55    psrp->szFilename[0] = TEXT('\0');
  56    psrp->fIncremental = FALSE;
  57
  58    if (lpiParent && lpiParent->fIsFileset())
  59       lpiParent->GetFilesetName (psrp->szFileset);
  60
  61    INT_PTR rc = ModalDialogParam (IDD_SET_RESTORE, GetActiveWindow(), (DLGPROC)Filesets_Restore_DlgProc, (LPARAM)psrp);
  62    if (rc != IDOK)
  63       {
  64       Delete (psrp);
  65       }
  66    else if (!ASSERT( psrp->lpi != NULL ))
  67       {
  68       Delete (psrp);
  69       }
  70    else if (!ASSERT( psrp->szFileset[0] != TEXT('\0') ))
  71       {
  72       Delete (psrp);
  73       }
  74    else if (!ASSERT( psrp->szFilename[0] != TEXT('\0') ))
  75       {
  76       Delete (psrp);
  77       }
  78    else
  79       {
  80       StartTask (taskSET_RESTORE, NULL, psrp);
  81       }
  82 }
  83
  84
  85 BOOL CALLBACK Filesets_Restore_DlgProc (HWND hDlg, UINT msg, WPARAM wp, LPARAM lp)
  86 {
  87    static LPSET_RESTORE_PARAMS psrp = NULL;
  88    if (msg == WM_INITDIALOG)
  89       psrp = (LPSET_RESTORE_PARAMS)lp;
  90
  91    if (HandleColumnNotify (hDlg, msg, wp, lp, &gr.viewAggRestore))
  92       return FALSE;
  93
  94    if (AfsAppLib_HandleHelp (IDD_SET_RESTORE, hDlg, msg, wp, lp))
  95       return TRUE;
  96
  97    if (psrp != NULL)
  98       {
  99       switch (msg)
 100          {
 101          case WM_INITDIALOG:
 102             FastList_SetTextCallback (GetDlgItem (hDlg, IDC_AGG_LIST), GetItemText, &gr.viewAggRestore);
 103             Filesets_Restore_OnInitDialog (hDlg, psrp);
 104             break;
 105
 106          case WM_CONTEXTMENU:
 107             HWND hList;
 108             hList = GetDlgItem (hDlg, IDC_AGG_LIST);
 109
 110             POINT ptScreen;
 111             POINT ptClient;
 112             ptScreen.x = LOWORD(lp);
 113             ptScreen.y = HIWORD(lp);
 114
 115             ptClient = ptScreen;
 116             ScreenToClient (hList, &ptClient);
 117
 118             if (FL_HitTestForHeaderBar (hList, ptClient))
 119                {
 120                HMENU hm = TaLocale_LoadMenu (MENU_COLUMNS);
 121                DisplayContextMenu (hm, ptScreen, hList);
 122                }
 123             break;
 124
 125          case WM_ENDTASK:
 126             LPTASKPACKET ptp;
 127             if ((ptp = (LPTASKPACKET)lp) != NULL)
 128                {
 129                if (ptp->idTask == taskSET_LOOKUP)
 130                   Filesets_Restore_OnEndTask_LookupFileset (hDlg, ptp, psrp);
 131                else if (ptp->idTask == taskSVR_ENUM_TO_COMBOBOX)
 132                   Filesets_Restore_OnEndTask_EnumServers (hDlg, psrp);
 133                else if (ptp->idTask == taskAGG_ENUM_TO_LISTVIEW)
 134                   Filesets_Restore_OnEndTask_EnumAggregates (hDlg, psrp);
 135                FreeTaskPacket (ptp);
 136                }
 137             break;
 138
 139          case WM_COLUMNS_CHANGED:
 140             LPARAM lpSel;
 141             lpSel = FL_GetSelectedData (GetDlgItem (hDlg, IDC_AGG_LIST));
 142             FL_RestoreView (GetDlgItem (hDlg, IDC_AGG_LIST), &gr.viewAggRestore);
 143             Filesets_Restore_OnSelectServer (hDlg, psrp);
 144             break;
 145
 146          case WM_COMMAND:
 147             switch (LOWORD(wp))
 148                {
 149                case IDOK:
 150                case IDCANCEL:
 151                   EndDialog (hDlg, LOWORD(wp));
 152                   break;
 153
 154                case IDC_RESTORE_INCREMENTAL:
 155                   psrp->fIncremental = IsDlgButtonChecked (hDlg, IDC_RESTORE_INCREMENTAL);
 156                   break;
 157
 158                case IDC_RESTORE_SERVER:
 159                   if (HIWORD(wp) == CBN_SELCHANGE)
 160                      Filesets_Restore_OnSelectServer (hDlg, psrp);
 161                   break;
 162
 163                case IDC_RESTORE_SETNAME:
 164                   if (HIWORD(wp) == EN_UPDATE)
 165                      Filesets_Restore_OnSetName (hDlg, psrp);
 166                   break;
 167
 168                case IDC_RESTORE_FILENAME:
 169                   Filesets_Restore_OnFileName (hDlg, psrp);
 170                   break;
 171
 172                case IDC_RESTORE_BROWSE:
 173                   Filesets_Restore_OnBrowse (hDlg, psrp);
 174                   break;
 175                }
 176             break;
 177
 178          case WM_DESTROY:
 179             FL_StoreView (GetDlgItem (hDlg, IDC_AGG_LIST), &gr.viewAggRestore);
 180             psrp = NULL;
 181             break;
 182
 183          case WM_NOTIFY:
 184             switch (((LPNMHDR)lp)->code)
 185                {
 186                case FLN_ITEMSELECT:
 187                   Filesets_Restore_OnAggregate (hDlg, psrp);
 188                   break;
 189                }
 190             break;
 191          }
 192       }
 193
 194    return FALSE;
 195 }
 196
 197
 198 static LONG_PTR procFilesetsRestoreList = 0;
 199
 200 LRESULT CALLBACK Filesets_Restore_SubclassListProc (HWND hList, UINT msg, WPARAM wp, LPARAM lp)
 201 {
 202    LRESULT rc;
 203
 204    if (procFilesetsRestoreList == 0)
 205       rc = DefWindowProc (hList, msg, wp, lp);
 206    else
 207       rc = (LRESULT) CallWindowProc ((WNDPROC)procFilesetsRestoreList, hList, msg, wp, lp);
 208
 209    switch (msg)
 210       {
 211       case WM_DESTROY:
 212          if (procFilesetsRestoreList != 0)
 213             SetWindowLongPtr (hList, GWLP_WNDPROC, procFilesetsRestoreList);
 214          break;
 215
 216       case WM_COMMAND:
 217          switch (LOWORD(wp))
 218             {
 219             case M_COLUMNS:
 220                ShowColumnsDialog (GetParent (hList), &gr.viewAggRestore);
 221                break;
 222             }
 223          break;
 224       }
 225
 226    return rc;
 227 }
 228
 229
 230 void Filesets_Restore_OnInitDialog (HWND hDlg, LPSET_RESTORE_PARAMS psrp)
 231 {
 232    HWND hList = GetDlgItem (hDlg, IDC_AGG_LIST);
 233    if (procFilesetsRestoreList == 0)
 234       procFilesetsRestoreList = GetWindowLongPtr (hList, GWLP_WNDPROC);
 235    SetWindowLongPtr (hList, GWLP_WNDPROC, (LONG_PTR)Filesets_Restore_SubclassListProc);
 236
 237    if (gr.viewAggRestore.lvsView == 0)  // never initialized this?
 238       {
 239       if (gr.viewAggMove.lvsView != 0)
 240          memcpy (&gr.viewAggRestore, &gr.viewAggMove, sizeof(VIEWINFO));
 241       else if (gr.viewAggCreate.lvsView != 0)
 242          memcpy (&gr.viewAggRestore, &gr.viewAggCreate, sizeof(VIEWINFO));
 243       else
 244          memcpy (&gr.viewAggRestore, &gr.viewAgg, sizeof(VIEWINFO));
 245       }
 246
 247    FL_RestoreView (GetDlgItem (hDlg, IDC_AGG_LIST), &gr.viewAggRestore);
 248
 249    SetDlgItemText (hDlg, IDC_RESTORE_FILENAME, psrp->szFilename);
 250    SetDlgItemText (hDlg, IDC_RESTORE_SETNAME, psrp->szFileset);
 251
 252    SetDlgItemText (hDlg, IDC_RESTORE_CREATE, TEXT(""));
 253    Filesets_Restore_OnEndTask_LookupFileset (hDlg, NULL, psrp);
 254
 255    CheckDlgButton (hDlg, IDC_RESTORE_INCREMENTAL, psrp->fIncremental);
 256
 257    // Fill in the Servers combobox, and select the default server
 258    // (if one was specified).
 259    //
 260    LPSVR_ENUM_TO_COMBOBOX_PACKET lpp = New (SVR_ENUM_TO_COMBOBOX_PACKET);
 261    lpp->hCombo = GetDlgItem (hDlg, IDC_RESTORE_SERVER);
 262    lpp->lpiSelect = (psrp->lpi) ? psrp->lpi->GetServer() : NULL;
 263    StartTask (taskSVR_ENUM_TO_COMBOBOX, hDlg, lpp);
 264 }
 265
 266
 267 void Filesets_Restore_OnSelectServer (HWND hDlg, LPSET_RESTORE_PARAMS psrp)
 268 {
 269    LPIDENT lpiServerNew = (LPIDENT)CB_GetSelectedData (GetDlgItem (hDlg, IDC_RESTORE_SERVER));
 270
 271    LPIDENT lpiSelect = NULL;
 272    if (psrp->lpi && !psrp->lpi->fIsServer())
 273       {
 274       if (psrp->lpi->GetServer() == lpiServerNew)
 275          lpiSelect = psrp->lpi->GetAggregate();
 276       }
 277
 278    SetWindowLongPtr (hDlg, DWLP_USER, (LONG_PTR)lpiServerNew);
 279
 280    LPAGG_ENUM_TO_LISTVIEW_PACKET lpp = New (AGG_ENUM_TO_LISTVIEW_PACKET);
 281    lpp->lpiServer = lpiServerNew;
 282    lpp->hList = GetDlgItem (hDlg, IDC_AGG_LIST);
 283    lpp->lpiSelect = lpiSelect;
 284    lpp->lpvi = &gr.viewAggRestore;
 285    StartTask (taskAGG_ENUM_TO_LISTVIEW, hDlg, lpp);
 286 }
 287
 288
 289 void Filesets_Restore_OnEndTask_EnumServers (HWND hDlg, LPSET_RESTORE_PARAMS psrp)
 290 {
 291    Filesets_Restore_OnSelectServer (hDlg, psrp);
 292 }
 293
 294
 295 void Filesets_Restore_OnEndTask_EnumAggregates (HWND hDlg, LPSET_RESTORE_PARAMS psrp)
 296 {
 297    ;
 298 }
 299
 300
 301 void Filesets_Restore_OnEndTask_LookupFileset (HWND hDlg, LPTASKPACKET ptp, LPSET_RESTORE_PARAMS psrp)
 302 {
 303    if (ptp)
 304       {
 305       psrp->lpi = (ptp->rc) ? TASKDATA(ptp)->lpi : NULL;
 306       }
 307    if (!psrp->lpi)
 308       {
 309       psrp->lpi = (LPIDENT)FL_GetSelectedData (GetDlgItem (hDlg, IDC_AGG_LIST));
 310       }
 311
 312    BOOL fCreate = (psrp->lpi && psrp->lpi->fIsFileset()) ? FALSE : TRUE;
 313
 314    TCHAR szFileset[ cchNAME ];
 315    GetDlgItemText (hDlg, IDC_RESTORE_SETNAME, szFileset, cchNAME);
 316
 317    LPTSTR pszText;
 318    if (szFileset[0] == TEXT('\0'))
 319       {
 320       pszText = CloneString (TEXT(""));
 321       }
 322    else if (fCreate)
 323       {
 324       pszText = FormatString (IDS_RESTORE_CREATESET, TEXT("%s"), szFileset);
 325       }
 326    else
 327       {
 328       TCHAR szServer[ cchNAME ];
 329       TCHAR szAggregate[ cchNAME ];
 330       psrp->lpi->GetServerName (szServer);
 331       psrp->lpi->GetAggregateName (szAggregate);
 332       pszText = FormatString (IDS_RESTORE_OVERWRITESET, TEXT("%s%s%s"), szServer, szAggregate, szFileset);
 333       }
 334    SetDlgItemText (hDlg, IDC_RESTORE_CREATE, pszText);
 335    FreeString (pszText);
 336
 337    EnableWindow (GetDlgItem (hDlg, IDC_RESTORE_SERVER), fCreate);
 338    EnableWindow (GetDlgItem (hDlg, IDC_AGG_LIST), fCreate);
 339
 340    if (psrp->lpi)
 341       {
 342       LPIDENT lpiServer = (LPIDENT)CB_GetSelectedData (GetDlgItem (hDlg, IDC_RESTORE_SERVER));
 343
 344       if (psrp->lpi->GetServer() != lpiServer)
 345          {
 346          CB_SetSelectedByData (GetDlgItem (hDlg, IDC_RESTORE_SERVER), (LPARAM)psrp->lpi->GetServer());
 347          Filesets_Restore_OnSelectServer (hDlg, psrp);
 348          }
 349       else if (!psrp->lpi->fIsServer())
 350          {
 351          FL_SetSelectedByData (GetDlgItem (hDlg, IDC_AGG_LIST), (LPARAM)psrp->lpi->GetAggregate());
 352          }
 353       }
 354
 355    Filesets_Restore_EnableOK (hDlg, psrp);
 356 }
 357
 358
 359 void Filesets_Restore_OnFileName (HWND hDlg, LPSET_RESTORE_PARAMS psrp)
 360 {
 361    GetDlgItemText (hDlg, IDC_RESTORE_FILENAME, psrp->szFilename, MAX_PATH);
 362    Filesets_Restore_EnableOK (hDlg, psrp);
 363 }
 364
 365
 366 void Filesets_Restore_OnSetName (HWND hDlg, LPSET_RESTORE_PARAMS psrp)
 367 {
 368    GetDlgItemText (hDlg, IDC_RESTORE_SETNAME, psrp->szFileset, cchNAME);
 369    EnableWindow (GetDlgItem (hDlg, IDOK), FALSE);
 370
 371    LPSET_LOOKUP_PACKET lpp = New (SET_LOOKUP_PACKET);
 372    lstrcpy (lpp->szFileset, psrp->szFileset);
 373    StartTask (taskSET_LOOKUP, hDlg, lpp);
 374 }
 375
 376
 377 void Filesets_Restore_OnAggregate (HWND hDlg, LPSET_RESTORE_PARAMS psrp)
 378 {
 379    if (!( psrp->lpi && psrp->lpi->fIsFileset() ))
 380       {
 381       psrp->lpi = (LPIDENT)FL_GetSelectedData (GetDlgItem (hDlg, IDC_AGG_LIST));
 382       }
 383
 384    Filesets_Restore_EnableOK (hDlg, psrp);
 385 }
 386
 387
 388 void Filesets_Restore_EnableOK (HWND hDlg, LPSET_RESTORE_PARAMS psrp)
 389 {
 390    BOOL fEnable = TRUE;
 391
 392    if (psrp->szFilename[0] == TEXT('\0'))
 393       fEnable = FALSE;
 394
 395    if (psrp->szFileset[0] == TEXT('\0'))
 396       fEnable = FALSE;
 397
 398    if (!psrp->lpi || psrp->lpi->fIsServer())
 399       fEnable = FALSE;
 400
 401    EnableWindow (GetDlgItem (hDlg, IDOK), fEnable);
 402 }
 403
 404
 405 void Filesets_Restore_OnBrowse (HWND hDlg, LPSET_RESTORE_PARAMS psrp)
 406 {
 407    TCHAR szFilter[ cchRESOURCE ];
 408    GetString (szFilter, IDS_RESTORE_FILTER);
 409    TCHAR chFilter = szFilter[ lstrlen(szFilter)-1 ];
 410    for (LPTSTR pszFilter = szFilter;
 411         (*pszFilter) && ((pszFilter = (LPTSTR)lstrchr (pszFilter, chFilter)) != NULL);
 412         ++pszFilter)
 413       {
 414       *pszFilter = TEXT('\0');
 415       }
 416
 417    TCHAR szFilename[ MAX_PATH ];
 418    GetDlgItemText (hDlg, IDC_RESTORE_FILENAME, szFilename, MAX_PATH);
 419
 420    OPENFILENAME ofn;
 421    memset (&ofn, 0x00, sizeof(ofn));
 422    ofn.lStructSize = sizeof(ofn);
 423    ofn.hwndOwner = hDlg;
 424    ofn.hInstance = THIS_HINST;
 425    ofn.lpstrFilter = szFilter;
 426    ofn.nFilterIndex = 1;
 427    ofn.lpstrFile = szFilename;
 428    ofn.nMaxFile = MAX_PATH;
 429    ofn.Flags = OFN_HIDEREADONLY | OFN_NOREADONLYRETURN |
 430                OFN_OVERWRITEPROMPT | OFN_PATHMUSTEXIST;
 431    ofn.lpstrDefExt = TEXT("dmp");
 432
 433    if (GetOpenFileName (&ofn))
 434       {
 435       SetDlgItemText (hDlg, IDC_RESTORE_FILENAME, szFilename);
 436       Filesets_Restore_OnFileName (hDlg, psrp);
 437       }
 438 }
 439