src/WINNT/afsadmsvr/TaAfsAdmSvrMain.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 "TaAfsAdmSvrInternal.h"
  19
  20 /*
  21  * ROUTINES ___________________________________________________________________
  22  *
  23  */
  24
  25 #ifdef DEBUG
  26 LPARAM CALLBACK AfsAdmSvr_Debug_ThreadProc (PVOID lp)
  27 {
  28    ShowMemoryManager();
  29
  30    MSG msg;
  31    while (GetMessage (&msg, 0, 0, 0))
  32       {
  33       if (!IsMemoryManagerMessage (&msg))
  34          {
  35          TranslateMessage (&msg);
  36          DispatchMessage (&msg);
  37          }
  38       }
  39
  40    return 0;
  41 }
  42 #endif
  43
  44
  45 int cdecl main (int argc, char **argv)
  46 {
  47    BOOL fSuccess = FALSE;
  48
  49    Print (TEXT("Initializing..."));
  50
  51    WSADATA Data;
  52    WSAStartup (0x0101, &Data);
  53
  54    // Parse the command-line
  55    //
  56    UINT_PTR dwAutoScope = AFSADMSVR_SCOPE_VOLUMES | AFSADMSVR_SCOPE_USERS;
  57
  58    for (--argc,++argv; argc; --argc,++argv)
  59       {
  60       if (!lstrcmpi (*argv, AFSADMSVR_KEYWORD_TIMED))
  61          AfsAdmSvr_EnableAutoShutdown (TRUE);
  62       else if (!lstrcmpi (*argv, AFSADMSVR_KEYWORD_MANUAL))
  63          dwAutoScope = 0;
  64       else if (!lstrcmpi (*argv, AFSADMSVR_KEYWORD_SCOPE_USERS))
  65          dwAutoScope &= ~AFSADMSVR_SCOPE_VOLUMES;
  66       else if (!lstrcmpi (*argv, AFSADMSVR_KEYWORD_SCOPE_VOLUMES))
  67          dwAutoScope &= ~AFSADMSVR_SCOPE_USERS;
  68 #ifdef DEBUG
  69       else if (!lstrcmpi (*argv, AFSADMSVR_KEYWORD_DEBUG))
  70          CreateThread (NULL, 0, (LPTHREAD_START_ROUTINE)AfsAdmSvr_Debug_ThreadProc, 0, 0, 0);
  71 #endif
  72       }
  73
  74    // Prepare to listen for RPCs
  75    //
  76    unsigned char *pszPROTOCOL = (unsigned char *)"ncacn_ip_tcp";
  77    unsigned char *pszENTRYNAME = (unsigned char *)AFSADMSVR_ENTRYNAME_DEFAULT;
  78    unsigned char *pszANNOTATION = (unsigned char *)"Transarc AFS Administrative Server";
  79    unsigned char szEndpoint[ 32 ];
  80    wsprintf ((LPTSTR)szEndpoint, "%lu", AFSADMSVR_ENDPOINT_DEFAULT);
  81    int cMAX_CALLS = 50;
  82
  83    // Clean up any broken interface registration
  84    //
  85    RpcServerUnregisterIf (ITaAfsAdminSvr_v1_0_s_ifspec, 0, FALSE);
  86 #ifdef notdef
  87    RpcNsBindingUnexport (RPC_C_NS_SYNTAX_DEFAULT, pszENTRYNAME, ITaAfsAdminSvr_v1_0_s_ifspec, NULL);
  88 #endif
  89
  90    // Register our interface
  91    //
  92    RPC_STATUS status;
  93    if ((status = RpcServerUseProtseq (pszPROTOCOL, cMAX_CALLS, NULL)) != 0)
  94       {
  95       Print (dlERROR, TEXT("RpcServerUseProtseq failed; error 0x%08lX"), status);
  96       }
  97    else if ((status = RpcServerRegisterIf (ITaAfsAdminSvr_v1_0_s_ifspec, 0, 0)) != 0)
  98       {
  99       Print (dlERROR, TEXT("RpcServerRegisterIf failed; error 0x%08lX"), status);
 100       }
 101    else
 102       {
 103       // Always try to register on port 1025; that's the easiest thing for
 104       // some clients to find. We'll only fail if we (a) can't use 1025, and
 105       // (b) can't export our bindings.
 106       //
 107       BOOL fGotPort = FALSE;
 108       if (RpcServerUseProtseqEp (pszPROTOCOL, cMAX_CALLS, szEndpoint, NULL) == 0)
 109          fGotPort = TRUE;
 110       else
 111          Print (dlWARNING, TEXT("RpcServerUseProtseqEp failed (benign); error 0x%08lX"), status);
 112
 113       RPC_BINDING_VECTOR *pBindingVector;
 114       if ((status = RpcServerInqBindings (&pBindingVector)) != 0)
 115          {
 116          Print (dlERROR, TEXT("RpcServerRegisterIf failed; error 0x%08lX"), status);
 117          }
 118       else if ((status = RpcEpRegister (ITaAfsAdminSvr_v1_0_s_ifspec, pBindingVector, NULL, pszANNOTATION)) != 0)
 119          {
 120          Print (dlERROR, TEXT("RpcEpRegister failed; error 0x%08lX"), status);
 121          }
 122       else
 123          {
 124          BOOL fExportedBinding = FALSE;
 125
 126 #ifdef notdef
 127          if ((status = RpcNsBindingExport (RPC_C_NS_SYNTAX_DEFAULT, pszENTRYNAME, ITaAfsAdminSvr_v1_0_s_ifspec, pBindingVector, NULL)) == 0)
 128             fExportedBinding = TRUE;
 129          else
 130             Print (dlWARNING, TEXT("RpcNsBindingExport failed (benign); error 0x%08lX"), status);
 131 #endif
 132
 133          if (!fExportedBinding && !fGotPort)
 134             {
 135             Print (dlERROR, TEXT("RpcNsBindingExport failed; error 0x%08lX"), status);
 136             Print (dlERROR, TEXT("Could not bind to port %s or export bindings; terminating"), szEndpoint);
 137             }
 138          else
 139             {
 140             AfsAdmSvr_Startup();
 141
 142             Print (TEXT("Ready.\n"));
 143
 144             // If not asked to open cells manually, fork a thread to start opening
 145             // the default local cell
 146             //
 147             if (dwAutoScope)
 148                {
 149                DWORD dwThreadID;
 150                CreateThread (NULL, 0, (LPTHREAD_START_ROUTINE)AfsAdmSvr_AutoOpen_ThreadProc, (PVOID)dwAutoScope, 0, &dwThreadID);
 151                }
 152
 153             // Listen for requests until someone calls StopListen
 154             //
 155             if ((status = RpcServerListen (1, cMAX_CALLS, FALSE)) != 0)
 156                {
 157                Print (dlERROR, TEXT("RpcServerListen failed; error 0x%08lX"), status);
 158                }
 159             else
 160                {
 161                fSuccess = TRUE;
 162                }
 163
 164             AfsAdmSvr_Shutdown();
 165             }
 166
 167 #ifdef notdef
 168          if (fExportedBinding)
 169             {
 170             if ((status = RpcNsBindingUnexport (RPC_C_NS_SYNTAX_DEFAULT, pszENTRYNAME, ITaAfsAdminSvr_v1_0_s_ifspec, NULL)) != 0)
 171                {
 172                Print (dlWARNING, TEXT("RpcNsBindingExport failed; error 0x%08lX"), status);
 173                }
 174             }
 175 #endif
 176          if ((status = RpcEpUnregister (ITaAfsAdminSvr_v1_0_s_ifspec, pBindingVector, NULL)) != 0)
 177             {
 178             Print (dlWARNING, TEXT("RpcEpUnregister failed; error 0x%08lX"), status);
 179             }
 180          }
 181       }
 182
 183    Print (TEXT("Shutting down...\n"));
 184
 185    if ((status = RpcServerUnregisterIf (0, 0, FALSE)) != 0)
 186       {
 187       Print (dlWARNING, TEXT("RpcServerUnregisterIf failed; error 0x%08lX"), status);
 188       exit (-1);
 189       }
 190
 191    return (fSuccess) ? (0) : (-1);
 192 }
 193
 194
 195 extern "C" void __RPC_FAR * __RPC_USER MIDL_user_allocate (size_t cbAllocate)
 196 {
 197    return (void __RPC_FAR *)Allocate (cbAllocate);
 198 }
 199
 200
 201 extern "C" void __RPC_USER MIDL_user_free (void __RPC_FAR *pData)
 202 {
 203    Free (pData);
 204 }
 205