![[BACK]](/cvsweb/icons/back.gif){kind=icon}
Return to rcmdsrv.c CVS log ![[TXT]](/cvsweb/icons/text.gif){kind=icon}
![[DIR]](/cvsweb/icons/dir.gif){kind=icon}
Up to [WindowsNT SDKs] / mstools / samples / sdktools / rshell / server|
Return to rcmdsrv.c CVS log | Up to [WindowsNT SDKs] / mstools / samples / sdktools / rshell / server |
1.1 ! root 1: /****************************** Module Header ******************************\ ! 2: * Module Name: rcmdsrv.c ! 3: * ! 4: * Copyright (c) 1991, Microsoft Corporation ! 5: * ! 6: * Remote shell server main module ! 7: * ! 8: * History: ! 9: * 06-29-92 Davidc Created. ! 10: \***************************************************************************/ ! 11: ! 12: #include "rcmdsrv.h" ! 13: ! 14: #define PIPE_NAME TEXT("\\\\.\\pipe\\rshell") ! 15: ! 16: // ! 17: // Number of pipe instances ! 18: // ! 19: ! 20: #define MAX_SESSIONS PIPE_UNLIMITED_INSTANCES ! 21: ! 22: ! 23: // ! 24: // Define pipe timeout (ms) ! 25: // Only used by WaitNamedPipe ! 26: // ! 27: ! 28: #define PIPE_TIMEOUT 1000 ! 29: ! 30: ! 31: ! 32: // ! 33: // Ctrl-C handler routine ! 34: // ! 35: ! 36: BOOL ! 37: CtrlHandler( ! 38: DWORD CtrlType ! 39: ) ! 40: { ! 41: // ! 42: // We'll handle Ctrl-C events ! 43: // ! 44: ! 45: return (CtrlType == CTRL_C_EVENT); ! 46: } ! 47: ! 48: ! 49: ! 50: // ! 51: // Main ! 52: // ! 53: ! 54: int ! 55: _CRTAPI1 main( ! 56: int argc, ! 57: char **argv ! 58: ) ! 59: { ! 60: SECURITY_ATTRIBUTES SecurityAttributes; ! 61: SECURITY_DESCRIPTOR SecurityDescriptor; ! 62: BOOL Result; ! 63: DWORD WaitResult; ! 64: HANDLE SessionHandle = NULL; ! 65: ! 66: // ! 67: // Set the error mode so we don't generate popups ! 68: // ! 69: ! 70: SetErrorMode(SEM_FAILCRITICALERRORS | SEM_NOOPENFILEERRORBOX); ! 71: ! 72: // ! 73: // Install a handler for Ctrl-C ! 74: // ! 75: ! 76: if (!SetConsoleCtrlHandler((PHANDLER_ROUTINE) &CtrlHandler, TRUE)) { ! 77: printf("Failed to install control-C handler, error = %d\n", GetLastError()); ! 78: return(1); ! 79: } ! 80: ! 81: ! 82: // ! 83: // Setup the security descriptor to put on the named pipe. ! 84: // For now give world access. ! 85: // ! 86: ! 87: Result = InitializeSecurityDescriptor(&SecurityDescriptor, SECURITY_DESCRIPTOR_REVISION); ! 88: assert(Result); ! 89: ! 90: Result = SetSecurityDescriptorDacl(&SecurityDescriptor, FALSE, NULL, FALSE); ! 91: assert(Result); ! 92: ! 93: SecurityAttributes.nLength = sizeof(SecurityAttributes); ! 94: SecurityAttributes.lpSecurityDescriptor = &SecurityDescriptor; ! 95: SecurityAttributes.bInheritHandle = FALSE; ! 96: ! 97: ! 98: // ! 99: // Tell the user what's happening ! 100: // ! 101: ! 102: printf("Remote command shell running.\n"); ! 103: printf("Use Ctrl-Break to terminate\n\n"); ! 104: ! 105: ! 106: // ! 107: // Do loop inside try-finally so we always delete any session on exit ! 108: // ! 109: ! 110: try { ! 111: ! 112: BOOL Done = FALSE; ! 113: ! 114: // ! 115: // Loop waiting for a client to connect. ! 116: // ! 117: ! 118: while (!Done) { ! 119: ! 120: HANDLE ConnectHandle; ! 121: SESSION_DISCONNECT_CODE DisconnectCode; ! 122: HANDLE PipeHandle; ! 123: ! 124: // ! 125: // Create an instance of the named pipe ! 126: // ! 127: ! 128: PipeHandle = CreateNamedPipe(PIPE_NAME, ! 129: PIPE_ACCESS_DUPLEX | FILE_FLAG_OVERLAPPED, ! 130: PIPE_TYPE_BYTE | PIPE_WAIT, ! 131: MAX_SESSIONS, // Number of pipes ! 132: 0, // Default out buffer size ! 133: 0, // Default in buffer size ! 134: PIPE_TIMEOUT, // Timeout in ms ! 135: &SecurityAttributes ! 136: ); ! 137: ! 138: if (PipeHandle == INVALID_HANDLE_VALUE ) { ! 139: printf("Failed to create named pipe instance, error = %d\n", GetLastError()); ! 140: break; // out of while ! 141: } ! 142: ! 143: ! 144: // ! 145: // Wait for a client to connect ! 146: // ! 147: ! 148: printf("Waiting for client connect\n"); ! 149: ! 150: ! 151: Result = ConnectNamedPipe(PipeHandle, NULL); ! 152: ! 153: if (!Result) { ! 154: ! 155: DWORD Error = GetLastError(); ! 156: ! 157: if (Error == ERROR_PIPE_CONNECTED) { ! 158: ! 159: // ! 160: // The client has already connected (fast work!) ! 161: // ! 162: ! 163: } else { ! 164: ! 165: printf("Connect named pipe failed, error = %d\n", GetLastError()); ! 166: MyCloseHandle(PipeHandle, "client pipe"); ! 167: break; ! 168: } ! 169: } ! 170: ! 171: ! 172: printf("Client connected\n"); ! 173: ! 174: ! 175: // ! 176: // Create a new session if necessary ! 177: // ! 178: ! 179: if (SessionHandle == NULL) { ! 180: ! 181: SessionHandle = CreateSession(); ! 182: ! 183: if (SessionHandle == NULL) { ! 184: printf("Failed to create session\n"); ! 185: MyCloseHandle(PipeHandle, "client pipe"); ! 186: break; ! 187: } ! 188: } ! 189: ! 190: ! 191: // ! 192: // Connect the pipe to our session ! 193: // ! 194: ! 195: ConnectHandle = ConnectSession(SessionHandle, PipeHandle); ! 196: if (ConnectHandle == NULL) { ! 197: MyCloseHandle(PipeHandle, "client pipe"); ! 198: printf("Failed to connect session\n"); ! 199: break; ! 200: } ! 201: ! 202: // ! 203: // Wait for session disconnect ! 204: // ! 205: ! 206: WaitResult = WaitForSingleObject(ConnectHandle, INFINITE); ! 207: if (WaitResult != 0) { ! 208: printf("Unexpected result from wait on connect handle, result = %d\n", WaitResult); ! 209: } ! 210: ! 211: // ! 212: // Disconnect the session to find out why it happened ! 213: // ! 214: ! 215: DisconnectCode = DisconnectSession(SessionHandle); ! 216: ! 217: switch (DisconnectCode) { ! 218: ! 219: case ShellEnded: ! 220: ! 221: // ! 222: // The shell ended, delete the session ! 223: // ! 224: ! 225: printf("Shell terminated\n"); ! 226: ! 227: DeleteSession(SessionHandle); ! 228: SessionHandle = NULL; ! 229: break; // out of switch ! 230: ! 231: case ClientDisconnected: ! 232: ! 233: // ! 234: // The client disconnected, keep the session and go wait ! 235: // for another client to connect to it. ! 236: // ! 237: ! 238: printf("Client disconnected\n"); ! 239: ! 240: // ............... bug workaround ............. ! 241: // To work around async read bug with exitted thread, ! 242: // always kill off the shell and start a new one ! 243: DeleteSession(SessionHandle); ! 244: SessionHandle = NULL; ! 245: // ............................................ ! 246: ! 247: break; ! 248: ! 249: case ConnectError: ! 250: case DisconnectError: ! 251: default: ! 252: ! 253: printf("Disconnect session returned unexpected code : %d\n", DisconnectCode); ! 254: Done = TRUE; ! 255: break; // out of switch ! 256: } ! 257: ! 258: ! 259: // ! 260: // Go back and wait for a client to connect ! 261: // ! 262: ! 263: } ! 264: ! 265: ! 266: } finally { ! 267: ! 268: DbgPrint("Finally being called\n"); ! 269: ! 270: // ! 271: // Delete any existing session ! 272: // ! 273: ! 274: if (SessionHandle != NULL) { ! 275: DeleteSession(SessionHandle); ! 276: } ! 277: ! 278: } ! 279: ! 280: return(0); ! 281: } ! 282:
This archive runs on limited infrastructure. Preserving old code on modern bandwidth. Automated agents are requested to crawl responsibly.