|
|
/*
* This program is used to edit the names that the NT LM server is using * on the network (while the LM server is running). * * Type 'srvname -?' for usage * * IsaacHe 3/24/94 * * Revision Histroy: * JeffJuny 3/20/97: Fix ANSI/Unicode CRT conversion problem * Mmccr 12/7/98: Added -m option, added param to err */
extern "C" {#include <nt.h>
#include <ntrtl.h>
#include <nturtl.h>
#include <windows.h>
#include <stdlib.h>
#include <stdio.h>
#include <lm.h>
#include <string.h>
#include <locale.h>
}
/*
* print out an error message and return */static voiderr( char *text, WCHAR *name, ULONG code ){ int i; char msg[ 100 ];
i = FormatMessageA( FORMAT_MESSAGE_FROM_SYSTEM | sizeof( msg ), NULL, code, 0, msg, sizeof(msg), NULL ); if( i ) fprintf( stderr, "%s%ws%ws%s %s\n", text?text:"", name?L" - ":L"", name?name:L"", text?" :":"", msg ); else fprintf( stderr, "%s%ws%ws%s error %X\n", text?text:"", name?L" - ":L"", name?name:L"", text?" :":"", code );}
/*
* Print out a handy usage message */voidUsage( char *name ){ fprintf( stderr, "Usage: %s [ options ] [ name ]\n", name ); fprintf( stderr, " Options:\n" ); fprintf( stderr, " -s server point to 'server'\n" ); fprintf( stderr, " -d delete 'name' from the server's list\n" ); fprintf( stderr, " 'name' is added by default if -d not present\n" ); fprintf( stderr, " -D domain have 'name' act as if it is in 'domain'\n" ); fprintf( stderr, " -m multiple Append numbers from 1 to 'multiple' to 'name'\n" ); fprintf( stderr, " results in machine having multiple + 1 names added\n" );}
__cdeclmain( int argc, char *argv[] ){ DWORD retval; DWORD NameMultiple = 0; DWORD NumberLength = 0; LPWSTR TargetServerName = NULL; WCHAR serverNameBuf[ 100 ]; LPWSTR DomainName = NULL; WCHAR domainNameBuf[ 100 ]; WCHAR newServerNameBuf[ 100 ]; WCHAR safenewServerNameBuf[ 100 ]; LPWSTR NewServerName = NULL; CHAR *NewName = NULL; CHAR ComputerNameBuf[MAX_COMPUTERNAME_LENGTH + 1]; DWORD dwbuflen =sizeof( ComputerNameBuf ); BOOLEAN DeleteTheName = FALSE; PSERVER_INFO_100 si100; DWORD j; int i; WCHAR wtempbuf[5]; char buf[ 500 ];
setlocale(LC_ALL, ""); for( i=1; i < argc; i++ ) { if( argv[i][0] == '-' || argv[i][0] == '/' ) { switch( argv[i][1] ) { case 's': if( i == argc-1 ) { fprintf( stderr, "Must supply a server name with -s option\n" ); return 1; } mbstowcs( serverNameBuf, argv[ ++i ], sizeof( serverNameBuf ) ); TargetServerName = serverNameBuf; break;
case 'D': if( DeleteTheName == TRUE ) { fprintf( stderr, "-d and -D can not be used together\n" ); return 1; }
if( i == argc-1 ) { fprintf( stderr, "Must supply a domain name with -D option\n" ); return 1; } mbstowcs( domainNameBuf, argv[ ++i ], sizeof( domainNameBuf ) ); DomainName = domainNameBuf; break;
case 'm': if( i == argc-1 ) { fprintf( stderr, "Must supply an integer multiple\n" ); return 1; }
NameMultiple = atoi(argv[++i]);
if( NameMultiple <= 0 ) { fprintf( stderr, "Multiple value is not a valid value\n" ); return 1; } if( NameMultiple > 1024 ) { fprintf( stderr, "Multiple value must not be larger than 1024\n" ); return 1; } NumberLength = strlen(argv[i]); break;
case 'd': DeleteTheName = TRUE; break;
default: fprintf( stderr, "%s : invalid option\n", argv[i] ); case '?': Usage( argv[0] ); return 1; } } else if( NewName == NULL ) { NewName = argv[i]; mbstowcs( newServerNameBuf, NewName, sizeof( newServerNameBuf ) ); NewServerName = newServerNameBuf;
} else { Usage( argv[0] ); return 1; } }
if( DeleteTheName == TRUE && NewName == NULL ) { fprintf( stderr, "You must supply the name to delete\n" ); return 1; }
if ((NewName == NULL)&&(NameMultiple == 0)) { //
// Print the current list of transports
//
DWORD entriesread = 0, totalentries = 0, resumehandle = 0; DWORD entriesread1 = 0; PSERVER_TRANSPORT_INFO_0 psti0; PSERVER_TRANSPORT_INFO_1 psti1; DWORD total;
retval = NetServerTransportEnum ( TargetServerName, 1, (LPBYTE *)&psti1, (DWORD)-1, &entriesread1, &totalentries, &resumehandle );
if( retval != NERR_Success ) { entriesread1 = 0; }
resumehandle = 0; totalentries = 0; retval = NetServerTransportEnum ( TargetServerName, 0, (LPBYTE *)&psti0, (DWORD)-1, &entriesread, &totalentries, &resumehandle );
if( retval != NERR_Success ) { err( "Could not get server transports", NULL, retval ); return retval; }
if( entriesread != totalentries ) { fprintf( stderr, "entries read = %d, total entries = %d\n", entriesread, totalentries ); fprintf( stderr, "Unable to read all the transport names!\n" ); return 1; }
for( total=i=0; i < (int)entriesread; i++ ) {
printf( "%-16.16s", psti0[i].svti0_transportaddress );
if( entriesread1 > (DWORD)i ) { printf( "%-16.16ws", psti1[i].svti1_domain); }
printf( " %-58.58ws", psti0[i].svti0_transportname );
printf( "%4u workstation%s\n", psti0[i].svti0_numberofvcs, psti0[i].svti0_numberofvcs != 1 ? "s" : "" );
total += psti0[i].svti0_numberofvcs; } if( total ) { printf( " %s-----\n", entriesread1?" ":"" ); printf( " %s%7u\n", entriesread1?" ":"", total ); } return 0; } else if (NewName == NULL) { if (TargetServerName == NULL) GetComputerName(newServerNameBuf, &dwbuflen); else wcscpy(newServerNameBuf, TargetServerName);
NewServerName = newServerNameBuf; wcstombs(ComputerNameBuf, NewServerName, sizeof(ComputerNameBuf)); NewName = ComputerNameBuf; }
if (strlen(NewName) > MAX_COMPUTERNAME_LENGTH) { fprintf( stderr, "The new name you have chosen exceeds the maximum computername length.\n" ); fprintf( stderr, "Please select a different name.\n" ); return 1; } if (NameMultiple > 0) { if ((NumberLength + strlen(NewName)) > MAX_COMPUTERNAME_LENGTH) { i = MAX_COMPUTERNAME_LENGTH - strlen(NewName); i = NumberLength - i; fprintf( stderr, "The Multiple you have chosen when concatinated with the servername\n" ); fprintf( stderr, "exceeds the maximum computername length. Try reducing your\n" ); if ((DWORD) i == NumberLength) { fprintf( stderr, "chosen servernames length.\n"); } else fprintf( stderr, "multiple by %d orders of magnitude\n", i ); return 1; } }
wcscpy(safenewServerNameBuf, NewServerName);
if( DeleteTheName == FALSE ) { for (j = 0; j <= NameMultiple; j++) { if (j) { wcscpy(NewServerName, safenewServerNameBuf); _itow(j, wtempbuf, 10); wcscat(NewServerName, wtempbuf); } //
// Add the new name to all of the transports
//
retval = NetServerComputerNameAdd( TargetServerName, DomainName, NewServerName );
if( retval != NERR_Success ) { err( "NetServerComputerNameAdd", NewServerName, retval ); } else { printf("Added Name: %ws on Server: %ws%ws%ws\n", NewServerName, TargetServerName, DomainName?L", Domain: ":L"", DomainName?DomainName:L""); } } return retval; }
//
// Must be wanting to delete the name from all the networks.
//
//
// Make sure we don't delete the 'real name' that the server is known by. Pre 3.51
// servers did not ACL protect this api!
//
retval = NetServerGetInfo( TargetServerName, 100, (LPBYTE *)&si100 ); if( retval != STATUS_SUCCESS ) { err( "Can not get target server name", NULL, retval ); return retval; }
if( si100 == NULL ) { fprintf( stderr, "NetServerGetInfo returned a NULL ptr, but no error!\n" ); return 1; }
for (j = 0; j <= NameMultiple; j++) { if (j) { wcscpy(NewServerName, safenewServerNameBuf); _itow(j, wtempbuf, 10); wcscat(NewServerName, wtempbuf); }
if( !_wcsicmp( si100->sv100_name, NewServerName ) ) { fprintf( stderr, "The primary name of %ws is %ws.\n", TargetServerName ? TargetServerName : L"this server", NewServerName ); fprintf( stderr, "\tYou can not delete the primary name of the server\n" ); } else { retval = NetServerComputerNameDel( TargetServerName, NewServerName ); if( retval != STATUS_SUCCESS ) { err( "NetServerComputerNameDelete", NewServerName, retval ); } else { printf("Deleted Name: %ws on Server: %ws%ws%ws\n", NewServerName, TargetServerName, DomainName?L", Domain: ":L"", DomainName?DomainName:L""); } } } return 0;}
|
