|
|
/*++
Copyright (c) 1998 Microsoft Corporation
Module Name:
Routing\Netsh\wins\srvrhndl.c
Abstract:
WINS Command dispatcher.
Created by:
Shubho Bhattacharya(a-sbhatt) on 12/10/98
--*/
#include "precomp.h"
extern ULONG g_ulSrvrNumTopCmds;extern ULONG g_ulSrvrNumGroups;
extern CMD_GROUP_ENTRY g_SrvrCmdGroups[];extern CMD_ENTRY g_SrvrCmds[];
DWORD g_dwSearchCount = 0;BOOL g_fHeader = FALSE;
WCHAR **LA_Table = NULL;LARGE_INTEGER **SO_Table = NULL;u_char **NBNames = NULL;WINSERVERS * WinServers = NULL;
LPWSTRGetDateTimeString(DWORD_PTR TimeStamp, BOOL fShort, int *piType);
DWORDHandleSrvrDump( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )
/*++
Routine Description : Dumps the current configuration of the Wins Server.
Arguments : Does not take any arguments.
Return Value: Returns the status of the operation.
--*/
{ DWORD Status = NO_ERROR;
if( dwArgCount > dwCurrentIndex ) { if( IsHelpToken(ppwcArguments[dwCurrentIndex]) is TRUE ) { DisplayMessage(g_hModule, HLP_WINS_DUMP_EX); } }
Status = WinsDumpServer(g_ServerIpAddressUnicodeString, g_ServerNetBiosName, g_hBind, g_BindData); if( Status is NO_ERROR ) { DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); } else if( Status is ERROR_FILE_NOT_FOUND ) { DisplayMessage(g_hModule, EMSG_WINS_NOT_CONFIGURED); } else { DisplayErrorMessage(EMSG_SRVR_DUMP, Status); } return Status;
}
DWORDHandleSrvrHelp( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Displays the help for Wins Server context.
Arguments : Does not take any arguments.
Return Value: Returns the status of the operation. NO_ERROR always.
--*/
{ DWORD i, j;
for(i = 0; i < g_ulSrvrNumTopCmds -2; i++) { if ((g_SrvrCmds[i].dwCmdHlpToken == WINS_MSG_NULL) || !g_SrvrCmds[i].pwszCmdToken[0] ) { continue; }
DisplayMessage(g_hModule, g_SrvrCmds[i].dwShortCmdHelpToken);
}
for(i = 0; i < g_ulSrvrNumGroups; i++) { if ((g_SrvrCmdGroups[i].dwShortCmdHelpToken == WINS_MSG_NULL) || !g_SrvrCmdGroups[i].pwszCmdGroupToken[0] ) { continue; }
DisplayMessage(g_hModule, g_SrvrCmdGroups[i].dwShortCmdHelpToken); } DisplayMessage(g_hModule, WINS_FORMAT_LINE);
return NO_ERROR;}
DWORDHandleSrvrAddName( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Adds and registers a name record to the WINS serverArguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory, Record Name and Ip Address Optional, Endchar, Scope, RecordType, NodeType, GroupType Note : GroupType is ignored if EndChar is specified.Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount = 0; TAG_TYPE pttTags[] = { {WINS_TOKEN_NAME, TRUE, FALSE}, {WINS_TOKEN_ENDCHAR, FALSE, FALSE}, {WINS_TOKEN_SCOPE, FALSE, FALSE}, {WINS_TOKEN_RECORDTYPE, FALSE, FALSE}, {WINS_TOKEN_GROUP, FALSE, FALSE}, {WINS_TOKEN_NODE, FALSE, FALSE}, {WINS_TOKEN_IP, TRUE, FALSE}, }; PDWORD pdwTagNum = NULL, pdwTagType = NULL;
WCHAR wszName[MAX_STRING_LEN+1] = {L'\0'}; BOOL fEndChar = FALSE; BOOL fDomain = FALSE; CHAR ch16thChar = 0x00;
BOOL fScope = FALSE; WCHAR wszScope[MAX_STRING_LEN] = {L'\0'}; BOOL fStatic = TRUE; DWORD dwRecType = WINSINTF_E_UNIQUE; BYTE rgbNodeType = WINSINTF_E_PNODE; PDWORD pdwIpAddress = NULL; DWORD dwIpCount = 0; DWORD dwStrLen = 0; LPWSTR pwszTemp = NULL; WINSINTF_RECORD_ACTION_T RecAction = {0}; PWINSINTF_RECORD_ACTION_T pRecAction = NULL; LPSTR pszTempName = NULL;
memset(&RecAction, 0x00, sizeof(WINSINTF_RECORD_ACTION_T)); RecAction.fStatic = fStatic;
//We need at least Name and Ip for the record.
if( dwArgCount < dwCurrentIndex + 2 ) { DisplayMessage(g_hModule, HLP_SRVR_ADD_NAME_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
//Start processing the arguments passed by ppwcArguments and dwArgCount
dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum); if( Status isnot NO_ERROR ) goto ErrorReturn; for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
i = dwTagCount; for( j=0; j<i; j++ ) { switch(pdwTagType[j]) { //Name of the record ( Compulsory )
case 0: { DWORD dwLen = 0; dwLen = wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); if( dwLen is 0 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dwLen = ( 255 < dwLen ) ? 255 : dwLen; memcpy(wszName, ppwcArguments[dwCurrentIndex+pdwTagNum[j]], dwLen*sizeof(WCHAR)); wszName[dwLen] = L'\0'; break; } //End Char or 16th Character( Optional )
case 1: { DWORD dwLen = 0, k=0; fEndChar = TRUE; dwLen = wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); if( dwLen > 2 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } for( k=0; k<dwLen; k++ ) { WCHAR wc = ppwcArguments[dwCurrentIndex+pdwTagNum[j]][k]; if( towlower(wc) >= L'a' and towlower(wc) <= L'z' ) { if( towlower(wc) > L'f' ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } }
ch16thChar = StringToHexA(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]);
break; } //Scope ( Optional )
case 2: { DWORD dwLen; dwLen = wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); fScope = TRUE; if( dwLen is 0 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dwLen = ( 255 < dwLen ) ? 255 : dwLen; memcpy(wszScope, ppwcArguments[dwCurrentIndex+pdwTagNum[j]], dwLen*sizeof(WCHAR)); wszScope[dwLen] = L'\0'; break; } //Record Type ie Static or Dynamic ( Optional )
case 3: { if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) > 1 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } else { WCHAR wc = ppwcArguments[dwCurrentIndex+pdwTagNum[j]][0]; if( wc is L'1' ) fStatic = FALSE; else if( wc is L'0' ) fStatic = TRUE; else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } break; } // Group Type ( Optional )
case 4: { DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 0); switch(dw) { case 0: { dwRecType = WINSINTF_E_UNIQUE; break; } case 1: { dwRecType = WINSINTF_E_NORM_GROUP; break; } case 2: { dwRecType = WINSINTF_E_SPEC_GROUP; break; } case 3: { dwRecType = WINSINTF_E_MULTIHOMED; break; } case 4: { fDomain = TRUE; dwRecType = WINSINTF_E_SPEC_GROUP; break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } break; } //Node Type( Optional )
case 5: { DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 0);
switch(dw) { case 0: { rgbNodeType = WINSINTF_E_BNODE; break; } case 1: { rgbNodeType = WINSINTF_E_PNODE; break; } case 3: { rgbNodeType = WINSINTF_E_HNODE; break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } break; } //IP Address corresponding to the record( Compulsory )
case 6: { LPWSTR pszIps = NULL; DWORD dwIpLen = 0; LPWSTR pTemp = NULL; dwIpCount = 0;
dwIpLen = wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); if( dwIpLen is 0 ) { break; } pszIps = WinsAllocateMemory((dwIpLen+1)*sizeof(WCHAR)); if( pszIps is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
wcscpy(pszIps, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); if( pszIps[0] isnot L'{' or pszIps[dwIpLen-1] isnot L'}') { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } pTemp = pszIps+1; pszIps[dwIpLen-1] = L'\0';
pTemp = wcstok(pTemp, L","); while(( pTemp isnot NULL ) && (dwIpCount < WINSINTF_MAX_MEM ) ) { PDWORD pdwTemp = NULL; if( IsIpAddress(pTemp) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dwIpCount++;
pdwTemp = WinsAllocateMemory(dwIpCount*sizeof(DWORD)); if( pdwTemp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } if( pdwIpAddress ) { memcpy(pdwTemp, pdwIpAddress, (dwIpCount-1)*sizeof(DWORD)); WinsFreeMemory(pdwIpAddress); pdwIpAddress = NULL; } pdwTemp[dwIpCount-1] = StringToIpAddress(pTemp); pdwIpAddress = pdwTemp; pTemp = wcstok(NULL, L","); } WinsFreeMemory(pszIps); pszIps = NULL; break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } } //Process Name, Scope and Endchar if specified.
_wcsupr(wszName); _wcsupr(wszScope);
RecAction.pName = WinsAllocateMemory(273); if( RecAction.pName is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } pszTempName = WinsUnicodeToOem(wszName, NULL);
if( pszTempName is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwStrLen = strlen(pszTempName);
if( dwStrLen >= 16 ) { DisplayMessage(g_hModule, EMSG_SRVR_INVALID_NETBIOS_NAME); Status = ERROR_INVALID_PARAMETER; WinsFreeMemory(pszTempName); pszTempName = NULL; goto ErrorReturn; }
strncpy(RecAction.pName, pszTempName, ( 16 > dwStrLen ) ? dwStrLen : 16);
WinsFreeMemory(pszTempName); pszTempName = NULL;
if( fDomain is TRUE && fEndChar is FALSE ) { ch16thChar = StringToHexA(L"1C"); fEndChar = TRUE; }
for( j=dwStrLen; j<16; j++ ) { RecAction.pName[j] = ' '; } if( fEndChar is TRUE ) { RecAction.pName[15] = (CHAR)ch16thChar; } RecAction.pName[16] = '\0';
dwStrLen = 16;
if( fScope ) { DWORD dwLen; RecAction.pName[dwStrLen] = '.';
pszTempName = WinsUnicodeToOem(wszScope, NULL); if( pszTempName is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } dwLen = strlen(pszTempName); dwLen = ( 255 - dwStrLen < dwLen ) ? 255 - dwStrLen : dwLen;
strncpy(RecAction.pName+dwStrLen+1, pszTempName, dwLen); WinsFreeMemory(pszTempName); pszTempName = NULL;
RecAction.pName[dwStrLen+dwLen+1] = '\0'; dwStrLen = strlen(RecAction.pName);
if( fEndChar and ch16thChar is 0x00 ) dwStrLen++; }
RecAction.NameLen = dwStrLen; RecAction.Cmd_e = WINSINTF_E_INSERT; RecAction.fStatic = fStatic;
if( pdwIpAddress is NULL ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } if( dwRecType isnot WINSINTF_E_NORM_GROUP and dwRecType isnot WINSINTF_E_SPEC_GROUP) { RecAction.NodeTyp = rgbNodeType; }
//Treat each of the rectype when no endchar is specified specially.
//This part of the code needs to be cleaned up after Beta3
if( fEndChar is FALSE ) { if( dwRecType is WINSINTF_E_SPEC_GROUP or dwRecType is WINSINTF_E_MULTIHOMED ) { RecAction.pAdd = WinsAllocateMemory(dwIpCount*sizeof(WINSINTF_ADD_T)); if( RecAction.pAdd is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } for( j=0; j<dwIpCount; j++ ) { (RecAction.pAdd+j)->IPAdd = pdwIpAddress[j]; (RecAction.pAdd+j)->Type = 0; (RecAction.pAdd+j)->Len = 4; } RecAction.NoOfAdds = dwIpCount; } else { RecAction.Add.IPAdd = pdwIpAddress[0]; RecAction.Add.Type = 0; RecAction.Add.Len = 4; }
switch(dwRecType) { case WINSINTF_E_UNIQUE: { CHAR Type[]={0x03, 0x20, 0x00};
for( i=0; i<3; i++ ) { *(RecAction.pName + 15) = Type[i]; if( Type[i] is 0x00 ) { RecAction.pName[16] = '\0'; RecAction.NameLen = 16; } RecAction.TypOfRec_e = dwRecType; pRecAction = &RecAction; Status = WinsRecordAction(g_hBind, &pRecAction); if( Status isnot NO_ERROR ) goto ErrorReturn; } break; } case WINSINTF_E_NORM_GROUP: { RecAction.pName[15] = (CHAR)0x1E; pRecAction = &RecAction; RecAction.TypOfRec_e = dwRecType; Status = WinsRecordAction(g_hBind, &pRecAction); if( Status isnot NO_ERROR ) goto ErrorReturn;
break; }
case WINSINTF_E_SPEC_GROUP: { RecAction.pName[15] = (CHAR)0x20; pRecAction = &RecAction; RecAction.TypOfRec_e = dwRecType; Status = WinsRecordAction(g_hBind, &pRecAction); if( Status isnot NO_ERROR ) goto ErrorReturn; break; }
case WINSINTF_E_MULTIHOMED: { CHAR Type[]={0x03, 0x20, 0x00}; for( i=0; i<3; i++ ) { *(RecAction.pName + 15) = Type[i]; if( Type[i] is 0x00 ) { *(RecAction.pName+16) = '\0'; RecAction.NameLen = 16; }
RecAction.TypOfRec_e = dwRecType; pRecAction = &RecAction; RecAction.NodeTyp = WINSINTF_E_PNODE; Status = WinsRecordAction(g_hBind, &pRecAction); if( Status isnot NO_ERROR ) goto ErrorReturn; } break; } } }
//Otherwise when Endchar is specified
else { //if endchar is 0x00, ignore the scope if spcefied
if( RecAction.pName[15] is 0x00 ) { RecAction.NameLen = 16; } //If endchar is 0x1C
if( RecAction.pName[15] is 0x1C ) { RecAction.TypOfRec_e = WINSINTF_E_SPEC_GROUP; RecAction.NodeTyp = 0; } //else if EndChar is 0x1E or 0x1D
else if( RecAction.pName[15] is 0x1E or RecAction.pName[15] is 0x1D ) { RecAction.TypOfRec_e = WINSINTF_E_NORM_GROUP; RecAction.NodeTyp = 0; }
if( RecAction.TypOfRec_e is WINSINTF_E_SPEC_GROUP ) { RecAction.pAdd = WinsAllocateMemory(dwIpCount*sizeof(WINSINTF_ADD_T)); if( RecAction.pAdd is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } for( j=0; j<dwIpCount; j++ ) { (RecAction.pAdd+j)->IPAdd = pdwIpAddress[j]; (RecAction.pAdd+j)->Type = 0; (RecAction.pAdd+j)->Len = 4; } RecAction.NoOfAdds = dwIpCount; } else { RecAction.Add.IPAdd = pdwIpAddress[0]; RecAction.Add.Type = 0; RecAction.Add.Len = 4; } pRecAction = &RecAction;
Status = WinsRecordAction(g_hBind, &pRecAction );
if( Status isnot NO_ERROR ) goto ErrorReturn; }
CommonReturn: if( Status is NO_ERROR ) DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS);
if( pszTempName ) { WinsFreeMemory(pszTempName); pszTempName = NULL; }
if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; } if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; } if( RecAction.pName ) { WinsFreeMemory(RecAction.pName); RecAction.pName = NULL; } if( RecAction.pAdd ) { WinsFreeMemory(RecAction.pAdd); RecAction.pAdd = NULL; }
if( pdwIpAddress ) { WinsFreeMemory(pdwIpAddress); pdwIpAddress = NULL; } return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_ADD_NAME, Status); goto CommonReturn;
}
DWORDHandleSrvrAddPartner( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Adds a partner ( either Push or Pull or Both ) to the WINS serverArguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory, ServerIpAddress Optional, ServerNetBios name and PartnerType. Note : Server NetBios name is required when Ip address can not be resolved to a name. PartherType by default is both. Otherwise whatever specified.Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount; TAG_TYPE pttTags[] = { {WINS_TOKEN_SERVER, TRUE, FALSE}, {WINS_TOKEN_NETBIOS, FALSE, FALSE}, {WINS_TOKEN_TYPE, FALSE, FALSE}, }; LPWSTR pwszServerName = NULL; WCHAR wcServerIpAdd[MAX_IP_STRING_LEN+1] = {L'\0'}; PDWORD pdwTagType=NULL, pdwTagNum=NULL; BOOL fPush = TRUE, fPull = TRUE; HKEY hServer = NULL, hPartner = NULL, hDefault = NULL, hKey = NULL;
LPWSTR pTemp = NULL; DWORD dwKeyLen = 0; DWORD dwData = 0, dwDataLen = 0, dwType = 0;
BOOL fIsNetBios = TRUE;
//Need at least the server Ip Address
if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, HLP_SRVR_ADD_PARTNER_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } //Start processing the arguments based on ppwcArguments and dwArgCount and dwCurrnetIndex
dwNumArgs = dwArgCount - dwCurrentIndex; dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum);
if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
for( j=0; j<dwTagCount; j++) { switch(pdwTagType[j]) { //Server IP Address. Try to resolve the IP to a name
case 0: { struct hostent * lpHostEnt = NULL; CHAR cAddr[16]; BYTE pbAdd[4]; char szAdd[4]; int k = 0, l=0; DWORD dwLen,nLen = 0; CHAR *pTemp = NULL; CHAR *pNetBios = NULL; if( IsIpAddress( ppwcArguments[dwCurrentIndex+pdwTagNum[j]] ) is FALSE ) { Status = ERROR_INVALID_IPADDRESS; goto ErrorReturn; }
//if( IsIpAddress( ppwcArguments[dwCurrentIndex+pdwTagNum[j]] ) )
else { DWORD dwIp = inet_addr(WinsUnicodeToAnsi(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL)); lpHostEnt = gethostbyaddr((char *)&dwIp, 4, AF_INET); if(lpHostEnt isnot NULL )//Valid IP Address
{ wcscpy(wcServerIpAdd, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); } else { wcscpy(wcServerIpAdd, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); fIsNetBios = FALSE; break; } }
dwLen = strlen(lpHostEnt->h_name); pTemp = WinsAllocateMemory(dwLen+1); if( pTemp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
strcpy(pTemp, lpHostEnt->h_name); pNetBios = strchr(pTemp, '.');
if( pNetBios isnot NULL ) { dwLen = (DWORD)(pNetBios - pTemp); pTemp[dwLen] = '\0'; }
pwszServerName = WinsAllocateMemory((dwLen+1)*sizeof(WCHAR)); if( pwszServerName is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } wcscpy(pwszServerName, WinsAnsiToUnicode(pTemp, NULL));
if( pTemp ) { WinsFreeMemory(pTemp); pTemp = NULL; } break; } //Server NetBios Name. Required only when Ip can not be resolved to a name.
//Otherwise ignored.
case 1: { if( fIsNetBios is FALSE ) { pwszServerName = WinsAllocateMemory((wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]])+1)*sizeof(WCHAR)); if( pwszServerName is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
wcscpy(pwszServerName, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); fIsNetBios = TRUE; } break; } //Partner Type. Default is BOTH
case 2: { DWORD dwType = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dwType = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 0); switch(dwType) { case 0: { fPull = TRUE; fPush = FALSE; break; } case 1: { fPull = FALSE; fPush = TRUE; break; } case 2: default: { fPull = TRUE; fPush = TRUE; break; } } break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
} }
if( fIsNetBios is FALSE ) { Status = ERROR_INVALID_IPADDRESS; goto ErrorReturn; } //Add the partner information to the registry and set the appropriate parameter
{ if( wcslen(g_ServerNetBiosName) > 0 ) { pTemp = g_ServerNetBiosName; } Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer);
if( Status isnot ERROR_SUCCESS ) goto ErrorReturn;
//Add the pull partner information
if( fPull ) { DWORD dwDisposition = 0; Status = RegCreateKeyEx(hServer, PULLROOT, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hPartner, &dwDisposition); if( Status isnot NO_ERROR ) { if( fPush ) goto PUSH; goto ErrorReturn; }
Status = RegCreateKeyEx(hPartner, wcServerIpAdd, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hKey, NULL); if( Status isnot NO_ERROR ) { if( fPush ) goto PUSH; goto ErrorReturn; } Status = RegSetValueEx(hKey, NETBIOSNAME, 0, REG_SZ, (LPBYTE)pwszServerName, (wcslen(pwszServerName)+1)*sizeof(WCHAR)); if(Status isnot NO_ERROR ) { if( fPush ) goto PUSH; goto ErrorReturn; } dwType = REG_DWORD; dwData = 0; dwDataLen = sizeof(DWORD);
Status = RegQueryValueEx( hPartner, PERSISTENCE, 0, &dwType, (LPBYTE)&dwData, &dwDataLen); if( Status isnot NO_ERROR ) { dwData = 0; dwDataLen = sizeof(DWORD); } Status = RegSetValueEx(hKey, PERSISTENCE, 0, REG_DWORD, (LPBYTE)&dwData, dwDataLen); if( Status isnot NO_ERROR ) { if( fPush ) goto PUSH; goto ErrorReturn; }
dwType = REG_DWORD; dwData = 0; dwDataLen = sizeof(DWORD);
Status = RegQueryValueEx(hPartner, WINSCNF_SELF_FND_NM, 0, &dwType, (LPBYTE)&dwData, &dwDataLen);
if( Status isnot NO_ERROR ) { dwData = 0; dwDataLen = sizeof(DWORD); }
Status = RegSetValueEx(hKey, WINSCNF_SELF_FND_NM, 0, REG_DWORD, (LPBYTE)&dwData, dwDataLen);
if( Status isnot NO_ERROR ) { if( fPush ) goto PUSH; goto ErrorReturn; }
Status = RegOpenKeyEx(hServer, DEFAULTPULL, 0, KEY_ALL_ACCESS, &hDefault); if( Status is NO_ERROR ) { dwType = REG_DWORD; dwData = 0; dwDataLen = sizeof(DWORD); Status = RegQueryValueEx(hDefault, WINSCNF_RPL_INTERVAL_NM, 0, &dwType, (LPBYTE)&dwData, &dwDataLen); if(Status isnot NO_ERROR ) { dwData = 1800; dwDataLen = sizeof(DWORD); } } else { dwData = 1800; dwDataLen = sizeof(DWORD); } Status = RegSetValueEx(hKey, WINSCNF_RPL_INTERVAL_NM, 0, REG_DWORD, (LPBYTE)&dwData, dwDataLen);
if( Status isnot NO_ERROR ) { if( fPush ) goto PUSH; goto ErrorReturn; } }
//Add the push partner information
PUSH: if( hKey ) { RegCloseKey(hKey); hKey = NULL; } if( hDefault ) { RegCloseKey(hDefault); hDefault = NULL; }
if( hPartner ) { RegCloseKey(hPartner); hPartner = NULL; } if( fPush ) { DWORD dwDisposition = 0; Status = RegCreateKeyEx(hServer, PUSHROOT, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hPartner, &dwDisposition); if( Status isnot NO_ERROR ) { goto ErrorReturn; } Status = RegCreateKeyEx(hPartner, wcServerIpAdd, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hKey, NULL); if( Status isnot NO_ERROR ) { goto ErrorReturn; } Status = RegSetValueEx(hKey, NETBIOSNAME, 0, REG_SZ, (LPBYTE)pwszServerName, (wcslen(pwszServerName)+1)*sizeof(WCHAR)); if(Status isnot NO_ERROR ) { goto ErrorReturn; }
dwType = REG_DWORD; dwData = 0; dwDataLen = sizeof(DWORD); Status = RegQueryValueEx( hPartner, PERSISTENCE, 0, &dwType, (LPBYTE)&dwData, &dwDataLen); if( Status isnot NO_ERROR ) { dwData = 0; dwDataLen = sizeof(DWORD); }
Status = RegSetValueEx(hKey, PERSISTENCE, 0, REG_DWORD, (LPBYTE)&dwData, dwDataLen); if( Status isnot NO_ERROR ) { goto ErrorReturn; }
dwType = REG_DWORD; dwData = 0; dwDataLen = sizeof(DWORD);
Status = RegQueryValueEx(hPartner, WINSCNF_SELF_FND_NM, 0, &dwType, (LPBYTE)&dwData, &dwDataLen);
if( Status isnot NO_ERROR ) { dwData = 0; dwDataLen = sizeof(DWORD); }
Status = RegSetValueEx(hKey, WINSCNF_SELF_FND_NM, 0, REG_DWORD, (LPBYTE)&dwData, dwDataLen);
if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegOpenKeyEx(hServer, DEFAULTPUSH, 0, KEY_ALL_ACCESS, &hDefault); if( Status is NO_ERROR ) { dwType = REG_DWORD; dwData = 0; dwDataLen = sizeof(DWORD); Status = RegQueryValueEx(hDefault, WINSCNF_UPDATE_COUNT_NM, 0, &dwType, (LPBYTE)&dwData, &dwDataLen); if(Status isnot NO_ERROR ) { dwData = 0; dwDataLen = sizeof(DWORD); } } else { dwData = 0; dwDataLen = sizeof(DWORD); } Status = RegSetValueEx(hKey, WINSCNF_UPDATE_COUNT_NM, 0, REG_DWORD, (LPBYTE)&dwData, dwDataLen);
if( Status isnot NO_ERROR ) { goto ErrorReturn; }
if( hKey ) { RegCloseKey(hKey); hKey = NULL; } if( hDefault ) { RegCloseKey(hDefault); hDefault = NULL; }
if( hPartner ) { RegCloseKey(hPartner); hPartner = NULL; }
} } CommonReturn: if( Status is NO_ERROR ) DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS);
if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; } if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; }
if( pwszServerName ) { WinsFreeMemory(pwszServerName); pwszServerName = NULL; }
if( hServer ) { RegCloseKey(hServer); hServer = NULL; } if( hDefault ) { RegCloseKey(hDefault); hDefault = NULL; } return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_ADD_PARTNER, Status);
if( hKey ) { RegCloseKey(hKey); hKey = NULL; }
if( hPartner ) { RegDeleteKey(hPartner, g_ServerIpAddressUnicodeString); RegCloseKey(hPartner); hPartner = NULL; }
goto CommonReturn;
}
BOOL CheckValidPgOp(HKEY hPartner, BOOL fGrata)/*++
Routine Description : Check whether Persona Mode allows operation for persona grata (fGrata) or non-grata (!fGrata)Arguments : hPartner = opened handle to the 'Partners' registry key fGrata specifies whether the check is done for a persona Grata (TRUE) operation or of a persona Non-Grata (FALSE) operationReturn Value: Returns TRUE if the operation is allowed, FALSE otherwise.
--*/{ DWORD dwPersMode = 0; // default (entry not existant) = Persona Non-Grata
DWORD dwType; DWORD dwSize = sizeof(DWORD);
// don't chare about the return code. If something goes wrong (entry not existant)
// consider Persona Mode as being 'Persona Non-Grata'.
RegQueryValueExA(hPartner, WINSCNF_PERSONA_MODE_NM, NULL, &dwType, (LPVOID)&dwPersMode, &dwSize);
return dwPersMode ? fGrata : !fGrata;}
DWORDHandleSrvrAddPersona( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Add Persona Non Grata servers for the WINS ServerArguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory, List of Server Ip addresses seperated by commas and enclosed by {} ( curly braces )Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount; TAG_TYPE pttTags[] = { {WINS_TOKEN_SERVERS, TRUE, FALSE}, }; DWORD dwTagType = 0, dwTagNum = 0; DWORD dwSize = 0, dwType = REG_MULTI_SZ, dwPngIp = 0, dwTotal = 0; LPWSTR pwszPngIp = NULL; LPWSTR pTemp = NULL; LPWSTR pwszTempKey = NULL; LPBYTE pbByte = NULL; HKEY hServer = NULL, hPartner = NULL; LPDWORD pdwPngIp = NULL; DWORD dwLenCount = 0, dwTemp = 0; BOOL fGrata;
fGrata = (wcsstr(CMD_SRVR_ADD_PNGSERVER, ppwcArguments[dwCurrentIndex-1]) == NULL);
//Needs a parameter always
if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, fGrata ? HLP_SRVR_ADD_PGSERVER_EX : HLP_SRVR_ADD_PNGSERVER_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dwNumArgs = dwArgCount - dwCurrentIndex;
if( wcslen(g_ServerNetBiosName) > 0 ) { pTemp = g_ServerNetBiosName; }
Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer);
if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegOpenKeyEx(hServer, PARTNERROOT, 0, KEY_ALL_ACCESS, &hPartner); if( Status isnot NO_ERROR ) goto ErrorReturn;
if (!CheckValidPgOp(hPartner, fGrata)) { DisplayMessage(g_hModule, fGrata ? EMSG_SRVR_PG_INVALIDOP : EMSG_SRVR_PNG_INVALIDOP); Status = ERROR_INVALID_PARAMETER; goto CommonReturn; }
if (fGrata) pwszTempKey = WinsOemToUnicode(WINSCNF_PERSONA_GRATA_NM, NULL); else pwszTempKey = WinsOemToUnicode(WINSCNF_PERSONA_NON_GRATA_NM, NULL);
if( pwszTempKey is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
Status = RegQueryValueEx(hPartner, pwszTempKey, NULL, &dwType, pbByte, &dwSize);
WinsFreeMemory(pwszTempKey); pwszTempKey = NULL;
if( Status isnot NO_ERROR and Status isnot 2 ) goto ErrorReturn;
if( dwSize > 7 ) { LPWSTR pwszPng = NULL; pbByte = WinsAllocateMemory(dwSize+2);
dwSize+=2;
if( pbByte is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } Status = RegQueryValueEx(hPartner, fGrata ? PGSERVER : PNGSERVER, NULL, &dwType, pbByte, &dwSize);
if( Status isnot NO_ERROR ) goto ErrorReturn;
pwszPng = (LPWSTR)pbByte;
pwszPngIp = WinsAllocateMemory(dwSize);
if( pwszPngIp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
memcpy(pwszPngIp, pbByte, dwSize);
for( i=0; i<(dwSize+2)/sizeof(WCHAR); i++ ) { if( pwszPng[i] is L'\0' and pwszPng[i+1] isnot L'\0') { pwszPng[i] = L','; i++; } }
dwPngIp = 0; pTemp = wcstok(pwszPng, L","); while(pTemp isnot NULL ) { LPDWORD pdwTemp = pdwPngIp; dwPngIp++; dwLenCount += wcslen(pTemp); pdwPngIp = WinsAllocateMemory(dwPngIp*sizeof(DWORD));
if( pdwPngIp is NULL ) { WinsFreeMemory(pdwTemp); pdwTemp = NULL; Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } if( pdwTemp isnot NULL ) { memcpy(pdwPngIp, pdwTemp, (dwPngIp-1)*sizeof(DWORD)); WinsFreeMemory(pdwTemp); pdwTemp = NULL; } pdwPngIp[dwPngIp-1] = StringToIpAddress(pTemp);
pTemp = wcstok(NULL, L","); dwLenCount++;
} }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, &dwTagType, &dwTagNum);
if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
for( j=0; j<dwTagCount; j++ ) { switch(dwTagType) { //Server Ip Address List
case 0: { DWORD dwIpLen = 0; LPWSTR pwszIps = NULL; dwIpLen = wcslen(ppwcArguments[dwCurrentIndex+dwTagNum]);
if( ppwcArguments[dwCurrentIndex+dwTagNum][0] isnot L'{' or ppwcArguments[dwCurrentIndex+dwTagNum][dwIpLen-1] isnot L'}') { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } ppwcArguments[dwCurrentIndex+dwTagNum][dwIpLen-1] =L'\0';
dwIpLen--;
pwszIps = WinsAllocateMemory((dwIpLen)*sizeof(WCHAR));
if( pwszIps is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } memcpy(pwszIps, ppwcArguments[dwCurrentIndex+dwTagNum]+1, (dwIpLen-1)*sizeof(WCHAR)); i=0; pTemp = wcstok(pwszIps, L","); while(pTemp isnot NULL) { BOOL fPresent = FALSE; DWORD dw = 0;
if( IsIpAddress(pTemp) is FALSE ) { DisplayMessage(g_hModule, EMSG_SRVR_IP_DISCARD, pTemp); pTemp = wcstok(NULL, L","); continue; } else { dw = StringToIpAddress(pTemp);
for( j=0; j<dwPngIp; j++ ) { if( dw is INADDR_NONE ) { continue; } if( pdwPngIp[j] is dw ) { fPresent = TRUE; break; } else { continue; } } if( fPresent is FALSE ) { LPDWORD pdwTemp = pdwPngIp;
pdwPngIp = WinsAllocateMemory((dwPngIp+1)*sizeof(DWORD)); if( pdwTemp ) { memcpy(pdwPngIp, pdwTemp, dwPngIp*sizeof(DWORD)); WinsFreeMemory(pdwTemp); pdwTemp = NULL; }
pdwPngIp[dwPngIp] = dw; dwPngIp++; dwTotal++; } else { DisplayMessage(g_hModule, EMSG_SRVR_DUPLICATE_DISCARD, pTemp); } } pTemp = wcstok(NULL, L","); }
if( pwszIps ) { WinsFreeMemory(pwszIps); pwszIps = NULL; } break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } } if( pwszPngIp ) { WinsFreeMemory(pwszPngIp); pwszPngIp = NULL; }
pwszPngIp = WinsAllocateMemory((dwPngIp*(MAX_IP_STRING_LEN+1)+2)*sizeof(WCHAR));
if( pwszPngIp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } dwTemp = 0;
for( i=0; i<dwPngIp; i++ ) { LPWSTR pwIp = IpAddressToString(pdwPngIp[i]); if( pwIp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } wcscpy(pwszPngIp+dwTemp, pwIp); dwTemp+= wcslen(pwIp);
dwTemp++; WinsFreeMemory(pwIp); pwIp = NULL; } pwszPngIp[dwTemp] = L'\0'; pwszPngIp[dwTemp+1] = L'\0';
Status = RegSetValueEx(hPartner, fGrata? PGSERVER : PNGSERVER, 0, REG_MULTI_SZ, (LPBYTE)pwszPngIp, (dwTemp+1)*sizeof(WCHAR));
if( Status isnot NO_ERROR ) goto ErrorReturn;
CommonReturn: if( Status is NO_ERROR ) { if( dwTotal is 0 ) { DisplayMessage(g_hModule, fGrata ? EMSG_SRVR_NO_IP_ADDED_PG : EMSG_SRVR_NO_IP_ADDED_PNG); } DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); } if( hPartner ) { RegCloseKey(hPartner); hPartner = NULL; }
if( hServer ) { RegCloseKey(hServer); hServer = NULL; }
if( pwszPngIp ) { WinsFreeMemory(pwszPngIp); pwszPngIp = NULL; }
if( pdwPngIp ) { WinsFreeMemory(pdwPngIp); pdwPngIp = NULL; }
if( pbByte ) { WinsFreeMemory(pbByte); pbByte = NULL; }
return Status;ErrorReturn: DisplayErrorMessage(fGrata ? EMSG_SRVR_ADD_PGSERVER : EMSG_SRVR_ADD_PNGSERVER, Status); goto CommonReturn;
}
DWORDHandleSrvrCheckDatabase( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Checks the consistency of the databaseArguments : No argumentsReturn Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount; TAG_TYPE pttTags[] = { {WINS_TOKEN_ALL, FALSE, FALSE}, {WINS_TOKEN_FORCE, FALSE, FALSE}, };
BOOL fAll = FALSE, fForce = FALSE; LPDWORD pdwTagType = NULL, pdwTagNum = NULL;
WINSINTF_SCV_REQ_T ScvReq;
if( dwArgCount > dwCurrentIndex ) { //Start processing the arguments
dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum); if( Status isnot NO_ERROR ) goto ErrorReturn;
for( j=0; j<dwTagCount; j++ ) { switch(pdwTagType[j]) { //Consistency check all or those older than verify interval
case 0: { DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 0); if( dw == 1 ) { fAll = TRUE; } else if( dw == 0 ) { fAll = FALSE; } else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } break; } //Override wins checking in overloaded condition
case 1 : { DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 0); if( dw == 1 ) { fForce = TRUE; } else if( dw == 0 ) { fForce = FALSE; } else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } } } if( fAll ) ScvReq.Age = 0; // check all the replicas
else ScvReq.Age = 1;
if( fForce ) ScvReq.fForce = TRUE; else ScvReq.fForce = FALSE;
ScvReq.Opcode_e = WINSINTF_E_SCV_VERIFY;
Status = WinsDoScavengingNew(g_hBind, &ScvReq);
if( Status isnot NO_ERROR ) goto ErrorReturn; DisplayMessage(g_hModule, MSG_WINS_COMMAND_QUEUED);
CommonReturn: if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; } if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; } return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_CHECK_DATABASE, Status); goto CommonReturn;}
DWORDHandleSrvrCheckName( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Checks a list of names against a list of WINS serversArguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory, either a list of names or a file which contains the list of names. Names should be in the format (name)*(16th char) and either a list of server IP addresses, separated by commas and enclosed by {} or a file that contains the list of ip address in comma seperated format. Optional, to include all partners in the server list.Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount; TAG_TYPE pttTags[] = { {WINS_TOKEN_NAMELIST, FALSE, FALSE}, {WINS_TOKEN_NAMEFILE, FALSE, FALSE}, {WINS_TOKEN_SERVERLIST, FALSE, FALSE}, {WINS_TOKEN_SERVERFILE, FALSE, FALSE}, {WINS_TOKEN_INCLPARTNER, FALSE, FALSE}, }; BOOL fNameFile = FALSE, fServerFile = FALSE; LPWSTR pwszNameFile = NULL, pwszServerFile = NULL; PDWORD pdwTagType = NULL, pdwTagNum = NULL; PDWORD pdwIpAddress = NULL; LPWSTR *ppNames = NULL; BOOL fInclPartner = FALSE, fIpEmpty = TRUE;
//Need at least a list of names either directly or thro' file and
//a list of server Ip either direcly or thro' file.
if( dwArgCount < dwCurrentIndex + 2 ) { DisplayMessage(g_hModule, HLP_SRVR_CHECK_NAME_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
//Start processing the arguments
dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum); if( Status isnot NO_ERROR ) goto ErrorReturn;
if( ( pttTags[0].bPresent is TRUE and pttTags[1].bPresent is TRUE ) or ( pttTags[2].bPresent is TRUE and pttTags[3].bPresent is TRUE ) ) { Status = ERROR_INVALID_PARAMETER_SPECIFICATION; goto ErrorReturn; } WinServers = WinsAllocateMemory(MAX_SERVERS*sizeof(*WinServers));
if( WinServers is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
NBNames = (u_char ** )WinsAllocateMemory(MAX_NB_NAMES*sizeof(u_char*));
if( NBNames is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
for( j=0; j<dwTagCount; j++ ) { switch(pdwTagType[j]) { case 0://NameList
{ LPWSTR pwszToken = L",\r\n\t "; LPWSTR pwszName = NULL; LPWSTR pwszTemp = NULL; int ilen = 0; LPWSTR pTemp = NULL; BOOL fPresent = FALSE; DWORD dw = 0, dwType = 0; DWORD dwLen = wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]);
if( dwLen < 2 ) { NumNBNames = 0; break; } if( ppwcArguments[dwCurrentIndex+pdwTagNum[j]][0] isnot L'{' or ppwcArguments[dwCurrentIndex+pdwTagNum[j]][dwLen-1] isnot L'}' ) { NumNBNames = 0; break; }
pwszTemp = WinsAllocateMemory((dwLen)*sizeof(WCHAR));
if(pwszTemp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } for( ilen=0; ilen<NumNBNames; ilen++ ) { if( NBNames[ilen] isnot NULL ) { WinsFreeMemory(NBNames[ilen]); NBNames[ilen] = NULL; } } NumNBNames = 0;
wcscpy(pwszTemp, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]+1); pwszTemp[dwLen-2] = L'\0'; pwszName = wcstok(pwszTemp, pwszToken); while(pwszName isnot NULL ) { dw = wcslen(pwszName); if( dw < 1 or dw > 18 ) { DisplayMessage(g_hModule, EMSG_WINS_INVALID_NAME, pwszName);
} else { pTemp = wcsstr(pwszName, L"*"); if( pTemp is NULL ) { DisplayMessage(g_hModule, EMSG_WINS_INVALID_NAME, pwszName); } else { CHAR chEnd = 0x00; dw = (DWORD)(pTemp - pwszName + 1); if( dw > 16 ) { DisplayMessage(g_hModule, EMSG_SRVR_INVALID_NETBIOS_NAME); goto ErrorReturn; } pwszName[dw-1] = L'\0'; chEnd = StringToHexA(pTemp+1);
if( dwType > 255 ) { DisplayMessage(g_hModule, EMSG_WINS_VALUE_OUTOFRANGE); } else { LPWSTR pwcTemp = WinsAllocateMemory((NBT_NONCODED_NMSZ+1)*sizeof(WCHAR)); LPWSTR pwTemp = NULL; if( pwcTemp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; if( pwszTemp isnot NULL ) { WinsFreeMemory(pwszTemp ); pwszTemp = NULL; } goto ErrorReturn; } wcscpy(pwcTemp, pwszName); wcsncat(pwcTemp, L" ", (16 - wcslen(pwszName))); pwcTemp[15] = chEnd; for( ilen=0; ilen<NumNBNames; ilen++) { pwTemp = WinsOemToUnicode(NBNames[ilen], NULL); if( pwTemp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } if(_wcsicmp(pwTemp, pwcTemp) is 0 ) { fPresent = TRUE; break; } WinsFreeMemory(pwTemp); pwTemp = NULL; } if( pwTemp ) { WinsFreeMemory(pwTemp); pwTemp = NULL; } if( fPresent is FALSE ) { LPSTR pcTemp = NULL;
NBNames[NumNBNames] = WinsAllocateMemory(17); if( NBNames[NumNBNames] is NULL ) { if( pwszTemp isnot NULL ) { WinsFreeMemory(pwszTemp ); pwszTemp = NULL; } Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } pcTemp = WinsUnicodeToOem(pwcTemp, NULL); if( pcTemp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } strncpy(NBNames[NumNBNames], pcTemp, 16); NBNames[NumNBNames][15] = (CHAR)chEnd; NBNames[NumNBNames][16] = '\0'; NumNBNames++; WinsFreeMemory(pcTemp); pcTemp = NULL; } if( pwcTemp ) { WinsFreeMemory(pwcTemp); pwcTemp = NULL; }
} } } pwszName = wcstok(NULL, pwszToken);
} if( pwszTemp ) { WinsFreeMemory(pwszTemp); pwszTemp = NULL; }
break; } case 1://NameFile
{ HANDLE hFile = NULL; DWORD dwLen = wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]), dwFileSize = 0, dw = 0, dwType = 0, dwBytesRead = 0; BOOL fPresent = FALSE; int ilen = 0; LPBYTE pbFileData = NULL; LPSTR pszToken = " ,\r\n\t", pszData = NULL, pTemp = NULL, pszName = NULL, pszTemp = NULL; LPWSTR pwszTempName = NULL;
if( dwLen < 1 ) { DisplayMessage(g_hModule, EMSG_WINS_INVALID_FILENAME, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); goto CommonReturn; }
hFile = CreateFile(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], GENERIC_READ | GENERIC_WRITE, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
if( hFile is INVALID_HANDLE_VALUE ) { Status = GetLastError(); goto ErrorReturn; }
dwFileSize = GetFileSize(hFile, NULL);
if( dwFileSize is 0 ) { DisplayMessage(g_hModule, EMSG_WINS_EMPTY_FILE, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); CloseHandle(hFile); hFile = NULL; goto CommonReturn; }
pbFileData = WinsAllocateMemory(dwFileSize+1);
if( pbFileData is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; CloseHandle(hFile); hFile = NULL; goto ErrorReturn; }
if( !ReadFile( hFile, pbFileData, dwFileSize, &dwBytesRead, NULL) ) { DisplayMessage(g_hModule, EMSG_WINS_FILEREAD_FAILED); CloseHandle(hFile); hFile = NULL; goto CommonReturn; } CloseHandle(hFile); hFile = NULL;
for( ilen=0; ilen<NumNBNames; ilen++ ) { if( NBNames[ilen] isnot NULL ) { WinsFreeMemory(NBNames[ilen]); NBNames[ilen] = NULL; } } NumNBNames = 0;
pszData = (LPSTR)pbFileData; pszName = strtok(pszData, pszToken); while( pszName isnot NULL ) { dw = strlen(pszName); if( dw < 1 or dw > 18 ) { pwszTempName = WinsOemToUnicode(pszName, NULL); if( pwszTempName is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } DisplayMessage(g_hModule, EMSG_WINS_INVALID_NAME, pwszTempName); WinsFreeMemory(pwszTempName); pwszTempName = NULL; } else { WCHAR wcEnd[2] = {L'\0'}; CHAR cEnd[2] = {L'\0'};
pTemp = strstr(pszName, "*"); if( pTemp is NULL ) { pwszTempName = WinsOemToUnicode(pszName, NULL); if( pwszTempName is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } DisplayMessage(g_hModule, EMSG_WINS_INVALID_NAME, pwszTempName); WinsFreeMemory(pwszTempName); pwszTempName = NULL; } else { CHAR chEndChar = 0x00; LPWSTR pwszTempBuf = NULL; dw = (DWORD)(pTemp - pszName + 1); if( dw > 16 ) { DisplayMessage(g_hModule, EMSG_SRVR_INVALID_NETBIOS_NAME); goto ErrorReturn; } pszName[dw-1] = L'\0'; pwszTempBuf = WinsOemToUnicode(pTemp+1, NULL);
if( pwszTempBuf is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
chEndChar = StringToHexA(pwszTempBuf); WinsFreeMemory(pwszTempBuf); pwszTempBuf = NULL;
if( dwType > 255 ) { DisplayMessage(g_hModule, EMSG_WINS_VALUE_OUTOFRANGE); } else { LPSTR pcTemp = WinsAllocateMemory(NBT_NONCODED_NMSZ+1); if( pcTemp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; if( pbFileData isnot NULL ) { WinsFreeMemory(pbFileData ); pbFileData = NULL; } goto ErrorReturn; } strcpy(pcTemp, pszName); strncat(pcTemp, " ", (16-strlen(pszName))); pcTemp[15] = chEndChar;
for( ilen=0; ilen<NumNBNames; ilen++) { if(_stricmp(NBNames[ilen], pcTemp) is 0 ) { fPresent = TRUE; break; } } if( fPresent is FALSE ) { LPSTR pszOem = NULL; NBNames[NumNBNames] = WinsAllocateMemory(17); if( NBNames[NumNBNames] is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } pszOem = WinsAnsiToOem(pcTemp); if( pszOem is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } strncpy(NBNames[NumNBNames], pszOem, 16); WinsFreeMemory(pszOem); pszOem = NULL;
NBNames[NumNBNames][15] = (CHAR)chEndChar; NumNBNames++; } if( pcTemp ) { WinsFreeMemory(pcTemp); pcTemp = NULL; } } } } pszName = strtok(NULL, pszToken); } if( pbFileData ) { WinsFreeMemory(pbFileData); pbFileData = NULL; } break; } case 2://ServerList
{ LPWSTR pwszToken = L",\r\n\t ", pwszName = NULL, pwszTemp = NULL; int ilen = 0; BOOL fPresent = FALSE; DWORD dwLen = wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]);
if( dwLen < 2 ) { NumWinServers = 0; break; } if( ppwcArguments[dwCurrentIndex+pdwTagNum[j]][0] isnot L'{' or ppwcArguments[dwCurrentIndex+pdwTagNum[j]][dwLen-1] isnot L'}' ) { NumWinServers = 0; break; }
pwszTemp = WinsAllocateMemory((dwLen)*sizeof(WCHAR));
if(pwszTemp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } if( fInclPartner is FALSE ) { memset(WinServers, 0x00, MAX_SERVERS*sizeof(WINSERVERS));
NumWinServers = 0; }
wcscpy(pwszTemp, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]+1); pwszTemp[dwLen-2] = L'\0'; pwszName = wcstok(pwszTemp, pwszToken); while(pwszName isnot NULL ) { fPresent = FALSE; if( IsIpAddress(pwszName) is FALSE ) { DisplayMessage(g_hModule, EMSG_WINS_INVALID_IPADDRESS, pwszName); } else { struct in_addr Temp; LPSTR pszTempAddr = NULL;
pszTempAddr = WinsUnicodeToAnsi(pwszName, NULL); if( pszTempAddr is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } Temp.s_addr = inet_addr(pszTempAddr); WinsFreeMemory(pszTempAddr); pszTempAddr = NULL;
if( Temp.s_addr isnot INADDR_NONE ) { for( ilen=0; ilen<NumWinServers; ilen++) { if( WinServers[ilen].Server.s_addr is Temp.s_addr ) { fPresent = TRUE; break; } } fIpEmpty = FALSE;
if( fPresent is FALSE ) { WinServers[NumWinServers].Server.s_addr = Temp.s_addr; NumWinServers++; } } } pwszName = wcstok(NULL, pwszToken); } break; } case 3://ServerFile
{ HANDLE hFile = NULL; DWORD dwLen = wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]), dwFileSize = 0, dwBytesRead = 0; BOOL fPresent = FALSE; int ilen = 0; LPBYTE pbFileData = NULL; LPSTR pszToken = " ,\r\n\t", pszData = NULL, pszName = NULL; LPWSTR pwszToken = L" ,\r\n\t", pwszName = NULL, pwszData = NULL; if( dwLen < 1 ) { DisplayMessage(g_hModule, EMSG_WINS_INVALID_FILENAME, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); goto CommonReturn; }
hFile = CreateFile(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], GENERIC_READ | GENERIC_WRITE, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
if( hFile is INVALID_HANDLE_VALUE ) { Status = GetLastError(); goto ErrorReturn; }
dwFileSize = GetFileSize(hFile, NULL);
if( dwFileSize is 0 ) { DisplayMessage(g_hModule, EMSG_WINS_EMPTY_FILE, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); CloseHandle(hFile); hFile = NULL; goto CommonReturn; }
pbFileData = WinsAllocateMemory(dwFileSize+1);
if( pbFileData is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; CloseHandle(hFile); hFile = NULL; goto ErrorReturn; }
if( !ReadFile( hFile, pbFileData, dwFileSize, &dwBytesRead, NULL) ) { CloseHandle(hFile); hFile = NULL; DisplayMessage(g_hModule, EMSG_WINS_FILEREAD_FAILED); goto CommonReturn; } CloseHandle(hFile); hFile = NULL; if( fInclPartner is FALSE ) { memset(WinServers, 0x00, MAX_SERVERS*sizeof(WINSERVERS));
NumWinServers = 0; }
pszData = (LPSTR)pbFileData; pwszData = WinsOemToUnicode(pszData, NULL); if( pwszData is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } pwszName = wcstok(pwszData, pwszToken); while( pwszName isnot NULL ) { fPresent = FALSE; pszName = WinsUnicodeToOem(pwszName, NULL); if( pszName is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
if( IsIpAddress(pwszName) is FALSE ) { DisplayMessage(g_hModule, EMSG_WINS_INVALID_IPADDRESS, pwszName); } else { struct in_addr Temp;
Temp.s_addr = inet_addr(pszName);
if( Temp.s_addr isnot INADDR_NONE ) { for( ilen=0; ilen<NumWinServers; ilen++) { if( WinServers[ilen].Server.s_addr is Temp.s_addr ) { fPresent = TRUE; break; } } fIpEmpty = FALSE; if( fPresent is FALSE ) { WinServers[NumWinServers].Server.s_addr = Temp.s_addr; NumWinServers++; } } } pwszName = wcstok(NULL, pwszToken); if( pszName ) { WinsFreeMemory(pszName); pszName = NULL; } }
if( pbFileData ) { WinsFreeMemory(pbFileData); pbFileData = NULL; } if( pwszData ) { WinsFreeMemory(pwszData); pwszData = NULL; }
break; } case 4://IncludePartners
{ HKEY hServer = NULL, hPull = NULL, hPush = NULL; LPWSTR pTemp = NULL; WCHAR wcKey[MAX_IP_STRING_LEN+1] = {L'\0'}; DWORD dw = 0, dwLen = MAX_IP_STRING_LEN+1, dwKeys = 0; if(wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) isnot 1 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
if( ppwcArguments[dwCurrentIndex+pdwTagNum[j]][0] isnot L'Y' and ppwcArguments[dwCurrentIndex+pdwTagNum[j]][0] isnot L'y' ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
if( wcslen(g_ServerNetBiosName) < 2 ) { pTemp = NULL; }
Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer ); if( Status isnot NO_ERROR ) { Status = NO_ERROR; break; } while(TRUE) { Status = RegOpenKeyEx(hServer, PULLROOT, 0, KEY_ALL_ACCESS, &hPull);
if( Status isnot NO_ERROR ) { Status = NO_ERROR; break; }
Status = RegQueryInfoKey(hPull, NULL, NULL, NULL, &dwKeys, NULL, NULL, NULL, NULL, NULL, NULL, NULL); if( Status isnot NO_ERROR ) { Status = NO_ERROR; break; }
if( dwKeys < 1 ) break;
for( dw=0; dw<dwKeys; dw++ ) { DWORD i = 0; LPSTR pszTempAddr = NULL; BOOL fPresent = FALSE; struct in_addr Temp;
dwLen = MAX_IP_STRING_LEN+1;
memset(wcKey, 0x00, (MAX_IP_STRING_LEN+1)*sizeof(WCHAR));
Status = RegEnumKeyEx(hPull, dw, wcKey, &dwLen, NULL, NULL, NULL, NULL);
if( Status isnot NO_ERROR ) continue;
pszTempAddr = WinsUnicodeToOem(wcKey, NULL); if( pszTempAddr is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } Temp.s_addr = inet_addr(pszTempAddr);
WinsFreeMemory(pszTempAddr); pszTempAddr = NULL;
if( Temp.s_addr isnot INADDR_NONE ) { fPresent = FALSE; if( MAX_SERVERS > NumWinServers ) { for( i=0; i<(DWORD)NumWinServers; i++) { if( WinServers[i].Server.s_addr is Temp.s_addr ) { fPresent = TRUE; break; } } if( fPresent is FALSE ) { WinServers[NumWinServers].Server.s_addr = Temp.s_addr; NumWinServers++; }
} else break; }
}
}
if( hPull ) { RegCloseKey(hPull); hPull = NULL; }
dw = dwKeys = 0;
while(TRUE) { Status = RegOpenKeyEx(hServer, PUSHROOT, 0, KEY_ALL_ACCESS, &hPush);
if( Status isnot NO_ERROR ) { Status = NO_ERROR; break; }
Status = RegQueryInfoKey(hPush, NULL, NULL, NULL, &dwKeys, NULL, NULL, NULL, NULL, NULL, NULL, NULL); if( Status isnot NO_ERROR ) { Status = NO_ERROR; break; }
if( dwKeys < 1 ) break;
for( dw=0; dw<dwKeys; dw++ ) { DWORD i = 0; BOOL fPresent = FALSE; LPSTR pszTempAddr = NULL; struct in_addr Temp;
dwLen = MAX_IP_STRING_LEN+1;
memset(wcKey, 0x00, (MAX_IP_STRING_LEN+1)*sizeof(WCHAR));
Status = RegEnumKeyEx(hPush, dw, wcKey, &dwLen, NULL, NULL, NULL, NULL);
if( Status isnot NO_ERROR ) continue;
pszTempAddr = WinsUnicodeToOem(wcKey, NULL); if( pszTempAddr is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
Temp.s_addr = inet_addr(pszTempAddr); WinsFreeMemory(pszTempAddr); pszTempAddr = NULL;
if( Temp.s_addr isnot INADDR_NONE ) { if( MAX_SERVERS > NumWinServers ) { for( i=0; i<(DWORD)NumWinServers; i++) { if( WinServers[i].Server.s_addr is Temp.s_addr ) { fPresent = TRUE; break; } } if( fPresent is FALSE ) { WinServers[NumWinServers].Server.s_addr = Temp.s_addr; NumWinServers++; }
} else break; }
}
} if( hPush ) { RegCloseKey(hPush); hPush = NULL; }
if( hServer ) { RegCloseKey(hServer); hServer = NULL; } break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } }
if( NumNBNames is 0 ) { DisplayMessage(g_hModule, EMSG_WINS_NO_NAMES); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
if( NumWinServers is 0 ) { DisplayMessage(g_hModule, EMSG_WINS_NO_SERVERS); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
if( fIpEmpty is TRUE ) { DisplayMessage(g_hModule, EMSG_WINS_NO_SERVERS); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } else { CheckNameConsistency(); }
if( Status is NO_ERROR ) DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS);
CommonReturn:
if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; }
if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; }
if( pdwIpAddress ) { WinsFreeMemory(pdwIpAddress); pdwIpAddress = NULL; }
if( ppNames ) { WinsFreeMemory(ppNames); ppNames = NULL; }
if( WinServers ) { WinsFreeMemory(WinServers); WinServers = NULL; }
if( NBNames ) { WinsFreeMemory(NBNames); NBNames = NULL; } return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_CHECK_NAME, Status); goto CommonReturn;}
DWORDHandleSrvrCheckVersion( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Checks the version number consistencies for the recordsArguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory, IP address of the server to start with. Optional, a File Name to store the output in proper format.Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount = 0; TAG_TYPE pttTags[] = { {WINS_TOKEN_SERVER, TRUE, FALSE}, {WINS_TOKEN_FILE, FALSE, FALSE}, };
PDWORD pdwTagNum = NULL, pdwTagType = NULL; BOOL fFile = FALSE; FILE *pFile = NULL; LPSTR pStartIp = NULL;
//Must provide the IP address of the server to start with
if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, HLP_SRVR_CHECK_VERSION_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn;
}
//Start processing the arguements based on the
dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum);
if( Status isnot NO_ERROR ) { goto ErrorReturn; }
for( j=0; j<dwTagCount; j++ ) { switch(pdwTagType[j]) { //Ip Address of the server to start with
case 0: { WCHAR wcServerIpAdd[MAX_IP_STRING_LEN+1] = {L'\0'}; if( IsIpAddress( ppwcArguments[dwCurrentIndex+pdwTagNum[j]] ) is FALSE ) { Status = ERROR_INVALID_IPADDRESS; goto ErrorReturn; }
wcscpy(wcServerIpAdd, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]);
pStartIp = WinsUnicodeToOem(wcServerIpAdd, NULL);
if( pStartIp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } break;
} //File to store the output data
case 1: { WCHAR wcFile[MAX_PATH] = {L'\0'}; if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) < 1 or wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) > MAX_PATH ) { wcscpy(wcFile, L"wins.rec"); } else { wcscpy(wcFile, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); }
pFile = _wfopen(wcFile, L"w+");
if( pFile is NULL ) { pFile = _wfopen(L"wins.rec", L"w+"); fFile = TRUE; } else { fFile = TRUE; } break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } }
CheckVersionNumbers( pStartIp, fFile, pFile);
CommonReturn: if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; }
if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; } if( pFile ) { fclose(pFile); pFile = NULL; } if( pStartIp) { WinsFreeMemory(pStartIp); pStartIp = NULL; }
return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_CHECK_VERSION, Status); goto CommonReturn;}
DWORDHandleSrvrDeleteName( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Deletes an record entry for the WINS server databaseArguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory, Record Name, Endchar Optional, ScopeReturn Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount = 0; TAG_TYPE pttTags[] = { {WINS_TOKEN_NAME, TRUE, FALSE}, {WINS_TOKEN_ENDCHAR, TRUE, FALSE}, {WINS_TOKEN_SCOPE, FALSE, FALSE} }; PDWORD pdwTagNum = NULL, pdwTagType = NULL;
WCHAR wszName[MAX_STRING_LEN] = {L'\0'}; BOOL fEndChar = FALSE;
CHAR ch16thChar = 0x00; BOOL fScope = FALSE; WCHAR wszScope[MAX_STRING_LEN] = {L'\0'}; LPSTR pszTempName = NULL; DWORD dwTempNameLen = 0; DWORD dwStrLen = 0; LPWSTR pwszTemp = NULL; WINSINTF_RECORD_ACTION_T RecAction = {0}; PWINSINTF_RECORD_ACTION_T pRecAction = NULL;
memset(&RecAction, 0x00, sizeof(WINSINTF_RECORD_ACTION_T)); RecAction.fStatic = FALSE;
//Must provide at least the record name and endchar
if( dwArgCount < dwCurrentIndex + 2 ) { DisplayMessage(g_hModule, HLP_SRVR_DELETE_NAME_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
//Start processing the arguments based on ppwcArguments, dwArgCount and dwCurrentIndex
dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum); if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
i = dwTagCount; for( j=0; j<i; j++ ) { switch(pdwTagType[j]) { //Record Name
case 0: { DWORD dwLen = wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); if( dwLen is 0 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dwLen = ( 255 < dwLen ) ? 255 : dwLen; memcpy(wszName, ppwcArguments[dwCurrentIndex+pdwTagNum[j]], dwLen*sizeof(WCHAR)); wszName[dwLen] = L'\0'; break; } //Endchar
case 1: { DWORD dwLen = 0, k=0; fEndChar = TRUE; dwLen = wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); if( dwLen > 2 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } for( k=0; k<dwLen; k++ ) { WCHAR wc = ppwcArguments[dwCurrentIndex+pdwTagNum[j]][k]; if( towlower(wc) >= L'a' and towlower(wc) <= L'z' ) { if( towlower(wc) > L'f' ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } }
ch16thChar = StringToHexA(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]);
break; } //Scope
case 2: { DWORD dwLen = wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); fScope = TRUE; if( dwLen is 0 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dwLen = ( 255 < dwLen ) ? 255 : dwLen; memcpy(wszScope, ppwcArguments[dwCurrentIndex+pdwTagNum[j]], dwLen*sizeof(WCHAR)); wszScope[dwLen] = L'\0'; break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } } _wcsupr(wszName); _wcsupr(wszScope);
pszTempName = WinsUnicodeToOem(wszName, NULL);
if( pszTempName is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwStrLen = strlen(pszTempName); if( dwStrLen >= 16 ) { DisplayMessage(g_hModule, EMSG_SRVR_INVALID_NETBIOS_NAME); Status = ERROR_INVALID_PARAMETER; WinsFreeMemory(pszTempName); pszTempName = NULL; goto ErrorReturn; } RecAction.pName = WinsAllocateMemory(273);
if( RecAction.pName is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
strncpy(RecAction.pName, pszTempName, dwStrLen); WinsFreeMemory(pszTempName); pszTempName = NULL;
for( i=dwStrLen; i < 16; i++ ) { RecAction.pName[i] = ' '; }
if( fEndChar ) { if( ch16thChar is 0x00 ) { RecAction.pName[15] = 0x00; } else { RecAction.pName[15] = ch16thChar; }
} RecAction.pName[16] = '\0'; dwStrLen = 16;
if( fScope ) { DWORD dwLen = 0;
RecAction.pName[dwStrLen] = '.'; pszTempName = WinsUnicodeToOem(wszScope, NULL); if( pszTempName is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } dwLen = strlen(pszTempName); dwLen = ( 255 - dwStrLen < dwLen ) ? 255 - dwStrLen : dwLen;
strncpy(RecAction.pName + dwStrLen + 1, pszTempName, dwLen); WinsFreeMemory(pszTempName); pszTempName = NULL; RecAction.pName[dwLen + dwStrLen + 1] = '\0'; if( fEndChar and ch16thChar is 0x00 ) dwStrLen = 16+dwLen+1; else dwStrLen = strlen(RecAction.pName); } else {
RecAction.pName[dwStrLen] = '\0'; }
RecAction.NameLen = dwStrLen;
RecAction.Cmd_e = WINSINTF_E_QUERY;
RecAction.OwnerId = StringToIpAddress(g_ServerIpAddressUnicodeString); RecAction.NameLen = dwStrLen; pRecAction = &RecAction;
Status = WinsRecordAction(g_hBind, &pRecAction);
if( Status isnot NO_ERROR ) goto ErrorReturn;
RecAction.Cmd_e = WINSINTF_E_DELETE; RecAction.State_e = WINSINTF_E_DELETED;
pRecAction = &RecAction;
Status = WinsRecordAction(g_hBind, &pRecAction); if( Status isnot NO_ERROR ) goto ErrorReturn;
CommonReturn: if( Status is NO_ERROR ) DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS);
if( pszTempName ) { WinsFreeMemory(pszTempName); pszTempName = NULL; }
if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; } if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; } if( RecAction.pName ) { WinsFreeMemory(RecAction.pName); RecAction.pName = NULL; } return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_DELETE_NAME, Status); goto CommonReturn;
return NO_ERROR;}
DWORDHandleSrvrDeletePartner( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Delete a partner from the WINS serverArguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : Confirmation Optional : Server IP and Type. Note : If no ip is provided, it deletes all partners in the list. Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount; PDWORD pdwTagType = NULL, pdwTagNum = NULL; TAG_TYPE pttTags[] = { {WINS_TOKEN_SERVER, FALSE, FALSE}, {WINS_TOKEN_TYPE, FALSE, FALSE}, {WINS_TOKEN_CONFIRM, TRUE, FALSE}, }; WCHAR wcServerIpAdd[MAX_IP_STRING_LEN + 1] = {L'\0'}; DWORD dwIpLen = (MAX_IP_STRING_LEN+1); BOOL fPull = TRUE, fPush = TRUE, fConfirm = FALSE;
HKEY hServer = NULL, hPartner = NULL;
LPWSTR pTemp = NULL;
//Must provide the confirmation in order for this to succeed.
if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, HLP_SRVR_DELETE_PARTNER_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } //Start processing the arguments based on ppwcArguments, dwCurrentIndex and dwArgCount
else { dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum); if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
for( j=0; j<dwTagCount; j++ ) { switch(pdwTagType[j]) { //Server Ip
case 0: { struct hostent * lpHostEnt = NULL; CHAR cAddr[16]; BYTE pbAdd[4]; char szAdd[4]; int k = 0, l=0; DWORD dwLen, nLen = 0; CHAR *pTemp = NULL; CHAR *pNetBios = NULL; if( IsIpAddress( ppwcArguments[dwCurrentIndex+pdwTagNum[j]] ) is FALSE ) { Status = ERROR_INVALID_IPADDRESS; goto ErrorReturn; }
wcscpy(wcServerIpAdd, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); break;
} //Partner Type
case 1: { DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 0); switch(dw) { case 0: { fPull = TRUE; fPush = FALSE; break; } case 1: { fPull = FALSE; fPush = TRUE; break; } case 2: { fPull = TRUE; fPush = TRUE; break; } default: { Status = ERROR_INVALID_PARAMETER; break; } } break; } //Confirmation
case 2 : { if( 0 is _wcsnicmp(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], L"Y", 1) ) fConfirm = TRUE; else if ( 0 is _wcsnicmp(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], L"N", 1) ) fConfirm = FALSE; else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } break;
} default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } } }
if( fConfirm is FALSE ) { DisplayMessage(g_hModule, MSG_WINS_CONFIRMATION_DENIED); goto CommonReturn; }
{ if( wcslen(g_ServerNetBiosName) > 0 ) { pTemp = g_ServerNetBiosName; } Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer);
if( Status isnot ERROR_SUCCESS ) goto ErrorReturn;
//PullPartners
if( fPull ) { Status = RegOpenKeyEx(hServer, PULLROOT, 0, KEY_ALL_ACCESS, &hPartner); if( Status isnot NO_ERROR ) { if (Status == ERROR_FILE_NOT_FOUND) Status = NO_ERROR;
goto PUSH; } if( wcslen(wcServerIpAdd) < 3 )//not a valid Ip, delete all partners
{ while(TRUE) { Status = RegEnumKeyEx(hPartner, 0, wcServerIpAdd, &dwIpLen, NULL, NULL, NULL, NULL); if (Status != ERROR_SUCCESS) break; RegDeleteKey(hPartner, wcServerIpAdd); dwIpLen = (MAX_IP_STRING_LEN + 1)*sizeof(WCHAR); memset(wcServerIpAdd, L'\0', MAX_IP_STRING_LEN+1); } } else { RegDeleteKey(hPartner, wcServerIpAdd); }
if( hPartner ) { RegCloseKey(hPartner); hPartner = NULL; }
if( Status is ERROR_NO_MORE_ITEMS ) { Status = NO_ERROR; goto PUSH; } if( Status is ERROR_FILE_NOT_FOUND ) { DisplayMessage(g_hModule, EMSG_INVALID_PARTNER_NAME); if( fPush ) goto PUSH; else goto CommonReturn; } if( Status isnot NO_ERROR ) { if( fPush ) goto PUSH; else goto CommonReturn; }
} //Push Partner
PUSH: if( fPush ) { Status = RegOpenKeyEx(hServer, PUSHROOT, 0, KEY_ALL_ACCESS, &hPartner);
if (Status is ERROR_FILE_NOT_FOUND) { DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); goto CommonReturn; }
if( Status isnot NO_ERROR ) goto ErrorReturn; dwIpLen = (MAX_IP_STRING_LEN + 1); if( wcslen(wcServerIpAdd) < 3 )//not a valid Ip, delete all partners
{ while( TRUE ) { Status = RegEnumKeyEx(hPartner, 0, wcServerIpAdd, &dwIpLen, NULL, NULL, NULL, NULL); if (Status != ERROR_SUCCESS) break; RegDeleteKey(hPartner, wcServerIpAdd); dwIpLen = (MAX_IP_STRING_LEN + 1)*sizeof(WCHAR); } } else { RegDeleteKey(hPartner, wcServerIpAdd); }
if( Status is ERROR_NO_MORE_ITEMS ) { Status = NO_ERROR; }
if( Status is ERROR_FILE_NOT_FOUND ) { DisplayMessage(g_hModule, EMSG_INVALID_PARTNER_NAME); goto ErrorReturn; }
if( Status isnot NO_ERROR ) goto ErrorReturn; if( hPartner ) { RegCloseKey(hPartner); hPartner = NULL; } } }
if( Status is NO_ERROR ) DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); else goto ErrorReturn;
CommonReturn:
if( hPartner ) { RegCloseKey(hPartner); hPartner = NULL; }
if( hServer ) { RegCloseKey(hServer); hServer = NULL; }
if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; } if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; } return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_DELETE_PARTNER, Status); goto CommonReturn;}
DWORDHandleSrvrDeleteRecords( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Delete or Tombstone records from a WINS server based on the versionArguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : Minimum and Maximum version numbers( range of version ) to be deleted/tombstoned Optional : Operation - tombstone(default) or delete Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount = 0; PDWORD pdwTagType = NULL, pdwTagNum = NULL;
TAG_TYPE pttTags[] = { {WINS_TOKEN_MINVER, TRUE, FALSE}, {WINS_TOKEN_MAXVER, TRUE, FALSE}, {WINS_TOKEN_OP, FALSE, FALSE}, }; BOOL fDelete = TRUE; WINSINTF_VERS_NO_T MinVer, MaxVer; WINSINTF_ADD_T WinsAdd; LPWSTR pwszTemp = NULL;
//Needs at least both Min Ver and Max ver
if( dwArgCount < dwCurrentIndex + 2 ) { DisplayMessage(g_hModule, HLP_SRVR_DELETE_RECORDS_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
//Start processing the arguments based on ppwcArguments, dwCurrentIndex and dwArgCount
dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); if( pdwTagType is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); if( pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum);
if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
i = dwTagCount;
for( j=0; j<i; j++ ) { switch(pdwTagType[j]) { case 0://{high,low} format, Min version
{ Status = GetVersionData(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], &MinVer); if( Status isnot NO_ERROR ) goto ErrorReturn; break; } case 1://{high,low} format, Max version
{ Status = GetVersionData(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], &MaxVer); if( Status isnot NO_ERROR ) goto ErrorReturn; break; } case 2: //Operation 0 - delete 1 - Tombstone
{ DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 0); if( dw is 0 ) fDelete = TRUE; else if( dw is 1 ) fDelete = FALSE; else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } }
WinsAdd.IPAdd = StringToIpAddress(g_ServerIpAddressUnicodeString); WinsAdd.Len = 4; WinsAdd.Type = 0;
if( fDelete ) { Status = WinsDelDbRecs(g_hBind, &WinsAdd, MinVer, MaxVer); } else { Status = WinsTombstoneDbRecs(g_hBind, &WinsAdd, MinVer, MaxVer); } if( Status isnot NO_ERROR ) goto ErrorReturn;CommonReturn: if( Status is NO_ERROR ) DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS);
if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; } if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; }
return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_DELETE_RECORDS, Status); goto CommonReturn;}
DWORDHandleSrvrDeleteWins( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Delete a partner from the WINS serverArguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : Confirmation Optional : Server IP and Type. Note : If no ip is provided, it deletes all partners in the list. Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount; TAG_TYPE pttTags[] = { {WINS_TOKEN_SERVERS, TRUE, FALSE}, {WINS_TOKEN_OP, FALSE, FALSE}, }; DWORD dwIpCount = 0; LPDWORD pdwIp = NULL, pdwTagType = NULL, pdwTagNum = NULL; BOOL fDelete = FALSE;
WINSINTF_ADD_T WinsAdd; handle_t hBind; WINSINTF_BIND_DATA_T BindData; if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, HLP_SRVR_DELETE_WINS_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn;
}
dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum);
if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
for( j=0; j<dwTagCount; j++ ) { switch(pdwTagType[j]) { case 0: { LPWSTR pszIps = NULL; DWORD dwIpLen = 0; LPWSTR pTemp = NULL; DWORD dwIp = 0; dwIpCount = 0;
dwIpLen = wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); if( dwIpLen < 2 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } pszIps = WinsAllocateMemory((dwIpLen+1)*sizeof(WCHAR)); if( pszIps is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } wcscpy(pszIps, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); if( pszIps[0] isnot L'{' or pszIps[dwIpLen-1] isnot L'}') { Status = ERROR_INVALID_PARAMETER; if( pszIps ) { WinsFreeMemory(pszIps); pszIps = NULL; } goto ErrorReturn; } pTemp = pszIps+1; pszIps[dwIpLen-1] = L'\0';
pTemp = wcstok(pTemp, L","); while(( pTemp isnot NULL ) && (dwIpCount < WINSINTF_MAX_MEM ) ) { PDWORD pdwTemp = NULL; dwIpCount++; pdwTemp = WinsAllocateMemory(dwIpCount*sizeof(DWORD)); if( pdwTemp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; if( pszIps ) { WinsFreeMemory(pszIps); pszIps = NULL; } goto ErrorReturn; } if( pdwIp ) { memcpy(pdwTemp, pdwIp, (dwIpCount-1)*sizeof(DWORD)); WinsFreeMemory(pdwIp); pdwIp = NULL; } dwIp = StringToIpAddress(pTemp);
if( dwIp is INADDR_NONE ) { Status = ERROR_INVALID_PARAMETER; if( pszIps ) { WinsFreeMemory(pszIps); pszIps = NULL; } if( pdwTemp ) { WinsFreeMemory(pdwTemp); pdwTemp = NULL; }
goto ErrorReturn; } pdwTemp[dwIpCount-1] = dwIp; pdwIp = pdwTemp; pTemp = wcstok(NULL, L","); } if( pszIps ) { WinsFreeMemory(pszIps); pszIps = NULL; } break; } case 1: { DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 10); if( dw is 0 ) { fDelete = FALSE; break; } else if( dw is 1 ) { fDelete = TRUE; break; } else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } }
for( i=0; i<dwIpCount; i++ ) { WinsAdd.Len = 4; WinsAdd.Type = 0; WinsAdd.IPAdd = pdwIp[i];
if( fDelete is TRUE ) { DisplayMessage(g_hModule, MSG_WINS_DELETING_RECORD, IpAddressToString(pdwIp[i]));
Status = WinsDeleteWins(g_hBind, &WinsAdd); } else { WINSINTF_RESULTS_T Results; WINSINTF_VERS_NO_T MinVer; WINSINTF_VERS_NO_T MaxVer; MaxVer.HighPart = 0; MaxVer.LowPart = 0; MinVer.HighPart = 0; MinVer.LowPart = 0;
DisplayMessage(g_hModule, MSG_WINS_TOMBSTONE_RECORD, IpAddressToString(pdwIp[i]));
Status = WinsTombstoneDbRecs(g_hBind, &WinsAdd, MinVer, MaxVer); } if( Status isnot NO_ERROR ) { DisplayMessage(g_hModule, EMSG_WINS_OPERATION_FAILED, IpAddressToString(pdwIp[i]));
DisplayErrorMessage(EMSG_SRVR_ERROR_MESSAGE, Status); continue; }
}
if( Status is NO_ERROR ) { DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); }
CommonReturn:
if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; }
if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; }
if( pdwIp ) { WinsFreeMemory(pdwIp); pdwIp = NULL; }
return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_DELETE_WINS, Status); goto CommonReturn;}
DWORDHandleSrvrDeletePersona( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Delete one or all PNG servers from the list of PNG serversArguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : A list of server IP Address separated by commas and enclosed by {}. If no server address is provided within {} it will delete all PNG servers. Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwTagCount, dwNumArgs; HKEY hServer = NULL, hPartner = NULL; TAG_TYPE pttTags[] = { {WINS_TOKEN_SERVERS,TRUE, FALSE}, };
LPBYTE pbData = NULL, pbValue = NULL; DWORD dwSize = 0, dwType = 0, dwCount = 0, dwTemp = 0; LPWSTR pTemp = NULL, pwszPng = NULL; DWORD dwTagType = 1, dwTagNum = 1, dwLenCount = 0, dwPngIp = 0; LPDWORD pdwPngIp = NULL; BOOL fAtleastone = FALSE; BOOL fGrata;
fGrata = (wcsstr(CMD_SRVR_DELETE_PNGSERVER, ppwcArguments[dwCurrentIndex-1]) == NULL);
if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, fGrata ? HLP_SRVR_DELETE_PGSERVER_EX : HLP_SRVR_DELETE_PNGSERVER_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, &dwTagType, &dwTagNum);
if( Status isnot NO_ERROR ) goto ErrorReturn;
if( pttTags[0].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
if( wcslen(g_ServerNetBiosName) > 0 ) pTemp = g_ServerNetBiosName;
Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer);
if( Status isnot NO_ERROR ) { goto ErrorReturn; }
Status = RegOpenKeyEx(hServer, PARTNERROOT, 0, KEY_ALL_ACCESS, &hPartner);
if( Status isnot NO_ERROR ) { goto ErrorReturn; }
if (!CheckValidPgOp(hPartner, fGrata)) { DisplayMessage(g_hModule, fGrata ? EMSG_SRVR_PG_INVALIDOP : EMSG_SRVR_PNG_INVALIDOP); Status = ERROR_INVALID_PARAMETER; goto CommonReturn; }
Status = RegQueryValueEx(hPartner, fGrata ? PGSERVER : PNGSERVER, NULL, &dwType, NULL, &dwSize);
if( Status isnot NO_ERROR && Status isnot ERROR_FILE_NOT_FOUND) goto ErrorReturn;
if( dwSize < 7 ) { DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); goto CommonReturn; } pbData = WinsAllocateMemory(dwSize);
if( pbData is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
Status = RegQueryValueEx(hPartner, fGrata ? PGSERVER : PNGSERVER, NULL, &dwType, pbData, &dwSize);
if( Status isnot NO_ERROR ) goto ErrorReturn;
pwszPng = (LPWSTR)pbData;
for( i=0; i<(dwSize+2)/sizeof(WCHAR); i++ ) { if( pwszPng[i] is L'\0' and pwszPng[i+1] isnot L'\0') { pwszPng[i] = L','; i++; } }
dwPngIp = 0; dwCount = 0; pTemp = wcstok(pwszPng, L","); while((pTemp isnot NULL) && (dwLenCount+sizeof(WCHAR)*7<dwSize)) { LPDWORD pdwTemp = pdwPngIp; dwPngIp++; dwLenCount += wcslen(pTemp)*sizeof(WCHAR); pdwPngIp = WinsAllocateMemory(dwPngIp*sizeof(DWORD));
if( pdwPngIp is NULL ) { WinsFreeMemory(pdwTemp); pdwTemp = NULL; Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } if( pdwTemp isnot NULL ) { memcpy(pdwPngIp, pdwTemp, (dwPngIp-1)*sizeof(DWORD)); WinsFreeMemory(pdwTemp); pdwTemp = NULL; } pdwPngIp[dwPngIp-1] = StringToIpAddress(pTemp);
pTemp = wcstok(NULL, L","); dwLenCount+=sizeof(WCHAR); }
//Now parse the data
{ LPWSTR pwszTemp = NULL; DWORD dwLen = wcslen(ppwcArguments[dwCurrentIndex]); if( ppwcArguments[dwCurrentIndex][0] isnot L'{' or ppwcArguments[dwCurrentIndex][dwLen-1] isnot L'}' ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
if( dwLen < 7 ) { BYTE rgbByte[] = {0x00, 0x00, 0x00, 0x00}; Status = RegSetValueEx(hPartner, fGrata ? PGSERVER : PNGSERVER, 0, REG_MULTI_SZ, rgbByte, sizeof(rgbByte));
if( Status isnot NO_ERROR ) goto ErrorReturn;
fAtleastone = TRUE;
DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); goto CommonReturn;
}
pwszTemp = WinsAllocateMemory((dwLen-1)*sizeof(WCHAR));
if( pwszTemp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
wcsncpy(pwszTemp, ppwcArguments[dwCurrentIndex]+1, dwLen-1); pwszTemp[dwLen-2] = L'\0'; pTemp = wcstok(pwszTemp, L","); dwCount = 0;
while(pTemp isnot NULL ) { DWORD dw = StringToIpAddress(pTemp); BOOL fPresent = TRUE; if( dw is INADDR_NONE ) { DisplayMessage(g_hModule, EMSG_WINS_INVALID_IPADDRESS, pTemp); } else { for( i=0; i<dwPngIp; i++ ) {
if( dw is pdwPngIp[i] ) { LPDWORD pdwTemp = pdwPngIp; pdwPngIp = WinsAllocateMemory((dwPngIp-1)*sizeof(DWORD));
if( pdwPngIp is NULL ) { WinsFreeMemory(pdwTemp); pdwTemp = NULL; WinsFreeMemory(pwszTemp); pwszTemp = NULL; Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } fAtleastone = TRUE; memcpy(pdwPngIp, pdwTemp, i*sizeof(DWORD)); for( j=i+1; j<dwPngIp; j++ ) { pdwPngIp[j-1] = pdwTemp[j]; } dwPngIp--; break; } else continue; } } pTemp = wcstok(NULL, L","); } dwTemp += 0;
pbValue = WinsAllocateMemory((dwPngIp*(MAX_IP_STRING_LEN+1)+1)*sizeof(WCHAR)); if( pbValue is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } pTemp = (LPWSTR)pbValue;
for( i=0; i<dwPngIp; i++ ) { LPWSTR pwIp = NULL;
pwIp = IpAddressToString(pdwPngIp[i]); if( pwIp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } wcscpy(pTemp+dwTemp, pwIp); dwTemp+=wcslen(pwIp); dwTemp++; WinsFreeMemory(pwIp); pwIp = NULL; }
pTemp[dwTemp++] = L'\0'; } Status = RegSetValueEx(hPartner, fGrata ? PGSERVER : PNGSERVER, 0, REG_MULTI_SZ, pbValue, dwTemp*sizeof(WCHAR));
if( Status isnot NO_ERROR ) goto ErrorReturn;
if( fAtleastone is FALSE ) { DisplayMessage(g_hModule, fGrata ? EMSG_SRVR_ATLEAST_ONE_PG : EMSG_SRVR_ATLEAST_ONE_PNG); } DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS);
CommonReturn:
if( pbData ) { WinsFreeMemory(pbData); pbData = NULL; }
if( pbValue ) { WinsFreeMemory(pbValue); pbValue = NULL; }
if( pdwPngIp ) { WinsFreeMemory(pdwPngIp); pdwPngIp = NULL; }
if( hServer ) { RegCloseKey(hServer); hServer = NULL; }
if( hPartner ) { RegCloseKey(hPartner); hPartner = NULL; }
return Status;ErrorReturn: DisplayErrorMessage(fGrata ? EMSG_SRVR_DELETE_PGSERVER : EMSG_SRVR_DELETE_PNGSERVER, Status); goto CommonReturn; }
DWORDHandleSrvrInitBackup( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Initiates a backup operation of WINS Server database.Arguments : All aguments are passes as array of wide char strings in ppwcArguments. Optional : Backup directory. If none is specified, it will assume the the default directory. Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount; PDWORD pdwNumTags = NULL, pdwTypeTags = NULL; LPSTR pszBackupPath = NULL; LPWSTR pwszTemp = NULL; BOOL fIncremental = FALSE; LPBYTE lpStr = NULL; TAG_TYPE pttTags[] = { {WINS_TOKEN_DIR, FALSE, FALSE}, {WINS_TOKEN_TYPE, FALSE, FALSE}, }; if( dwArgCount > dwCurrentIndex ) { dwNumArgs = dwArgCount - dwCurrentIndex;
pdwNumTags = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwNumTags is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
pdwTypeTags = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTypeTags is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTypeTags, pdwNumTags);
if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
i = dwTagCount;
for( j = 0; j < i ; j++ ) { switch(pdwTypeTags[j]) { case 0: { pszBackupPath = WinsUnicodeToOem(ppwcArguments[dwCurrentIndex+pdwNumTags[j]], NULL);
if( pszBackupPath is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } break; } case 1: { DWORD dwType = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwNumTags[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dwType = STRTOUL(ppwcArguments[dwCurrentIndex+pdwNumTags[j]], NULL, 0); if( dwType is 0 ) fIncremental = FALSE; else if( dwType is 1 ) fIncremental = TRUE; else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } } } if( pszBackupPath is NULL ) //Look for the previously set backup path from Registry
{ HKEY hServer = NULL, hParameter = NULL; WCHAR wcTempSrc[1024] = {L'\0'}, wcTempDst[1024] = {L'\0'}; LPSTR pszTempPath = NULL; DWORD dwType = REG_EXPAND_SZ, dwTempLen = 1024*sizeof(WCHAR);
if( wcslen(g_ServerNetBiosName) > 0 ) pwszTemp = g_ServerNetBiosName;
Status = RegConnectRegistry(pwszTemp, HKEY_LOCAL_MACHINE, &hServer); if( Status isnot NO_ERROR ) { goto ErrorReturn; }
Status = RegOpenKeyEx(hServer, PARAMETER, 0, KEY_ALL_ACCESS, &hParameter);
if( Status isnot NO_ERROR ) { RegCloseKey(hServer); hServer = NULL; goto ErrorReturn; }
Status = RegQueryValueEx(hParameter, WINSCNF_BACKUP_DIR_PATH_NM, 0, &dwType, (LPBYTE)wcTempSrc, &dwTempLen);
RegCloseKey(hParameter); hParameter = NULL; RegCloseKey(hServer); hServer = NULL;
if( Status is ERROR_FILE_NOT_FOUND or wcslen(wcTempSrc) is 0 ) { DisplayMessage(g_hModule, EMSG_SRVR_NOBACKUP_PATH); Status = ERROR_INVALID_PARAMETER; goto CommonReturn; } if( Status isnot NO_ERROR ) { goto ErrorReturn; }
dwTempLen = ExpandEnvironmentStrings(wcTempSrc, wcTempDst, 1024);
if( dwTempLen is 0 ) { goto ErrorReturn; }
pszTempPath = WinsUnicodeToOem(wcTempDst, NULL); if( pszTempPath is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } dwTempLen = MAX(dwTempLen, strlen(pszTempPath));
pszBackupPath = WinsAllocateMemory(dwTempLen+1);
if( pszBackupPath is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
strcpy(pszBackupPath, pszTempPath);
WinsFreeMemory(pszTempPath); pszTempPath = NULL; } if( pszBackupPath[strlen(pszBackupPath) - 1] is '\\' ) { pszBackupPath[strlen(pszBackupPath) - 1] = '\0'; } Status = WinsBackup(g_hBind, (LPBYTE)pszBackupPath, (short)fIncremental);
if( Status isnot NO_ERROR ) goto ErrorReturn; CommonReturn: if( Status is NO_ERROR ) DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); if( pdwTypeTags ) { WinsFreeMemory(pdwTypeTags); pdwTypeTags = NULL; } if( pdwNumTags ) { WinsFreeMemory(pdwNumTags); pdwNumTags = NULL; } if( pszBackupPath ) { WinsFreeMemory(pszBackupPath); pszBackupPath = NULL; } return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_INIT_BACKUP, Status); goto CommonReturn; }
DWORDHandleSrvrInitImport( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Initiates import of records from LMHOSTS file.Arguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : Lmhosts file name. Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount; TAG_TYPE pttTags[] = { {WINS_TOKEN_FILE, TRUE, FALSE}, }; PDWORD pdwTagType = NULL, pdwTagNum = NULL; LPWSTR pwszFileName = NULL; WCHAR wcTemp[2042] = {L'\0'};
if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, HLP_SRVR_INIT_IMPORT_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum);
if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
for( j=0; j<dwTagCount; j++ ) { switch(pdwTagType[j]) { case 0: { DWORD dwLen = wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); if( dwLen is 0 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } pwszFileName = WinsAllocateMemory((dwLen+1)*sizeof(WCHAR)); if( pwszFileName is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } wcscpy(pwszFileName, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } } if( pwszFileName ) { //
// If this is a local connection, we copy the file to
// temporary name (the source may be on a remote drive
// which is not accessible to the WINS service.
//
// If this is not a local connection, attempt to copy
// the file to a temp name on C$ of the WINS server
//
do { if (IsLocalServer()) { wcscpy(wcTemp ,_wtempnam(NULL, L"WINS")); //
// First copy file to a temporary name (since the file
// could be remote), and then import and delete this file
//
if (!CopyFile(pwszFileName, wcTemp, TRUE)) { Status = GetLastError(); break; } //
// Now import the temporary file, and delete the file
// afterwards.
//
Status = ImportStaticMappingsFile(wcTemp, TRUE); DeleteFile(wcTemp); } else { //
// Try copying to the remote machine C: drive
//
wcscat(wcTemp, L"\\\\"); wcscat(wcTemp, g_ServerNameUnicode); wcscat(wcTemp, L"\\"); wcscat(wcTemp, L"C$"); wcscat(wcTemp, L"\\"); wcscat(wcTemp, L"WINS"); j = wcslen(wcTemp);
i=0; while (TRUE) { WCHAR Buffer[10] = {L'\0'}; DWORD dwErr = 0; _itow(i, Buffer, 10); wcscat(wcTemp,Buffer);
if (GetFileAttributes(wcTemp) == -1) { dwErr = GetLastError(); if (dwErr is ERROR_FILE_NOT_FOUND) { break; } } wcTemp[j] = L'\0'; i++; } //
// First copy file to a temporary name (since the file
// could be remote), and then import and delete this file
//
if (!CopyFile(pwszFileName, wcTemp, TRUE)) { Status = GetLastError(); break; }
//
// Now import the temporary file, and delete the file
// afterwards.
//
Status = ImportStaticMappingsFile(wcTemp, TRUE); DeleteFile(wcTemp); } }while(FALSE);
if (Status isnot NO_ERROR ) { goto ErrorReturn; }
}
CommonReturn: if( Status is NO_ERROR ) DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS);
if( pwszFileName ) { WinsFreeMemory(pwszFileName); pwszFileName = NULL; }
if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; }
if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; } return Status;
ErrorReturn: DisplayErrorMessage(EMSG_SRVR_INIT_IMPORT, Status); goto CommonReturn;}
DWORDHandleSrvrInitPull( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Initiates and send pull trigger to the specified pull partner.Arguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : Server Ip Address Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; WINSINTF_ADD_T WinsAdd; DWORD i, j, dwNumArgs, dwTagCount = 0; PDWORD pdwTagType = NULL, pdwTagNum = NULL; TAG_TYPE pttTags[] = { {WINS_TOKEN_SERVER, TRUE, FALSE}, }; LPWSTR pwszTemp = NULL;
if( dwArgCount < dwCurrentIndex+1 ) { DisplayMessage(g_hModule, HLP_SRVR_INIT_PULL_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if(pdwTagType is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); if(pdwTagType is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum); if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
i = dwTagCount;
for( j=0; j<i; j++ ) { switch(pdwTagType[j]) { case 0: { WinsAdd.Len = 4; WinsAdd.Type = 0; if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is 0 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
if( IsIpAddress(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) ) { WinsAdd.IPAdd = StringToIpAddress(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); break; } else //Server UNC name
{ struct hostent * lpHostEnt = NULL; CHAR cAddr[16]; BYTE pbAdd[4]; char szAdd[4]; int k = 0, l=0; DWORD nLen = 0;
if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) > 2 and _wcsnicmp(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], L"\\\\", 2) is 0 ) k = 2; lpHostEnt = gethostbyname(WinsUnicodeToAnsi(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]+k, NULL)); if( lpHostEnt is NULL ) { DisplayMessage(g_hModule, EMSG_WINS_INVALID_COMPUTER_NAME, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]+k); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
memcpy(pbAdd, lpHostEnt->h_addr_list[0], 4); for( l=0;l<4; l++) { _itoa((int)pbAdd[l], szAdd, 10); memcpy(cAddr+nLen, szAdd, strlen(szAdd)); nLen += strlen(szAdd); *(cAddr+nLen) = '.'; nLen++; } *(cAddr+nLen-1) = '\0'; WinsAdd.IPAdd = WinsDottedStringToIpAddress(cAddr); } break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } }
for( j=0; j<sizeof(pttTags)/sizeof(TAG_TYPE); j++) { if( pttTags[j].dwRequired is TRUE && pttTags[j].bPresent is FALSE ) { DisplayMessage(g_hModule, EMSG_WINS_REQUIRED_PARAMETER, j); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
Status = WinsTrigger(g_hBind, &WinsAdd, WINSINTF_E_PULL);
if( Status isnot NO_ERROR ) goto ErrorReturn;
CommonReturn: if( Status is NO_ERROR ) { DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); } if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; } if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; } return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_INIT_PULL, Status); goto CommonReturn;
}
DWORDHandleSrvrInitPullrange( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Initiates and pulls a range of database from a particular server.Arguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : Pulls a range of database from a particular server owned by a particular server within the given version range. Note : If no ip is provided, it deletes all partners in the list. Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwTagCount, dwNumArgs; TAG_TYPE pttTags[] = { {WINS_TOKEN_OWNER, TRUE, FALSE}, {WINS_TOKEN_SERVER, TRUE, FALSE}, {WINS_TOKEN_MAXVER, TRUE, FALSE}, {WINS_TOKEN_MINVER, TRUE, FALSE}, };
LPDWORD pdwTagType = NULL, pdwTagNum = NULL;
WINSINTF_VERS_NO_T MinVer, MaxVer; WINSINTF_ADD_T PullAdd, OwnerAdd ;
if( dwArgCount < dwCurrentIndex + 4 ) { DisplayMessage(g_hModule, HLP_SRVR_INIT_PULLRANGE_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum);
if( Status isnot NO_ERROR ) goto ErrorReturn;
for( j=0; j<sizeof(pttTags)/sizeof(TAG_TYPE); j++ ) { if( pttTags[j].dwRequired is TRUE and pttTags[j].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
for( j=0; j<dwTagCount; j++ ) { switch(pdwTagType[j]) { case 0: { if( IsIpAddress(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } OwnerAdd.IPAdd = StringToIpAddress(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); OwnerAdd.Type = 0; OwnerAdd.Len = 4; break; } case 1: { if( IsIpAddress(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } PullAdd.IPAdd = StringToIpAddress(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); PullAdd.Type = 0; PullAdd.Len = 4; break; } case 2: { Status = GetVersionData(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], &MaxVer); if( Status isnot NO_ERROR ) goto ErrorReturn; break; } case 3: { Status = GetVersionData(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], &MinVer); if( Status isnot NO_ERROR ) goto ErrorReturn; break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
}
Status = WinsPullRange(g_hBind, &PullAdd, &OwnerAdd, MinVer, MaxVer);
if( Status isnot NO_ERROR ) goto ErrorReturn;
DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS);
CommonReturn:
if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; }
if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; }
return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_INIT_PULLRANGE, Status); goto CommonReturn; }
DWORDHandleSrvrInitPush( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Initiate and sends push trigger to a particular Push partner.Arguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : Push partner's IP Address Optional : If user wants to propagate the push trigger Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; WINSINTF_ADD_T WinsAdd; DWORD i, j, dwNumArgs, dwTagCount=0; PDWORD pdwTagType = NULL, pdwTagNum = NULL; TAG_TYPE pttTags[] = { {WINS_TOKEN_SERVER, TRUE, FALSE}, {WINS_TOKEN_PROPAGATION, FALSE, FALSE}, }; LPWSTR pwszTemp = NULL; DWORD dwChoice = WINSINTF_E_PUSH;
if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, HLP_SRVR_INIT_PUSH_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if(pdwTagType is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); if(pdwTagType is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum); if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
i = dwTagCount;
for( j=0; j<i; j++ ) { switch(pdwTagType[j]) { case 0: { WinsAdd.Len = 4; WinsAdd.Type = 0;
if( IsIpAddress(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) ) { WinsAdd.IPAdd = StringToIpAddress(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); } else //Server UNC name
{ struct hostent * lpHostEnt = NULL; CHAR cAddr[16]; BYTE pbAdd[4]; char szAdd[4]; int k = 0, l=0; DWORD nLen = 0;
if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) > 2 and _wcsnicmp(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], L"\\\\", 2) is 0 ) k = 2; lpHostEnt = gethostbyname(WinsUnicodeToAnsi(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]+k, NULL)); if( lpHostEnt is NULL ) { DisplayMessage(g_hModule, EMSG_WINS_INVALID_COMPUTER_NAME, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]+k); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
memcpy(pbAdd, lpHostEnt->h_addr_list[0], 4); for( l=0;l<4; l++) { _itoa((int)pbAdd[l], szAdd, 10); memcpy(cAddr+nLen, szAdd, strlen(szAdd)); nLen += strlen(szAdd); *(cAddr+nLen) = '.'; nLen++; } *(cAddr+nLen-1) = '\0'; WinsAdd.IPAdd = WinsDottedStringToIpAddress(cAddr); } break; } case 1: { DWORD dwVal = 0; if( ( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) or ( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) > 1 ) ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } if( ppwcArguments[dwCurrentIndex+pdwTagNum[j]][0] is L'1' ) dwChoice = WINSINTF_E_PUSH_PROP; else if( ppwcArguments[dwCurrentIndex+pdwTagNum[j]][0] is L'0' ) dwChoice = WINSINTF_E_PUSH; else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
break;
} default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } }
for( j=0; j<sizeof(pttTags)/sizeof(TAG_TYPE); j++) { if( pttTags[j].dwRequired is TRUE && pttTags[j].bPresent is FALSE ) { DisplayMessage(g_hModule, EMSG_WINS_REQUIRED_PARAMETER, j); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
Status = WinsTrigger(g_hBind, &WinsAdd, dwChoice);
if( Status isnot NO_ERROR ) goto ErrorReturn;
CommonReturn: if( Status is NO_ERROR ) { DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); } if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; } if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; } return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_INIT_PUSH, Status); goto CommonReturn;
}
DWORDHandleSrvrInitReplicate( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Initiates a database replication with the partnersArguments : All aguments are passes as array of wide char strings in ppwcArguments. NONE. Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; HKEY hServer = NULL, hPullRoot = NULL, hPushRoot = NULL; DWORD dw = 0, dwKeyLen = MAX_IP_STRING_LEN+1, dwPullKeys = 0, dwPushKeys = 0;
WCHAR wcIpAddress[MAX_IP_STRING_LEN+1] = {L'\0'}; LPWSTR pTemp = NULL;
WINSINTF_ADD_T WinsAdd;
if( wcslen(g_ServerNetBiosName) > 0 ) { pTemp = g_ServerNetBiosName; }
Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer);
if( Status isnot NO_ERROR ) { goto ErrorReturn; }
Status = RegOpenKeyEx(hServer, PULLROOT, 0, KEY_ALL_ACCESS, &hPullRoot);
if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegQueryInfoKey(hPullRoot, NULL, NULL, NULL, &dwPullKeys, NULL, NULL, NULL, NULL, NULL, NULL, NULL);
if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegOpenKeyEx(hServer, PUSHROOT, 0, KEY_ALL_ACCESS, &hPushRoot);
if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegQueryInfoKey(hPushRoot, NULL, NULL, NULL, &dwPushKeys, NULL, NULL, NULL, NULL, NULL, NULL, NULL);
if( Status isnot NO_ERROR ) goto ErrorReturn;
DisplayMessage(g_hModule, MSG_WINS_TIME_WARNING);
WinsAdd.Len = 4; WinsAdd.Type = 0;
if( dwPullKeys is 0 ) { DisplayMessage(g_hModule, EMSG_WINS_NO_PULLPARTNER); }
if( dwPushKeys is 0 ) { DisplayMessage(g_hModule, EMSG_WINS_NO_PUSHPARTNER); }
DisplayMessage(g_hModule, WINS_FORMAT_LINE);
for( dw=0; dw<dwPullKeys; dw++ ) { dwKeyLen = MAX_IP_STRING_LEN+1; Status = RegEnumKeyEx(hPullRoot, dw, wcIpAddress, &dwKeyLen, NULL, NULL, NULL, NULL); if( Status isnot NO_ERROR ) continue;
WinsAdd.IPAdd = StringToIpAddress(wcIpAddress);
DisplayMessage(g_hModule, MSG_WINS_SEND_PULL, wcIpAddress); Status = WinsTrigger(g_hBind, &WinsAdd, WINSINTF_E_PULL); if( Status isnot NO_ERROR ) { DisplayMessage(g_hModule, EMSG_WINS_PULL_FAILED); } else { DisplayMessage(g_hModule, MSG_WINS_TRIGGER_DONE); }
}
WinsAdd.Len = 4; WinsAdd.Type = 0;
for( dw=0; dw<dwPushKeys; dw++ ) { dwKeyLen = MAX_IP_STRING_LEN+1;
Status = RegEnumKeyEx(hPushRoot, dw, wcIpAddress, &dwKeyLen, NULL, NULL, NULL, NULL); if( Status isnot NO_ERROR ) continue;
WinsAdd.IPAdd = StringToIpAddress(wcIpAddress);
DisplayMessage(g_hModule, MSG_WINS_SEND_PUSH, wcIpAddress); Status = WinsTrigger(g_hBind, &WinsAdd, WINSINTF_E_PUSH_PROP);
if( Status isnot NO_ERROR ) { DisplayMessage(g_hModule, EMSG_WINS_PUSH_FAILED); } else { DisplayMessage(g_hModule, MSG_WINS_TRIGGER_DONE); } } if( Status isnot NO_ERROR ) goto CommonReturn;
DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS);CommonReturn: if( hServer ) { RegCloseKey(hServer); hServer = NULL; }
if( hPullRoot ) { RegCloseKey(hPullRoot); hPullRoot = NULL; }
if( hPushRoot ) { RegCloseKey(hPushRoot); hPushRoot = NULL; }
return Status;
ErrorReturn: if( Status is ERROR_FILE_NOT_FOUND ) { DisplayMessage(g_hModule, MSG_WINS_NO_PARTNER); } else { DisplayErrorMessage(EMSG_SRVR_INIT_REPLICATE, Status); } goto CommonReturn;
}
DWORDHandleSrvrInitRestore( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Initiates and restore database Arguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : Directory to do restore from Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR, Status1 = NO_ERROR, i, j, dwNumArgs, dwTagCount=0; PDWORD pdwNumTags = NULL, pdwTypeTags = NULL; TAG_TYPE pttTags[] = { {WINS_TOKEN_DIR, TRUE, FALSE}, {WINS_TOKEN_VERSION, FALSE, FALSE}, }; CHAR szRestorePath[MAX_PATH+1] = {'\0'}; LPWSTR pwszTemp = NULL; DWORD dwService = NO_ERROR; DbVersion eVersion = 3; BOOL fBackupOnTerm = FALSE;
handle_t wbdhBind = g_hBind; WINSINTF_BIND_DATA_T wbdBindData = g_BindData; DWORD dwError = NO_ERROR; if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, HLP_SRVR_INIT_RESTORE_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
if( IsLocalServer() is FALSE ) { DisplayMessage(g_hModule, EMSG_WINS_LOCAL_SERVER, g_ServerIpAddressUnicodeString); return ERROR_INVALID_PARAMETER; }
dwNumArgs = dwArgCount - dwCurrentIndex;
pdwNumTags = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwNumTags is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
pdwTypeTags = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTypeTags is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTypeTags, pdwNumTags); if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
i = dwTagCount;
for( j = 0; j < i ; j++ ) { switch(pdwTypeTags[j]) { case 0: { DWORD dw = 0; LPSTR pszTempPath = NULL; pszTempPath = WinsUnicodeToOem(ppwcArguments[dwCurrentIndex+pdwNumTags[j]], NULL);
if( pszTempPath is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } dw = strlen(pszTempPath);
if( dw < 2 ) { Status = ERROR_INVALID_PARAMETER; WinsFreeMemory( pszTempPath ); pszTempPath = NULL; goto ErrorReturn; }
strncpy(szRestorePath, pszTempPath, ( dw > MAX_PATH - 1 ) ? MAX_PATH - 1 : dw );
WinsFreeMemory(pszTempPath); pszTempPath = NULL;
break; } case 1: { DWORD eVer = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwNumTags[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } eVer = STRTOUL(ppwcArguments[dwCurrentIndex+pdwNumTags[j]], NULL, 0); if( eVer > DbVersionMin && eVer < DbVersionMax ) eVersion = eVer; else { Status = ERROR_INVALID_DB_VERSION; goto ErrorReturn; } break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } } //To restore the database, first stop WINS Service, Cleanup earlier dabase
//do DB restore and then restart WINS
{
//Unbind WINS
WinsUnbind(&g_BindData, g_hBind); g_hBind = NULL; //Reset Backup on Termination flag
{ HKEY hServer = NULL, hParameter = NULL;
LPWSTR pTemp = NULL; DWORD dwSize = sizeof(DWORD); DWORD dwType = REG_DWORD; DWORD dwData = 0;
if( wcslen(g_ServerNetBiosName) > 0 ) pTemp = g_ServerNetBiosName; Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer); if( Status isnot NO_ERROR ) goto EXIT;
Status = RegOpenKeyEx(hServer, PARAMETER, 0, KEY_ALL_ACCESS, &hParameter);
if( Status isnot NO_ERROR ) goto EXIT; Status = RegQueryValueEx(hParameter, WINSCNF_DO_BACKUP_ON_TERM_NM, NULL, &dwType, (LPBYTE)&dwData, &dwSize);
if( Status is NO_ERROR ) { if( dwData is 1 ) { DWORD dw = 0; Status = RegSetValueEx(hParameter, WINSCNF_DO_BACKUP_ON_TERM_NM, 0, REG_DWORD, (LPBYTE)&dw, sizeof(DWORD)); if( Status is NO_ERROR ) fBackupOnTerm = TRUE;
} else { fBackupOnTerm = FALSE; } }
EXIT: if( Status isnot NO_ERROR ) { DisplayMessage(g_hModule, EMSG_WINS_RESTORE_IMPROPER); }
if( hServer ) { RegCloseKey(hServer); hServer = NULL; }
if( hParameter ) { RegCloseKey(hParameter); hParameter = NULL; }
Status = NO_ERROR; }
//Stop WINS Service
Status = ControlWINSService(TRUE);
if( Status isnot NO_ERROR ) { if( Status isnot ERROR_SERVICE_NOT_ACTIVE ) { g_BindData = wbdBindData; g_hBind = WinsBind(&g_BindData); goto ErrorReturn; } else { dwService = ERROR_SERVICE_NOT_ACTIVE; Status = NO_ERROR; }
}
//Now try restoring the database
Status1 = WinsRestore((LPBYTE)szRestorePath);
if( Status1 isnot NO_ERROR ) { DisplayErrorMessage(EMSG_SRVR_INIT_RESTORE, Status1); }
if( dwService isnot ERROR_SERVICE_NOT_ACTIVE ) Status = ControlWINSService(FALSE); }
if( Status isnot NO_ERROR ) goto ErrorReturn;
if( Status1 isnot NO_ERROR ) { Status = Status1; goto ErrorReturn; }
DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS);
CommonReturn: g_BindData = wbdBindData; g_hBind = WinsBind(&g_BindData); //Reset the DoBackUpOnTerm value
if( fBackupOnTerm is TRUE ) { HKEY hServer = NULL, hParameter = NULL; LPWSTR pTemp = NULL; DWORD dwErr = NO_ERROR; if( wcslen(g_ServerNetBiosName) > 0 ) { pTemp = g_ServerNetBiosName; } dwErr = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer); if( dwErr isnot NO_ERROR ) { DisplayErrorMessage(EMSG_WINS_REGCONNECT_FAILED, dwErr); } else { dwErr = RegOpenKeyEx(hServer, PARAMETER, 0, KEY_ALL_ACCESS, &hParameter); if( dwErr isnot NO_ERROR ) { RegCloseKey(hServer); hServer = NULL; DisplayErrorMessage(EMSG_WINS_REGOPEN_FAILED, dwErr); } else { DWORD dw = 1; dwErr = RegSetValueEx(hParameter, WINSCNF_DO_BACKUP_ON_TERM_NM, 0, REG_DWORD, (LPBYTE)&dw, sizeof(DWORD)); if( dwErr isnot NO_ERROR ) { DisplayErrorMessage(EMSG_WINS_REGSETVAL_FAILED, dwErr); }
RegCloseKey(hParameter); hParameter = NULL; RegCloseKey(hServer); hServer = NULL; } } } if( pdwTypeTags ) { WinsFreeMemory(pdwTypeTags); pdwTypeTags = NULL; } if( pdwNumTags ) { WinsFreeMemory(pdwNumTags); pdwNumTags = NULL; }
return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_INIT_RESTORE, Status); goto CommonReturn; }
DWORDHandleSrvrInitScavenge( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Initiates scavenging of database for the serverArguments : All aguments are passes as array of wide char strings in ppwcArguments. NONE. Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; struct in_addr InAddr; BOOL fIpAddress = TRUE; DWORD j = 0;
WCHAR wcName[256] = {L'\0'}; DWORD dwLen = 0;
WINSINTF_ADD_T WinsAdd; WINSINTF_RESULTS_T Results; WINSINTF_VERS_NO_T MaxVer, MinVer; WINSINTF_RECS_T Recs; PWINSINTF_RECORD_ACTION_T pRow = NULL;
Status = WinsDoScavenging( g_hBind ); if( Status is NO_ERROR ) { DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); }
else { DisplayErrorMessage(EMSG_SRVR_INIT_SCAVENGE, Status); }
return Status;}
DWORDHandleSrvrInitSearch( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Search WINS database based on Name, 16th char, case and optionally stores the result to a file,Arguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : Name to search for Optional : 16th char, case sensitive or not and if the result to be stored in a file, then the file Name. Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, k, dwNumArgs, dwTagCount; TAG_TYPE pttTags[]={ {WINS_TOKEN_NAME, TRUE, FALSE}, {WINS_TOKEN_ENDCHAR, FALSE, FALSE}, {WINS_TOKEN_CASE, FALSE, FALSE}, {WINS_TOKEN_FILE, FALSE, FALSE}, }; PDWORD pdwTagType = NULL, pdwTagNum = NULL; CHAR chEndChar = (CHAR)0x00; BOOL fEndChar = FALSE; WCHAR wcName[256+1] = {L'\0'}; WCHAR wcFile[MAX_PATH] = {L'\0'}; BOOL fFile = FALSE; DWORD dwLen = 0; BOOL fIpAddress = TRUE; BOOL fMatch = FALSE; BOOL fCase = FALSE; BOOL fNew = TRUE; struct in_addr InAddr; LPSTR pszName = NULL;
WINSINTF_ADD_T WinsAdd; WINSINTF_RESULTS_T Results = {0}; WINSINTF_RESULTS_NEW_T ResultsN = {0}; WINSINTF_VERS_NO_T MaxVer, MinVer; WINSINTF_RECS_T Recs; PWINSINTF_RECORD_ACTION_T pRow = NULL;
i = j = k = dwNumArgs = dwTagCount = 0;
if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, HLP_SRVR_INIT_SEARCH_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum); if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
for( j=0; j<dwTagCount; j++) { switch(pdwTagType[j]) { case 0: { DWORD dwLen = 0; dwLen = wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); if( dwLen is 0 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dwLen = ( 255 < dwLen ) ? 255 : dwLen; memcpy(wcName, ppwcArguments[dwCurrentIndex+pdwTagNum[j]], dwLen*sizeof(WCHAR)); wcName[dwLen] = L'\0'; fIpAddress = FALSE; break; } case 1: { DWORD dwLen = 0, k=0; fEndChar = TRUE; dwLen = wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); if( dwLen > 2 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } for( k=0; k<dwLen; k++ ) { WCHAR wc = ppwcArguments[dwCurrentIndex+pdwTagNum[j]][k]; if( towlower(wc) >= L'a' and towlower(wc) <= L'z' ) { if( towlower(wc) > L'f' ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } } chEndChar = StringToHexA(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); break; } case 2: { DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 10); if( dw is 0 ) { fCase = FALSE; break; } if( dw is 1 ) { fCase = TRUE; break; } else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
} case 3: { DWORD dwLen = wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); if( dwLen is 0 ) { wcscpy(wcFile, L"wins.rec"); fFile = TRUE; break; } wcscpy(wcFile, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); fFile = TRUE; break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } } pszName = WinsUnicodeToOem(wcName, NULL);
if( pszName is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwLen = strlen(pszName); if( dwLen >= 16 ) { DisplayMessage(g_hModule, EMSG_SRVR_INVALID_NETBIOS_NAME); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
if( fEndChar ) { LPSTR pTemp = pszName; pszName = WinsAllocateMemory(17); if( pszName is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } memcpy(pszName, pTemp, 15); for( i=strlen(pszName); i<16; i++ ) { pszName[i] = ' '; } pszName[15] = chEndChar; pszName[16] = '\0'; WinsFreeMemory(pTemp); pTemp = NULL; dwLen = 16; }
ResultsN.WinsStat.NoOfPnrs = 0; ResultsN.WinsStat.pRplPnrs = 0; ResultsN.NoOfWorkerThds = 1; ResultsN.pAddVersMaps = NULL;
Status = WinsStatusNew(g_hBind, WINSINTF_E_CONFIG_ALL_MAPS, &ResultsN); if( Status is RPC_S_PROCNUM_OUT_OF_RANGE ) { //Try old API
Results.WinsStat.NoOfPnrs = 0; Results.WinsStat.pRplPnrs = 0; Status = WinsStatus(g_hBind, WINSINTF_E_CONFIG_ALL_MAPS, &Results); fNew = FALSE; } if( Status isnot NO_ERROR ) goto ErrorReturn;
g_dwSearchCount = 0; g_fHeader = FALSE;
if( fNew is FALSE ) { if( Results.NoOfOwners is 0 ) goto CommonReturn;
for( j=0; j<Results.NoOfOwners; j++) { WinsAdd.Len = 4; WinsAdd.Type = 0; WinsAdd.IPAdd = Results.AddVersMaps[j].Add.IPAdd;
if( WinsAdd.IPAdd is 0 ) continue; DisplayMessage(g_hModule, MSG_WINS_SEARCHING_STATUS, IpAddressToString(Results.AddVersMaps[j].Add.IPAdd));
InAddr.s_addr = htonl(Results.AddVersMaps[j].Add.IPAdd); MaxVer = Results.AddVersMaps[j].VersNo; MinVer.HighPart = 0; MinVer.LowPart = 0; Status = GetDbRecs(MinVer, MaxVer, &WinsAdd, inet_ntoa(InAddr), fIpAddress ? FALSE : TRUE, pszName, dwLen, FALSE, 0, FALSE, fCase, fFile, fFile ? wcFile : NULL); if(Status isnot NO_ERROR ) { if( Results.WinsStat.pRplPnrs) { WinsFreeMem(Results.WinsStat.pRplPnrs); Results.WinsStat.pRplPnrs = NULL; } goto ErrorReturn; } } if( Results.WinsStat.pRplPnrs) { WinsFreeMem(Results.WinsStat.pRplPnrs); Results.WinsStat.pRplPnrs = NULL; } if( Status isnot NO_ERROR ) goto ErrorReturn; } else { if( ResultsN.NoOfOwners is 0 ) goto CommonReturn; for( j=0; j<ResultsN.NoOfOwners; j++) {
WinsAdd.Len = 4; WinsAdd.Type = 0; WinsAdd.IPAdd = ResultsN.pAddVersMaps[j].Add.IPAdd;
if( WinsAdd.IPAdd is 0 ) continue;
DisplayMessage(g_hModule, MSG_WINS_SEARCHING_STATUS, IpAddressToString(ResultsN.pAddVersMaps[j].Add.IPAdd));
InAddr.s_addr = htonl(ResultsN.pAddVersMaps[j].Add.IPAdd); MaxVer = ResultsN.pAddVersMaps[j].VersNo; MinVer.HighPart = 0; MinVer.LowPart = 0; Status = GetDbRecs(MinVer, MaxVer, &WinsAdd, inet_ntoa(InAddr), fIpAddress ? FALSE : TRUE, pszName, dwLen, FALSE, 0, FALSE, fCase, fFile, fFile ? wcFile : NULL);
if(Status isnot NO_ERROR ) { if( ResultsN.WinsStat.pRplPnrs) { WinsFreeMem(ResultsN.WinsStat.pRplPnrs); ResultsN.WinsStat.pRplPnrs = NULL; } goto ErrorReturn; } } if( ResultsN.WinsStat.pRplPnrs) { WinsFreeMem(ResultsN.WinsStat.pRplPnrs); ResultsN.WinsStat.pRplPnrs = NULL; } if( Status isnot NO_ERROR ) goto ErrorReturn; }
if( pszName ) { WinsFreeMemory(pszName); pszName = NULL; }
DisplayMessage(g_hModule, MSG_SRVR_SEARCH_COUNT, g_dwSearchCount);
if( fFile ) { FILE * pFile = _wfopen(wcFile, L"a+");
if( pFile isnot NULL ) { DumpMessage(g_hModule, pFile, MSG_SRVR_SEARCH_COUNT, g_dwSearchCount);
fclose(pFile); pFile = NULL; } }
CommonReturn: if( Status is NO_ERROR ) { DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); }
if( pszName ) { WinsFreeMemory(pszName); pszName = NULL; } if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; } if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; } g_dwSearchCount = 0; g_fHeader = FALSE;
return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_INIT_SEARCH, Status); goto CommonReturn; }
DWORDHandleSrvrResetCounter( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Resets the version counterArguments : All aguments are passes as array of wide char strings in ppwcArguments. NONE.Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR;
Status = WinsResetCounters( g_hBind ); if( Status is NO_ERROR ) { DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); }
else { DisplayErrorMessage(EMSG_SRVR_INIT_BACKUP, Status); } return Status;}
DWORDHandleSrvrSetAutopartnerconfig( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Sets automatic partner configuration parameters.Arguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : State to set or disable Optional : Time Interval or TTL valueReturn Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount; PDWORD pdwTagType = NULL, pdwTagNum = NULL; LPWSTR pTemp = NULL;
TAG_TYPE pttTags[] = { {WINS_TOKEN_STATE, TRUE, FALSE}, {WINS_TOKEN_INTERVAL, FALSE, FALSE}, {WINS_TOKEN_TTL, FALSE, FALSE}, }; DWORD dwInterval = WINSCNF_DEF_MCAST_INTVL; DWORD dwTTL = WINSCNF_DEF_MCAST_TTL; BOOL fState = FALSE, fTTL = FALSE, fInterval = FALSE;
HKEY hServer = NULL, hParameter = NULL;
if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, HLP_SRVR_SET_AUTOPARTNERCONFIG_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
{ dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum);
if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
for( j=0; j<dwTagCount; j++ ) { switch(pdwTagType[j]) { case 0: { WCHAR wc = ppwcArguments[dwCurrentIndex+pdwTagNum[j]][0]; if( wc is L'1' ) { fState = TRUE; } else if( wc is L'0' ) { fState = FALSE; } else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } break; } case 1: { DWORD dw = 0; fInterval = TRUE; if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) < 1 ) break; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 0);
if( dw >= ONEDAY ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dwInterval = dw; break;
} case 2: { DWORD dw = 0; fTTL = TRUE; if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) < 1 ) break; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 0); if( dw > 32 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } if( dw <= 0 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dwTTL = dw; break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } } } { if( wcslen(g_ServerNetBiosName) > 0 ) { pTemp = g_ServerNetBiosName; } Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer); if( Status isnot NO_ERROR ) goto ErrorReturn; Status = RegOpenKeyEx(hServer, PARAMETER, 0, KEY_ALL_ACCESS, &hParameter);
if( Status isnot NO_ERROR ) goto ErrorReturn; Status = RegSetValueEx(hParameter, WINSCNF_USE_SELF_FND_PNRS_NM, 0, REG_DWORD, (LPBYTE)&fState, sizeof(BOOL)); if( Status isnot NO_ERROR ) goto ErrorReturn;
if( fInterval ) { Status = RegSetValueEx(hParameter, WINSCNF_MCAST_INTVL_NM, 0, REG_DWORD, (LPBYTE)&dwInterval, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto ErrorReturn; }
if( fTTL ) { Status = RegSetValueEx(hParameter, WINSCNF_MCAST_TTL_NM, 0, REG_DWORD, (LPBYTE)&dwTTL, sizeof(DWORD)); if( Status isnot NO_ERROR ) goto ErrorReturn; } }CommonReturn: if( Status is NO_ERROR ) { DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); }
if( hParameter ) { RegCloseKey(hParameter); hParameter = NULL; }
if( hServer ) { RegCloseKey(hServer); hServer = NULL; }
if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; } if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; } return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_SET_AUTOPARTNERCONFIG, Status); goto CommonReturn; }
DWORDHandleSrvrSetBackuppath( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Sets the backup path for the WINS databaseArguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : Backup dir Optional : Enable backup at server shutdown.Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount; PDWORD pdwTagType = NULL, pdwTagNum = NULL;
TAG_TYPE pttTags[] = { {WINS_TOKEN_DIR, FALSE, FALSE}, {WINS_TOKEN_ATSHUTDOWN, FALSE, FALSE}, };
LPWSTR pwszDir = L"C:\\\\"; BOOL fDir = FALSE; BOOL fAtShutDown = FALSE; LPWSTR pTemp = NULL; HKEY hServer = NULL, hParameter = NULL;
if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, HLP_SRVR_SET_BACKUPPATH_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum);
if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
for( j=0; j<dwTagCount; j++ ) { switch(pdwTagType[j]) { case 0: { if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) > 0 ) { pwszDir = ppwcArguments[dwCurrentIndex+pdwTagNum[j]]; } else { pwszDir = L""; }
fDir = TRUE;
break; } case 1: { WCHAR wc = ppwcArguments[dwCurrentIndex+pdwTagNum[j]][0]; if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) isnot 1 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } if( wc is L'0' ) fAtShutDown = FALSE; else if( wc is L'1' ) fAtShutDown = TRUE; else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
} }
if( wcslen(g_ServerNetBiosName) > 0 ) { pTemp = g_ServerNetBiosName; }
Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer); if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegCreateKeyEx(hServer, PARAMETER, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hParameter, NULL);
if( Status isnot NO_ERROR ) goto ErrorReturn;
if( fDir ) { Status = RegSetValueEx(hParameter, WINSCNF_BACKUP_DIR_PATH_NM, 0, REG_EXPAND_SZ, (LPBYTE)pwszDir, (wcslen(pwszDir)+1)*sizeof(WCHAR));
if( Status isnot NO_ERROR ) { goto ErrorReturn; } } Status = RegSetValueEx(hParameter, WINSCNF_DO_BACKUP_ON_TERM_NM, 0, REG_DWORD, (LPBYTE)&fAtShutDown, sizeof(BOOL)); if( Status isnot NO_ERROR ) goto ErrorReturn;
CommonReturn: if( Status is NO_ERROR ) { DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); }
if( hParameter ) { RegCloseKey(hParameter); hParameter = NULL; }
if( hServer ) { RegCloseKey(hServer); hServer = NULL; }
if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; } if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; } return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_SET_BACKUPPATH, Status); goto CommonReturn;
}
DWORDHandleSrvrSetDefaultparam( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Sets the default values for all the configuration parameter. This command must be run at least once before running the dump command.Arguments : All aguments are passes as array of wide char strings in ppwcArguments. NONE.Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; HKEY hServer = NULL, hWins = NULL, hParameter = NULL, hDefault = NULL, hDefaultPull = NULL, hDefaultPush = NULL, hPartner = NULL, hCheck = NULL, hPullPart = NULL, hPushPart = NULL; DWORD dwSize = 0, dwType = 0, dwData = 0; LPWSTR pTemp = NULL;
if( wcslen(g_ServerNetBiosName) > 2 ) pTemp = g_ServerNetBiosName;
Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer);
if( Status isnot NO_ERROR ) { goto RETURN; }
//Open all required Registry key handles
Status = RegCreateKeyEx(hServer, PARAMETER, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hParameter, NULL);
if( Status isnot NO_ERROR ) goto RETURN;
Status = RegCreateKeyEx(hServer, PARTNERROOT, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hPartner, NULL);
if( Status isnot NO_ERROR ) goto RETURN;
Status = RegCreateKeyEx(hServer, PULLROOT, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hPullPart, NULL);
if( Status isnot NO_ERROR ) goto RETURN;
Status = RegCreateKeyEx(hServer, PUSHROOT, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hPushPart, NULL);
if( Status isnot NO_ERROR ) goto RETURN;
Status = RegCreateKeyEx(hServer, DEFAULTPULL, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hDefaultPull, NULL);
if( Status isnot NO_ERROR ) goto RETURN;
Status = RegCreateKeyEx(hServer, DEFAULTPUSH, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hDefaultPush, NULL);
if( Status isnot NO_ERROR ) goto RETURN;
//Start setting the default values.
Status = RegSetValueEx(hParameter, WINSCNF_BACKUP_DIR_PATH_NM, 0, REG_EXPAND_SZ, (LPBYTE)L"", (wcslen(L"")+1)*sizeof(WCHAR));
if( Status isnot NO_ERROR ) goto RETURN;
dwData = 0;
Status = RegSetValueEx(hParameter, WINSCNF_DO_BACKUP_ON_TERM_NM, 0, REG_DWORD, (LPBYTE)&dwData, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto RETURN;
dwData = 0x7e900;
Status = RegSetValueEx(hParameter, WINSCNF_REFRESH_INTVL_NM, 0, REG_DWORD, (LPBYTE)&dwData, sizeof(DWORD)); if( Status isnot NO_ERROR ) goto RETURN;
dwData = 0x54600;
Status = RegSetValueEx(hParameter, WINSCNF_TOMBSTONE_INTVL_NM, 0, REG_DWORD, (LPBYTE)&dwData, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto RETURN;
dwData = 0x7e900;
Status = RegSetValueEx(hParameter, WINSCNF_TOMBSTONE_TMOUT_NM, 0, REG_DWORD, (LPBYTE)&dwData, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto RETURN;
dwData = 0x1fa400;
Status = RegSetValueEx(hParameter, WINSCNF_VERIFY_INTVL_NM, 0, REG_DWORD, (LPBYTE)&dwData, sizeof(DWORD)); if( Status isnot NO_ERROR ) goto RETURN;
dwData = 1;
Status = RegSetValueEx(hParameter, WINSCNF_BURST_HANDLING_NM, 0, REG_DWORD, (LPBYTE)&dwData, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto RETURN;
dwData = 0x1f4;
Status = RegSetValueEx(hParameter, WINSCNF_BURST_QUE_SIZE_NM, 0, REG_DWORD, (LPBYTE)&dwData, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto RETURN;
dwData = 0;
Status = RegSetValueEx(hParameter, WINSCNF_LOG_DETAILED_EVTS_NM, 0, REG_DWORD, (LPBYTE)&dwData, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto RETURN;
dwData = 1;
Status = RegSetValueEx(hParameter, WINSCNF_LOG_FLAG_NM, 0, REG_DWORD, (LPBYTE)&dwData, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto RETURN;
Status = RegSetValueEx(hParameter, WINSCNF_LOG_FILE_PATH_NM, 0, REG_EXPAND_SZ, (LPBYTE)L"%windir%\\system32\\wins", (wcslen(L"%windir%\\system32\\wins")+1)*sizeof(WCHAR));
if( Status isnot NO_ERROR ) goto RETURN;
dwData = 0;
Status = RegSetValueEx(hParameter, WINSCNF_USE_SELF_FND_PNRS_NM, 0, REG_DWORD, (LPBYTE)&dwData, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto RETURN;
dwData = 0;
Status = RegSetValueEx(hParameter, WINSCNF_MIGRATION_ON_NM, 0, REG_DWORD, (LPBYTE)&dwData, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto RETURN;
dwData = 0x1; Status = RegSetValueEx(hParameter, WINSCNF_RPL_ONLY_W_CNF_PNRS_NM, 0, REG_DWORD, (LPBYTE)&dwData, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto RETURN; Status = RegSetValueEx(hParameter, WINSCNF_DB_FILE_NM, 0, REG_EXPAND_SZ, (LPBYTE)L"%windir%\\system32\\wins\\wins.mdb", (wcslen(L"%windir%\\system32\\wins\\wins.mdb")+1)*sizeof(WCHAR));
if( Status isnot NO_ERROR ) goto RETURN;
dwData = 2;
Status = RegSetValueEx(hParameter, WINSCNF_MCAST_TTL_NM, 0, REG_DWORD, (LPBYTE)&dwData, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto RETURN;
dwData = 0x960;
Status = RegSetValueEx(hParameter, WINSCNF_MCAST_INTVL_NM, 0, REG_DWORD, (LPBYTE)&dwData, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto RETURN;
dwData = 0;
Status = RegSetValueEx(hParameter, WINSCNF_INIT_VERSNO_VAL_HW_NM, 0, REG_DWORD, (LPBYTE)&dwData, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto RETURN;
dwData = 0;
Status = RegSetValueEx(hParameter, WINSCNF_INIT_VERSNO_VAL_LW_NM, 0, REG_DWORD, (LPBYTE)&dwData, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto RETURN;
dwData = 0x708;
Status = RegSetValueEx(hDefaultPull, WINSCNF_RPL_INTERVAL_NM, 0, REG_DWORD, (LPBYTE)&dwData, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto RETURN;
dwData = 0; Status = RegSetValueEx(hDefaultPush, WINSCNF_UPDATE_COUNT_NM, 0, REG_DWORD, (LPBYTE)&dwData, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto RETURN;
dwData = 0x3;
Status = RegSetValueEx(hPullPart, WINSCNF_RETRY_COUNT_NM, 0, REG_DWORD, (LPBYTE)&dwData, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto RETURN;
dwData = 0x1;
Status = RegSetValueEx(hPullPart, WINSCNF_INIT_TIME_RPL_NM, 0, REG_DWORD, (LPBYTE)&dwData, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto RETURN; dwData = 0x1;
Status = RegSetValueEx(hPullPart, WINSCNF_PRS_CONN_NM, 0, REG_DWORD, (LPBYTE)&dwData, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto RETURN;
dwData = 0x0;
Status = RegSetValueEx(hPushPart, WINSCNF_INIT_TIME_RPL_NM, 0, REG_DWORD, (LPBYTE)&dwData, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto RETURN; dwData = 0x1;
Status = RegSetValueEx(hPushPart, WINSCNF_PRS_CONN_NM, 0, REG_DWORD, (LPBYTE)&dwData, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto RETURN;
dwData = 0x0;
Status = RegSetValueEx(hPushPart, WINSCNF_ADDCHG_TRIGGER_NM, 0, REG_DWORD, (LPBYTE)&dwData, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto RETURN;
DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS);RETURN:
if( Status isnot NO_ERROR ) DisplayErrorMessage(EMSG_SRVR_SET_DEFAULTPARAM, Status); if( hPushPart ) { RegCloseKey(hPushPart); hPushPart = NULL; }
if( hPullPart ) { RegCloseKey(hPullPart); hPullPart = NULL; }
if( hPartner ) { RegCloseKey(hPartner); hPartner = NULL; }
if( hCheck ) { RegCloseKey(hCheck); hCheck = NULL; }
if( hDefaultPull ) { RegCloseKey(hDefaultPull); hDefaultPull = NULL; }
if( hDefaultPush ) { RegCloseKey(hDefaultPush); hDefaultPush = NULL; }
if( hDefault ) { RegCloseKey(hDefault); hDefault = NULL; }
if( hParameter ) { RegCloseKey(hParameter); hParameter = NULL; }
if( hWins ) { RegCloseKey(hWins); hWins = NULL; }
if( hServer ) { RegCloseKey(hServer); hServer = NULL; }
return Status;}
DWORDHandleSrvrSetMigrateflag( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Set the migrate on/off flagArguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : Set or disableReturn Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount; PDWORD pdwTagType = NULL, pdwTagNum = NULL;
TAG_TYPE pttTags[] = { {WINS_TOKEN_STATE, TRUE, FALSE}, };
BOOL fMigrate = FALSE; LPWSTR pTemp = NULL; HKEY hServer = NULL, hParameter = NULL;
if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, HLP_SRVR_SET_MIGRATEFLAG_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum);
if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
for( j=0; j<dwTagCount; j++ ) { switch(pdwTagType[j]) { case 0: { DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE or wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) < 1 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 0); if( dw is 1 ) fMigrate = TRUE; else if( dw is 0 ) fMigrate = FALSE; else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } }
{ if( wcslen(g_ServerNetBiosName) > 0 ) pTemp = g_ServerNetBiosName;
Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer); if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegCreateKeyEx(hServer, PARAMETER, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hParameter, NULL);
if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegSetValueEx(hParameter, WINSCNF_MIGRATION_ON_NM, 0, REG_DWORD, (LPBYTE)&fMigrate, sizeof(BOOL));
if( Status isnot NO_ERROR ) goto ErrorReturn; }
CommonReturn: if( Status is NO_ERROR ) { DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); }
if( hParameter ) { RegCloseKey(hParameter); hParameter = NULL; }
if( hServer ) { RegCloseKey(hServer); hServer = NULL; }
if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; } if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; } return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_SET_MIGRATEFLAG, Status); goto CommonReturn;
}
DWORDHandleSrvrSetNamerecord( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Set the name record parameters for the server.Arguments : All aguments are passes as array of wide char strings in ppwcArguments. Optional : Renew, extinction interval, extinction timeout and verification values. Note : All parameters are optional. Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount; PDWORD pdwTagType = NULL, pdwTagNum = NULL;
TAG_TYPE pttTags[] = { {WINS_TOKEN_RENEW, FALSE, FALSE}, {WINS_TOKEN_EXTINCTION, FALSE, FALSE}, {WINS_TOKEN_EXTIMEOUT, FALSE, FALSE}, {WINS_TOKEN_VERIFICATION, FALSE, FALSE}, };
DWORD dwRenew = SIX_DAYS, dwExtinction = SIX_DAYS, dwExTimeOut = SIX_DAYS, dwVerify = WINSCNF_MIN_VERIFY_INTERVAL;
BOOL fRenew = FALSE, fExtinction = FALSE, fExTimeOut = FALSE, fVerify = FALSE;
LPWSTR pTemp = NULL; HKEY hServer = NULL, hParameter = NULL; if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); return NO_ERROR; }
dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum);
if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
for( j=0; j<dwTagCount; j++ ) { switch(pdwTagType[j]) { case 0: { DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) < 1 ) { dwRenew = SIX_DAYS; } else { dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 10); dwRenew = dw; } fRenew = TRUE; break; } case 1: { DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) < 1 ) { dwExtinction = SIX_DAYS; } else { dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 0); dwExtinction = dw; } fExtinction = TRUE; break; } case 2: {
DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) < 1 ) { dwExTimeOut = SIX_DAYS; } else { dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 0); dwExTimeOut = dw; } fExTimeOut = TRUE; break; } case 3: { DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) < 1 ) { dwVerify = WINSCNF_MIN_VERIFY_INTERVAL; } else { dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 0); dwVerify = dw; } fVerify = TRUE; break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } }
{ DWORD dwValue = 0, dwType = REG_DWORD, dwSize = sizeof(DWORD); if( wcslen(g_ServerNetBiosName) > 0 ) pTemp = g_ServerNetBiosName;
Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer); if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegCreateKeyEx(hServer, PARAMETER, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hParameter, NULL);
if( Status isnot NO_ERROR ) goto ErrorReturn;
//First retrieve the older values for all the parameters
Status = RegQueryValueEx(hParameter, WINSCNF_REFRESH_INTVL_NM, NULL, &dwType, (LPBYTE)&dwValue, &dwSize);
if( Status is NO_ERROR ) { if( fRenew is FALSE ) { dwRenew = dwValue; } } dwSize = sizeof(DWORD);
Status = RegQueryValueEx(hParameter, WINSCNF_TOMBSTONE_INTVL_NM, NULL, &dwType, (LPBYTE)&dwValue, &dwSize); if( Status is NO_ERROR ) { if( fExtinction is FALSE ) dwExtinction = dwValue; }
dwSize = sizeof(DWORD);
Status = RegQueryValueEx(hParameter, WINSCNF_TOMBSTONE_TMOUT_NM, NULL, &dwType, (LPBYTE)&dwValue, &dwSize);
if( Status is NO_ERROR ) { if( fExTimeOut is FALSE ) dwExTimeOut = dwValue; }
dwSize = sizeof(DWORD);
Status = RegQueryValueEx(hParameter, WINSCNF_VERIFY_INTVL_NM, NULL, &dwType, (LPBYTE)&dwValue, &dwSize);
if( Status is NO_ERROR ) { if( fVerify is FALSE ) dwVerify = dwValue; }
//Check the validity and range of values
{ if( dwRenew < WINSCNF_MIN_REFRESH_INTERVAL ) { DisplayMessage(g_hModule, EMSG_SRVR_RENEW_INTERVAL); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } if( dwRenew > ONE_YEAR ) { dwRenew = ONE_YEAR; }
if( dwExTimeOut < WINSCNF_MIN_TOMBSTONE_TIMEOUT ) { DisplayMessage(g_hModule, EMSG_SRVR_TOMBSTONE_TIMEOUT, WINSCNF_MIN_TOMBSTONE_TIMEOUT); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } if( dwExTimeOut > ONE_YEAR ) { dwExTimeOut = ONE_YEAR; }
if( dwExTimeOut < dwRenew ) { DisplayMessage(g_hModule, EMSG_SRVR_TOMBSTONE_TIMEOUT, dwRenew); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
if( dwExtinction < WINSCNF_MAKE_TOMB_INTVL_0_M(dwRenew) ) { DisplayMessage(g_hModule, EMSG_SRVR_TOMBSTONE_INTERVAL, WINSCNF_MAKE_TOMB_INTVL_0_M(dwRenew) ); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
if( dwVerify < WINSCNF_MAKE_VERIFY_INTVL_M(dwExtinction) ) { if( WINSCNF_MAKE_VERIFY_INTVL_M(dwExtinction) == TWENTY_FOUR_DAYS ) { DisplayMessage(g_hModule, EMSG_SRVR_VERIFY_INTERVAL, TOMB_MULTIPLIER_FOR_VERIFY, WINSCNF_MAKE_VERIFY_INTVL_M(dwExtinction)); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
} }
Status = RegSetValueEx(hParameter, WINSCNF_REFRESH_INTVL_NM, 0, REG_DWORD, (LPBYTE)&dwRenew, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegSetValueEx(hParameter, WINSCNF_TOMBSTONE_INTVL_NM, 0, REG_DWORD, (LPBYTE)&dwExtinction, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegSetValueEx(hParameter, WINSCNF_TOMBSTONE_TMOUT_NM, 0, REG_DWORD, (LPBYTE)&dwExTimeOut, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegSetValueEx(hParameter, WINSCNF_VERIFY_INTVL_NM, 0, REG_DWORD, (LPBYTE)&dwVerify, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto ErrorReturn;
}
if( Status is NO_ERROR ) { DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); }CommonReturn:
if( hParameter ) { RegCloseKey(hParameter); hParameter = NULL; }
if( hServer ) { RegCloseKey(hServer); hServer = NULL; }
if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; } if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; } return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_SET_NAMERECORD, Status); goto CommonReturn; }
DWORDHandleSrvrSetPeriodicdbchecking( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Set the periodic database checking parameters for the serverArguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : State Optional : Maximum record count, and other parameters.Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount; PDWORD pdwTagType = NULL, pdwTagNum = NULL;
TAG_TYPE pttTags[] = { {WINS_TOKEN_STATE, TRUE, FALSE}, {WINS_TOKEN_MAXRECORDCOUNT, FALSE, FALSE}, {WINS_TOKEN_CHECKAGAINST, FALSE, FALSE}, {WINS_TOKEN_CHECKEVERY, FALSE, FALSE}, {WINS_TOKEN_START, FALSE, FALSE}, };
DWORD dwMaxRec = WINSCNF_CC_DEF_RECS_AAT, dwEvery = WINSCNF_CC_DEF_INTERVAL, dwStart = WINSCNF_DEF_CC_SP_HR*60*60;
BOOL fPartner = WINSCNF_CC_DEF_USE_RPL_PNRS, fIsPartner = FALSE, fMax = FALSE, fEvery = FALSE, fStart = FALSE, fState = TRUE;
LPWSTR pTemp = NULL; HKEY hServer = NULL, hCCRoot = NULL;
if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, HLP_SRVR_SET_PERIODICDBCHECKING_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum);
if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
for( j=0; j<dwTagCount; j++ ) { switch(pdwTagType[j]) { case 0: { DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 0); if( dw is 0 ) fState = FALSE; else fState = TRUE; break; } case 1: { DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 0); dwMaxRec = dw; fMax = TRUE; break; } case 2: { DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 0); if( dw is 1 ) fPartner = TRUE; else fPartner = FALSE; fIsPartner = TRUE; break; } case 3: { DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 0); dwEvery = dw*60*60; fEvery = TRUE; break; } case 4: { DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 0); dwStart = dw; fStart = TRUE; break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } }
if( fState is FALSE ) { if( wcslen(g_ServerNetBiosName) > 0 ) pTemp = g_ServerNetBiosName;
Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer); if( Status isnot NO_ERROR ) goto ErrorReturn;
RegDeleteKey(hServer, CCROOT); } else { if( fMax is TRUE or fEvery is TRUE or fIsPartner is TRUE or fStart is TRUE ) { if( wcslen(g_ServerNetBiosName) > 0 ) pTemp = g_ServerNetBiosName;
Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer); if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegCreateKeyEx(hServer, CCROOT, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hCCRoot, NULL);
if( Status isnot NO_ERROR ) goto ErrorReturn;
if( fMax ) { Status = RegSetValueEx(hCCRoot, WINSCNF_CC_MAX_RECS_AAT_NM, 0, REG_DWORD, (LPBYTE)&dwMaxRec, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto ErrorReturn; }
if( fIsPartner ) { Status = RegSetValueEx(hCCRoot, WINSCNF_CC_USE_RPL_PNRS_NM, 0, REG_DWORD, (LPBYTE)&fPartner, sizeof(BOOL));
if( Status isnot NO_ERROR ) goto ErrorReturn; } if( fEvery ) { Status = RegSetValueEx(hCCRoot, WINSCNF_CC_INTVL_NM, 0, REG_DWORD, (LPBYTE)&dwEvery, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto ErrorReturn; }
if( fStart ) { LPWSTR pwszTime = MakeTimeString(dwStart); if( pwszTime is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } Status = RegSetValueEx(hCCRoot, WINSCNF_SP_TIME_NM, 0, REG_SZ, (LPBYTE)pwszTime, (wcslen(pwszTime)+1)*sizeof(WCHAR));
WinsFreeMemory(pwszTime); pwszTime = NULL; if( Status isnot NO_ERROR ) goto ErrorReturn; }
} }
CommonReturn: if( Status is NO_ERROR ) { DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); }
if( hCCRoot ) { RegCloseKey(hCCRoot); hCCRoot = NULL; }
if( hServer ) { RegCloseKey(hServer); hServer = NULL; }
if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; } if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; } return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_SET_PERIODICDBCHECKING, Status); if( hServer ) { if( hCCRoot ) { RegCloseKey(hCCRoot); hCCRoot = NULL; } RegDeleteKey(hServer, CCROOT); } goto CommonReturn; }
DWORDHandleSrvrSetPullpersistentconnection( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Set the pull partner configuration parametersArguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : Server IP and persistence state Optional : Start value and Time interval Note : All parameters are optional. Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount; PDWORD pdwTagType = NULL, pdwTagNum = NULL;
TAG_TYPE pttTags[] = { {WINS_TOKEN_STATE, TRUE, FALSE}, {WINS_TOKEN_SERVER, TRUE, FALSE}, {WINS_TOKEN_START, FALSE, FALSE}, {WINS_TOKEN_INTERVAL, FALSE, FALSE}, };
DWORD dwStart = 0, dwInterval = 1800;
WCHAR wcServerIpAdd[MAX_IP_STRING_LEN + 1] = {L'\0'};
BOOL fState = TRUE, fStart = FALSE, fInterval = FALSE;
LPWSTR pTemp = NULL; HKEY hServer = NULL, hPullRoot = NULL, hPullServer = NULL;
if( dwArgCount < dwCurrentIndex + 2 ) { DisplayMessage(g_hModule, HLP_SRVR_SET_PULLPERSISTENTCONNECTION_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum);
if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
for( j=0; j<dwTagCount; j++ ) { switch(pdwTagType[j]) { case 0: { WCHAR wc = ppwcArguments[dwCurrentIndex+pdwTagNum[j]][0]; if( wc is L'0' ) fState = FALSE; else if ( wc is L'1' ) fState = TRUE; else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } break; } case 1: { struct hostent * lpHostEnt = NULL; CHAR cAddr[16]; BYTE pbAdd[4]; char szAdd[4]; int k = 0, l=0; DWORD dwLen, nLen = 0;
if( IsIpAddress( ppwcArguments[dwCurrentIndex+pdwTagNum[j]] ) ) { DWORD dwIp = inet_addr(WinsUnicodeToAnsi(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL)); lpHostEnt = gethostbyaddr((char *)&dwIp, 4, AF_INET); if(lpHostEnt isnot NULL )//Valid IP Address
{ wcscpy(wcServerIpAdd, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); break; } else { Status = ERROR_INVALID_IPADDRESS; goto ErrorReturn; } }
if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) > 2 and _wcsnicmp(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], L"\\\\", 2) is 0 ) k = 2;
lpHostEnt = gethostbyname(WinsUnicodeToAnsi(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]+k, NULL)); if( lpHostEnt is NULL ) { DisplayMessage(g_hModule, EMSG_WINS_INVALID_COMPUTER_NAME, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]+k); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
memcpy(pbAdd, lpHostEnt->h_addr_list[0], 4); for( l=0; l<4; l++) { _itoa((int)pbAdd[l], szAdd, 10); memcpy(cAddr+nLen, szAdd, strlen(szAdd)); nLen += strlen(szAdd); *(cAddr+nLen) = '.'; nLen++;
} *(cAddr+nLen-1) = '\0'; WinsAnsiToUnicode(cAddr, wcServerIpAdd); break; } case 2: { DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 0); dwStart = dw; fStart = TRUE; break; } case 3: { DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 0); dwInterval = dw; fInterval = TRUE; break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } }
{ DWORD dwDisposition = 0; if( wcslen(g_ServerNetBiosName) > 0 ) pTemp = g_ServerNetBiosName;
Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer); if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegOpenKeyEx(hServer, PULLROOT, 0, KEY_ALL_ACCESS, &hPullRoot);
if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegOpenKeyEx(hPullRoot, wcServerIpAdd, 0, KEY_ALL_ACCESS, &hPullServer);
if( Status is ERROR_FILE_NOT_FOUND ) { DisplayMessage(g_hModule, EMSG_SRVR_INVALID_PARTNER, wcServerIpAdd); goto CommonReturn; } if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegSetValueEx(hPullServer, PERSISTENCE, 0, REG_DWORD, (LPBYTE)&fState, sizeof(BOOL)); if( Status isnot NO_ERROR ) { goto ErrorReturn; } if( fInterval ) { Status = RegSetValueEx(hPullServer, WINSCNF_RPL_INTERVAL_NM, 0, REG_DWORD, (LPBYTE)&dwInterval, sizeof(DWORD));
if( Status isnot NO_ERROR ) { goto ErrorReturn; } } if( fStart ) { LPWSTR pwszTime = MakeTimeString(dwStart); if( pwszTime is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } Status = RegSetValueEx(hPullServer, WINSCNF_SP_TIME_NM, 0, REG_SZ, (LPBYTE)pwszTime, (wcslen(pwszTime)+1)*sizeof(WCHAR));
WinsFreeMemory(pwszTime); pwszTime = NULL; if( Status isnot NO_ERROR ) { goto ErrorReturn; } }
} CommonReturn: if( Status is NO_ERROR ) { DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); }
if( hPullServer ) { RegCloseKey(hPullServer); hPullServer = NULL; }
if( hPullRoot ) { RegCloseKey(hPullRoot); hPullRoot = NULL; }
if( hServer ) { RegCloseKey(hServer); hServer = NULL; }
if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; } if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; } return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_SET_PULLPERSISTENTCONNECTION, Status); goto CommonReturn; }
DWORDHandleSrvrSetPushpersistentconnection( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Set the push partner configuration parametersArguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : Server IP and persistence state Optional : Update count Note : All parameters are optional. Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount; PDWORD pdwTagType = NULL, pdwTagNum = NULL;
TAG_TYPE pttTags[] = { {WINS_TOKEN_STATE, TRUE, FALSE}, {WINS_TOKEN_SERVER, TRUE, FALSE}, {WINS_TOKEN_UPDATE,FALSE, FALSE}, };
DWORD dwUpdate = 0;
WCHAR wcServerIpAdd[MAX_IP_STRING_LEN + 1] = {L'\0'};
BOOL fState = TRUE, fUpdate = FALSE;
LPWSTR pTemp = NULL; HKEY hServer = NULL, hPushRoot = NULL, hPushServer = NULL;
if( dwArgCount < dwCurrentIndex + 2 ) { DisplayMessage(g_hModule, HLP_SRVR_SET_PUSHPERSISTENTCONNECTION_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum);
if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
for( j=0; j<dwTagCount; j++ ) { switch(pdwTagType[j]) { case 0: { WCHAR wc = ppwcArguments[dwCurrentIndex+pdwTagNum[j]][0]; if( wc is L'0' ) fState = FALSE; else if ( wc is L'1' ) fState = TRUE; else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } break; } case 1: { struct hostent * lpHostEnt = NULL; CHAR cAddr[16]; BYTE pbAdd[4]; char szAdd[4]; int k = 0, l=0; DWORD dwLen, nLen = 0;
if( IsIpAddress( ppwcArguments[dwCurrentIndex+pdwTagNum[j]] ) ) { DWORD dwIp = inet_addr(WinsUnicodeToAnsi(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL)); lpHostEnt = gethostbyaddr((char *)&dwIp, 4, AF_INET); if(lpHostEnt isnot NULL )//Valid IP Address
{ wcscpy(wcServerIpAdd, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); break; } else { Status = ERROR_INVALID_IPADDRESS; goto ErrorReturn; } }
if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) > 2 and _wcsnicmp(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], L"\\\\", 2) is 0 ) k = 2;
lpHostEnt = gethostbyname(WinsUnicodeToAnsi(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]+k, NULL)); if( lpHostEnt is NULL ) { DisplayMessage(g_hModule, EMSG_WINS_INVALID_COMPUTER_NAME, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]+k); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
memcpy(pbAdd, lpHostEnt->h_addr_list[0], 4); for( l=0;l<4; l++) { _itoa((int)pbAdd[l], szAdd, 10); memcpy(cAddr+nLen, szAdd, strlen(szAdd)); nLen += strlen(szAdd); *(cAddr+nLen) = '.'; nLen++;
} *(cAddr+nLen-1) = '\0'; WinsAnsiToUnicode(cAddr, wcServerIpAdd); break; } case 2: { DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 0); dwUpdate = dw; fUpdate = TRUE; break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } }
{ DWORD dwDisposition = 0; if( wcslen(g_ServerNetBiosName) > 0 ) pTemp = g_ServerNetBiosName;
Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer); if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegOpenKeyEx(hServer, PUSHROOT, 0, KEY_ALL_ACCESS, &hPushRoot);
if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegOpenKeyEx(hPushRoot, wcServerIpAdd, 0, KEY_ALL_ACCESS, &hPushServer);
if( Status is ERROR_FILE_NOT_FOUND ) { DisplayMessage(g_hModule, EMSG_SRVR_INVALID_PARTNER, wcServerIpAdd); goto CommonReturn; }
if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegSetValueEx(hPushServer, PERSISTENCE, 0, REG_DWORD, (LPBYTE)&fState, sizeof(BOOL)); if( Status isnot NO_ERROR ) { goto ErrorReturn; } if( fUpdate ) { Status = RegSetValueEx(hPushServer, WINSCNF_UPDATE_COUNT_NM, 0, REG_DWORD, (LPBYTE)&dwUpdate, sizeof(DWORD));
if( Status isnot NO_ERROR ) { goto ErrorReturn; } } } CommonReturn: if( Status is NO_ERROR ) { DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); }
if( hPushServer ) { RegCloseKey(hPushServer); hPushServer = NULL; }
if( hPushRoot ) { RegCloseKey(hPushRoot); hPushRoot = NULL; }
if( hServer ) { RegCloseKey(hServer); hServer = NULL; }
if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; } if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; } return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_SET_PUSHPERSISTENTCONNECTION, Status); goto CommonReturn; }
DWORDHandleSrvrSetPullparam( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Set the default pull parametersArguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : persistence state Optional : Startup value, Start time and Time interval and retry count Note : All parameters are optional. Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount; PDWORD pdwTagType = NULL, pdwTagNum = NULL;
TAG_TYPE pttTags[] = { {WINS_TOKEN_STATE, TRUE, FALSE}, {WINS_TOKEN_STARTUP, FALSE, FALSE}, {WINS_TOKEN_START, FALSE, FALSE}, {WINS_TOKEN_INTERVAL, FALSE, FALSE}, {WINS_TOKEN_RETRY, FALSE, FALSE}, };
DWORD dwStart = 0, dwInterval = 1800, dwRetry = 3;
BOOL fStart = FALSE, fStartup = FALSE, fInterval = FALSE, fRetry = FALSE, fState = TRUE;
LPWSTR pTemp = NULL; HKEY hServer = NULL, hDefault = NULL, hPullRoot = NULL;
if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, HLP_SRVR_SET_PULLPARAM_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum);
if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
for( j=0; j<dwTagCount; j++ ) { switch(pdwTagType[j]) { case 0: { WCHAR wc = ppwcArguments[dwCurrentIndex+pdwTagNum[j]][0]; if( wc is L'0' ) fState = FALSE; else if ( wc is L'1' ) fState = TRUE; else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } break; } case 1: { WCHAR wc = ppwcArguments[dwCurrentIndex+pdwTagNum[j]][0]; if( wc is L'1' ) { fStartup = TRUE; } else if ( wc is L'0' ) { fStartup = FALSE; } else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } break; } case 2: { DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 0); dwStart = dw; fStart = TRUE; break; } case 3: { DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 0);
if( dw > 0 && dw < 60 ) // 1 minute
{ Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dwInterval = dw; fInterval = TRUE; break; } case 4: { DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 0); dwRetry = dw; fRetry = TRUE; break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } } { if( wcslen(g_ServerNetBiosName) > 0 ) pTemp = g_ServerNetBiosName;
Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer); if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegCreateKeyEx(hServer, PULLROOT, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hPullRoot, NULL);
if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegSetValueEx(hPullRoot, PERSISTENCE, 0, REG_DWORD, (LPBYTE)&fState, sizeof(BOOL));
if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegSetValueEx(hPullRoot, WINSCNF_INIT_TIME_RPL_NM, 0, REG_DWORD, (LPBYTE)&fStartup, sizeof(BOOL));
if( Status isnot NO_ERROR ) goto ErrorReturn;
if( fRetry ) { Status = RegSetValueEx(hPullRoot, WINSCNF_RETRY_COUNT_NM, 0, REG_DWORD, (LPBYTE)&dwRetry, sizeof(DWORD)); if( Status isnot NO_ERROR ) goto ErrorReturn; }
if( fStart is TRUE or fInterval is TRUE ) { Status = RegCreateKeyEx(hServer, DEFAULTPULL, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hDefault, NULL);
if( Status isnot NO_ERROR ) goto ErrorReturn;
if( fStart ) { LPWSTR pwszTime = MakeTimeString(dwStart); if( pwszTime is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } Status = RegSetValueEx(hDefault, WINSCNF_SP_TIME_NM, 0, REG_SZ, (LPBYTE)pwszTime, (wcslen(pwszTime)+1)*sizeof(WCHAR)); WinsFreeMemory(pwszTime); pwszTime = NULL; if( Status isnot NO_ERROR ) goto ErrorReturn; }
if( fInterval ) { Status = RegSetValueEx(hDefault, WINSCNF_RPL_INTERVAL_NM, 0, REG_DWORD, (LPBYTE)&dwInterval, sizeof(DWORD)); if( Status isnot NO_ERROR ) goto ErrorReturn; } } }
CommonReturn: if( Status is NO_ERROR ) { DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); }
if( hPullRoot ) { RegCloseKey(hPullRoot); hPullRoot = NULL; }
if( hDefault ) { RegCloseKey(hDefault); hDefault = NULL; } if( hServer ) { RegCloseKey(hServer); hServer = NULL; }
if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; } if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; } return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_SET_PULLPARAM, Status); goto CommonReturn; }
DWORDHandleSrvrSetPushparam( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Set the default push parametersArguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : Server IP and persistence state Optional : Start value and Time intervalReturn Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount; PDWORD pdwTagType = NULL, pdwTagNum = NULL;
TAG_TYPE pttTags[] = { {WINS_TOKEN_STATE, TRUE, FALSE}, {WINS_TOKEN_STARTUP, FALSE, FALSE}, {WINS_TOKEN_ADDRESSCHANGE, FALSE, FALSE}, {WINS_TOKEN_UPDATE, FALSE, FALSE}, };
DWORD dwUpdate = 3;
BOOL fStartup = FALSE, IsStartup = FALSE, fAddressChange = FALSE, IsAddChange = FALSE, fAdd = FALSE, IsAdd = FALSE, fUpdate = FALSE, IsUpdate = FALSE, fState = TRUE;
LPWSTR pTemp = NULL; HKEY hServer = NULL, hDefault = NULL, hPushRoot = NULL;
if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, HLP_SRVR_SET_PUSHPARAM_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum);
if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
for( j=0; j<dwTagCount; j++ ) { switch(pdwTagType[j]) { case 0: { WCHAR wc = ppwcArguments[dwCurrentIndex+pdwTagNum[j]][0]; if( wc is L'0' ) fState = FALSE; else if ( wc is L'1' ) fState = TRUE; else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } break; } case 1: { WCHAR wc ; IsStartup = TRUE; if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) < 1 ) break; wc = ppwcArguments[dwCurrentIndex+pdwTagNum[j]][0]; if( wc is L'1' ) { fStartup = TRUE; } else if( wc is L'0' ) fStartup = 0; else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } break; } case 2: { WCHAR wc ; IsAddChange = TRUE; fAdd = TRUE; if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) < 1 ) break; wc = ppwcArguments[dwCurrentIndex+pdwTagNum[j]][0]; if( wc is L'1' ) fAddressChange = TRUE; else if( wc is L'0' ) fAddressChange = FALSE; else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } break; } case 3: { DWORD dw = 0; IsUpdate = TRUE; fUpdate = TRUE; if( wcslen( ppwcArguments[dwCurrentIndex+pdwTagNum[j]] ) < 1 ) break; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 0); if( dw < 1000 ) dwUpdate = dw; else dwUpdate = 999;
break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } } { if( wcslen(g_ServerNetBiosName) > 0 ) pTemp = g_ServerNetBiosName;
Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer); if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegCreateKeyEx(hServer, PUSHROOT, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hPushRoot, NULL);
if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegSetValueEx(hPushRoot, PERSISTENCE, 0, REG_DWORD, (LPBYTE)&fState, sizeof(BOOL));
if( Status isnot NO_ERROR ) goto ErrorReturn;
if( IsStartup ) { Status = RegSetValueEx(hPushRoot, WINSCNF_INIT_TIME_RPL_NM, 0, REG_DWORD, (LPBYTE)&fStartup, sizeof(BOOL));
if( Status isnot NO_ERROR ) goto ErrorReturn; }
if( fAdd ) { Status = RegSetValueEx(hPushRoot, WINSCNF_ADDCHG_TRIGGER_NM, 0, REG_DWORD, (LPBYTE)&fAddressChange, sizeof(DWORD)); if( Status isnot NO_ERROR ) goto ErrorReturn; }
if( fUpdate is TRUE ) { Status = RegCreateKeyEx(hServer, DEFAULTPUSH, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hDefault, NULL);
if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegSetValueEx(hDefault, WINSCNF_UPDATE_COUNT_NM, 0, REG_DWORD, (LPBYTE)&dwUpdate, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto ErrorReturn; } }
CommonReturn: if( Status is NO_ERROR ) { DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); }
if( hPushRoot ) { RegCloseKey(hPushRoot); hPushRoot = NULL; }
if( hDefault ) { RegCloseKey(hDefault); hDefault = NULL; } if( hServer ) { RegCloseKey(hServer); hServer = NULL; }
if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; } if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; } return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_SET_PUSHPARAM, Status); goto CommonReturn; }
DWORDHandleSrvrSetReplicateflag( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Set the replication flagArguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : Flag stateReturn Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount; PDWORD pdwTagType = NULL, pdwTagNum = NULL;
TAG_TYPE pttTags[] = { {WINS_TOKEN_STATE, TRUE, FALSE}, };
BOOL fReplicate = FALSE; LPWSTR pTemp = NULL; HKEY hServer = NULL, hParameter = NULL;
if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, HLP_SRVR_SET_REPLICATEFLAG_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum);
if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
for( j=0; j<dwTagCount; j++ ) { switch(pdwTagType[j]) { case 0: { DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 0); if( dw is 1 ) fReplicate = TRUE; else if( dw is 0 ) fReplicate = FALSE; else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } }
{ if( wcslen(g_ServerNetBiosName) > 0 ) pTemp = g_ServerNetBiosName;
Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer); if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegCreateKeyEx(hServer, PARAMETER, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hParameter, NULL);
if( Status isnot NO_ERROR ) goto ErrorReturn; Status = RegSetValueEx(hParameter, WINSCNF_RPL_ONLY_W_CNF_PNRS_NM, 0, REG_DWORD, (LPBYTE)&fReplicate, sizeof(BOOL));
if( Status isnot NO_ERROR ) goto ErrorReturn; }
CommonReturn: if( Status is NO_ERROR ) { DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); }
if( hParameter ) { RegCloseKey(hParameter); hParameter = NULL; }
if( hServer ) { RegCloseKey(hServer); hServer = NULL; }
if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; } if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; } return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_SET_REPLICATEFLAG, Status); goto CommonReturn; }
DWORDHandleSrvrSetLogparam( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Set the logging parametersArguments : All aguments are passes as array of wide char strings in ppwcArguments. Optional : Log database change and detail event log options Note : All parameters are optional. Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount; TAG_TYPE pttTags[] = { {WINS_TOKEN_DBCHANGE, FALSE, FALSE}, {WINS_TOKEN_EVENT, FALSE, FALSE}, }; PDWORD pdwTagType = NULL, pdwTagNum = NULL;
BOOL fDbChange = FALSE, IsDbChange = FALSE, fEvent = FALSE, IsEvent = FALSE;
if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); goto CommonReturn; }
dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum);
if( Status isnot NO_ERROR ) goto ErrorReturn;
for( j=0; j<dwTagCount; j++ ) { switch(pdwTagType[j]) { case 0: { WCHAR wc = ppwcArguments[dwCurrentIndex+pdwTagNum[j]][0];
if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is 0 ) { fDbChange = FALSE; IsDbChange = TRUE; break; } if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) isnot 1 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } if( wc is L'0' ) { fDbChange = FALSE; } else if( wc is L'1' ) { fDbChange = TRUE; } else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
IsDbChange = TRUE; break;
} case 1: { WCHAR wc = ppwcArguments[dwCurrentIndex+pdwTagNum[j]][0]; if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is 0 ) { fEvent = FALSE; IsEvent = TRUE; break; }
if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) isnot 1 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } if( wc is L'0' ) { fEvent = FALSE; } else if( wc is L'1' ) { fEvent = TRUE; } else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
IsEvent = TRUE; break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } }
{ HKEY hServer = NULL, hParameter = NULL; LPWSTR pTemp = NULL; DWORD dwType = REG_DWORD, dwSize = sizeof(BOOL);
if( wcslen(g_ServerNetBiosName) > 0 ) pTemp = g_ServerNetBiosName;
Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer);
if( Status isnot NO_ERROR ) goto ErrorReturn; Status = RegOpenKeyEx(hServer, PARAMETER, 0, KEY_ALL_ACCESS, &hParameter);
if( Status isnot NO_ERROR ) { RegCloseKey(hServer); hServer = NULL; goto ErrorReturn; } if( IsDbChange ) { Status = RegSetValueEx(hParameter, WINSCNF_LOG_FLAG_NM, 0, dwType, (LPBYTE)&fDbChange, dwSize);
if( Status isnot NO_ERROR ) { RegCloseKey(hServer); hServer = NULL; RegCloseKey(hParameter); hParameter = NULL; goto ErrorReturn; } }
if( IsEvent ) { Status = RegSetValueEx(hParameter, WINSCNF_LOG_DETAILED_EVTS_NM, 0, dwType, (LPBYTE)&fEvent, dwSize);
if( Status isnot NO_ERROR ) { RegCloseKey(hServer); hServer = NULL; RegCloseKey(hParameter); hParameter = NULL; goto ErrorReturn; } } if( hServer ) { RegCloseKey(hServer); hServer = NULL; }
if( hParameter ) { RegCloseKey(hParameter); hParameter = NULL; }
}
DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS);CommonReturn: if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; }
if( pdwTagType ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; }
return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_SET_LOGPARAM, Status); goto CommonReturn;}
DWORDHandleSrvrSetBurstparam( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Set the burst handling parametersArguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : To enable or disable burst handling Optional : Burst handling value.Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount; TAG_TYPE pttTags[] = { {WINS_TOKEN_STATE, TRUE, FALSE}, {WINS_TOKEN_VALUE, FALSE, FALSE}, }; PDWORD pdwTagType = NULL, pdwTagNum = NULL;
BOOL fState = FALSE, fValue = FALSE;
DWORD dwValue = 0;
if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, HLP_SRVR_SET_BURSTPARAM_EX);
Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum);
if( Status isnot NO_ERROR ) goto ErrorReturn;
for( j=0; j<dwTagCount; j++ ) { switch(pdwTagType[j]) { case 0: { WCHAR wc = ppwcArguments[dwCurrentIndex+pdwTagNum[j]][0]; if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is 0 ) { fState = FALSE; break; } if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) isnot 1 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } if( wc is L'0' ) { fState = FALSE; } else if( wc is L'1' ) { fState = TRUE; } else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } break;
} case 1: { if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dwValue = wcstoul(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 10);
if( dwValue < 50 or dwValue > 5000 ) { DisplayMessage(g_hModule, EMSG_SRVR_BURST_PARAM_OUTOFRANGE); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn;
}
fValue = TRUE; break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } }
{ HKEY hServer = NULL, hParameter = NULL; LPWSTR pTemp = NULL; DWORD dwType = REG_DWORD, dwSize = sizeof(BOOL);
if( wcslen(g_ServerNetBiosName) > 0 ) pTemp = g_ServerNetBiosName;
Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer);
if( Status isnot NO_ERROR ) goto ErrorReturn; Status = RegOpenKeyEx(hServer, PARAMETER, 0, KEY_ALL_ACCESS, &hParameter);
if( Status isnot NO_ERROR ) { RegCloseKey(hServer); hServer = NULL; goto ErrorReturn; } Status = RegSetValueEx(hParameter, WINSCNF_BURST_HANDLING_NM, 0, dwType, (LPBYTE)&fState, dwSize);
if( Status isnot NO_ERROR ) { RegCloseKey(hServer); hServer = NULL; RegCloseKey(hParameter); hParameter = NULL; goto ErrorReturn; }
if( fValue ) { dwSize = sizeof(DWORD); Status = RegSetValueEx(hParameter, WINSCNF_BURST_QUE_SIZE_NM, 0, dwType, (LPBYTE)&dwValue, dwSize);
if( Status isnot NO_ERROR ) { RegCloseKey(hServer); hServer = NULL; RegCloseKey(hParameter); hParameter = NULL; goto ErrorReturn; } } if( hServer ) { RegCloseKey(hServer); hServer = NULL; }
if( hParameter ) { RegCloseKey(hParameter); hParameter = NULL; }
}
DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS);CommonReturn: if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; }
if( pdwTagType ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; }
return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_SET_BURSTPARAM, Status); goto CommonReturn;}
DWORDHandleSrvrSetStartversion( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Set start value of the version counterArguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : Version counter value in {high,low} formatReturn Value: Returns the status of the operation.
--*/{ DWORD Status = 0; DWORD i, j, dwTagCount, dwNumArgs; TAG_TYPE pttTags[] = { {WINS_TOKEN_VERSION, TRUE, FALSE}, }; WINSINTF_VERS_NO_T Version={0}; LPWSTR pServer = NULL; PDWORD pdwTagNum = NULL, pdwTagType = NULL;
HKEY hServer = NULL, hParameter = NULL;
if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, HLP_SRVR_SET_STARTVERSION_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum);
if( Status isnot NO_ERROR ) goto ErrorReturn;
for( j=0; j<dwTagCount; j++ ) { switch(pdwTagType[0]) { case 0: { Status = GetVersionData(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], &Version); if( Status isnot NO_ERROR ) goto ErrorReturn; break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } } if( wcslen(g_ServerNetBiosName) > 0 ) pServer = g_ServerNetBiosName;
Status = RegConnectRegistry(pServer, HKEY_LOCAL_MACHINE, &hServer);
if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegOpenKeyEx(hServer, PARAMETER, 0, KEY_ALL_ACCESS, &hParameter);
if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegSetValueEx(hParameter, WINSCNF_INIT_VERSNO_VAL_HW_NM, 0, REG_DWORD, (LPBYTE)&Version.HighPart, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegSetValueEx(hParameter, WINSCNF_INIT_VERSNO_VAL_LW_NM, 0, REG_DWORD, (LPBYTE)&Version.LowPart, sizeof(DWORD));
if( Status isnot NO_ERROR ) goto ErrorReturn;
DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS);
CommonReturn:
if( hServer ) { RegCloseKey(hServer); hServer = NULL; }
if( hParameter ) { RegCloseKey(hParameter); hParameter = NULL; }
if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; }
if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; }
return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_SET_STARTVERSION, Status); goto CommonReturn;}
DWORDHandleSrvrSetPersMode( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Set Persona Grata/Non-Grata modeArguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : Grat|Non-GrataReturn Value: Returns the status of the operation.
--*/{ DWORD Status = 0; DWORD dwTagCount; DWORD dwPersMode; PDWORD pdwTagNum = NULL; PDWORD pdwTagType = NULL; TAG_TYPE pttTags[] = {{WINS_TOKEN_MODE, TRUE, FALSE},}; LPWSTR lpwszMode; LPWSTR pTemp; HKEY hServer, hPartner;
if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, HLP_SRVR_SET_PGMODE_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
pdwTagType = WinsAllocateMemory((dwArgCount - dwCurrentIndex)*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory((dwArgCount - dwCurrentIndex)*sizeof(DWORD));
if( pdwTagType is NULL || pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum);
if( Status isnot NO_ERROR ) goto ErrorReturn;
lpwszMode = ppwcArguments[dwCurrentIndex+pdwTagNum[0]]; if (wcslen(lpwszMode) == 1 && (lpwszMode[0] == L'0' || lpwszMode[0] == L'1')) { // set the value for 'persona grata mode'
dwPersMode = lpwszMode[0] == L'0' ? PERSMODE_NON_GRATA : PERSMODE_GRATA; } else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
if( wcslen(g_ServerNetBiosName) > 0 ) { pTemp = g_ServerNetBiosName; }
Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer);
if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegOpenKeyEx(hServer, PARTNERROOT, 0, KEY_ALL_ACCESS, &hPartner);
if (Status isnot NO_ERROR) goto ErrorReturn;
Status = RegSetValueExA( hPartner, WINSCNF_PERSONA_MODE_NM, 0, REG_DWORD, (LPVOID)&dwPersMode, sizeof(DWORD));
DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS);
CommonReturn: if( hServer ) { RegCloseKey(hServer); hServer = NULL; }
if( hPartner ) { RegCloseKey(hPartner); hPartner = NULL; }
if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; }
if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; }
return Status;
ErrorReturn: DisplayErrorMessage(EMSG_SRVR_SET_PGMODE, Status); goto CommonReturn; }
DWORDHandleSrvrShowDatabase( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Displays wins database based on (optionally)different filtering conditionsArguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : Server ip whose database to be displayed Optional : Different conditionsReturn Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; WINSINTF_ADD_T WinsAdd = {0}; TAG_TYPE pttTags[] = { {WINS_TOKEN_SERVERS, TRUE, FALSE}, {WINS_TOKEN_RECORDTYPE, FALSE, FALSE}, {WINS_TOKEN_RECCOUNT, FALSE, FALSE}, {WINS_TOKEN_START, FALSE, FALSE}, {WINS_TOKEN_ENDCHAR, FALSE, FALSE}, {WINS_TOKEN_FILE, FALSE, FALSE}, }; PDWORD pdwTagType = NULL, pdwTagNum = NULL, pdwIp = NULL; WCHAR wcFilter = L'\0'; CHAR chFilter = 0x00; WCHAR wcServerIpAdd[MAX_IP_STRING_LEN+1]={L'\0'}; WCHAR wcFile[MAX_PATH] = {L'\0'}; DWORD i, j, dwNumArgs, dwTagCount; DWORD dwStart = WINSINTF_BEGINNING; DWORD NoOfRecsDesired = (DWORD)~0; DWORD TypeOfRec, dwIpCount = 0, dwRecCount = 0, dwTotal = 0, dw=0; WINSINTF_RECS_T Recs = {0}; PWINSINTF_RECORD_ACTION_T pRow = NULL; BOOL fFilter = FALSE, fFile = FALSE; BOOL fAll = FALSE, fHeader = FALSE, fError = FALSE, fOnce = FALSE, fNew = TRUE; WINSINTF_RESULTS_T Results = {0}; WINSINTF_RESULTS_NEW_T ResultsN = {0}; FILE *pFile = NULL; BOOL fOpenFile = FALSE; WCHAR wszFilter[3] = {L'\0'}; LPWSTR pwszTemp = NULL; LPWSTR pwszTime = NULL;
NoOfRecsDesired = (DWORD)~0; TypeOfRec = WINSINTF_BOTH; if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, HLP_SRVR_SHOW_DATABASE_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } { dwNumArgs = dwArgCount - dwCurrentIndex; pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE); Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum); if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
for( j=0; j<dwTagCount; j++ ) { switch(pdwTagType[j]) { case 0: { LPWSTR pwcTemp = NULL, pwcTag = L",\r\n", pwcToken = NULL, pwszIps = NULL; DWORD dwLen = 0;
dwLen = wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]);
if( ppwcArguments[dwCurrentIndex+pdwTagNum[j]][0] isnot L'{' or ppwcArguments[dwCurrentIndex+pdwTagNum[j]][dwLen-1] isnot L'}') { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } pwcTemp = ppwcArguments[dwCurrentIndex+pdwTagNum[j]] + 1; dwLen--; ppwcArguments[dwCurrentIndex+pdwTagNum[j]][dwLen] = L'\0'; dwLen--; if( dwLen <= 0 ) { fAll = TRUE; break; } if( dwLen < 7 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
pwszIps = WinsAllocateMemory(dwLen*sizeof(WCHAR));
if( pwszIps is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } wcscpy(pwszIps, pwcTemp); pwcToken = wcstok(pwszIps, pwcTag);
dwIpCount = 0;
while( pwcToken isnot NULL ) { PDWORD pdwTemp = NULL;
if( IsIpAddress( pwcToken ) is FALSE ) { Status = ERROR_INVALID_PARAMETER; WinsFreeMemory(pwszIps); pwszIps = NULL; goto ErrorReturn; }
pdwTemp = pdwIp; pdwIp = WinsAllocateMemory((dwIpCount+1)*sizeof(DWORD)); if( pdwIp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; WinsFreeMemory(pwszIps); pwszIps = NULL; if( pdwTemp ) { WinsFreeMemory(pdwTemp); pdwTemp = NULL; } goto ErrorReturn; }
if( pdwTemp ) { memcpy(pdwIp, pdwTemp, dwIpCount*sizeof(DWORD)); WinsFreeMemory(pdwTemp); pdwTemp = NULL; }
pdwIp[dwIpCount] = StringToIpAddress(pwcToken); dwIpCount++; pwcToken = wcstok(NULL, pwcTag);
} if( pwszIps ) { WinsFreeMemory(pwszIps); pwszIps = NULL; }
break;
} case 1: { DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 10); if( dw is 1 ) TypeOfRec = WINSINTF_STATIC; else if( dw is 2 ) TypeOfRec = WINSINTF_DYNAMIC; else if( dw is 0 ) TypeOfRec = WINSINTF_BOTH; else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } break; } case 2: { DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 10); NoOfRecsDesired = dw; break; } case 3: { DWORD dw = 0; if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dw = STRTOUL(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL, 10); if( dw is 0 ) { dwStart = WINSINTF_BEGINNING; } else if ( dw is 1 ) { dwStart = WINSINTF_END; } else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } break; } case 4: { DWORD dwLen = 0, k=0; dwLen = wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); if( dwLen > 2 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } for( k=0; k<dwLen; k++ ) { WCHAR wc = ppwcArguments[dwCurrentIndex+pdwTagNum[j]][k]; if( isalpha(wc) is TRUE ) { if( towlower(wc) < L'a' or towlower(wc) > L'f' ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } } chFilter = StringToHexA(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); wcsncpy(wszFilter, ppwcArguments[dwCurrentIndex+pdwTagNum[j]], 2); fFilter = TRUE; break; } case 5: { DWORD dwLen = wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); if( dwLen is 0 or dwLen > MAX_PATH ) { wcscpy(wcFile, L"wins.rec"); fFile = TRUE; break; } wcscpy(wcFile, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); fFile = TRUE; break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } } } if( fAll ) { ResultsN.WinsStat.NoOfPnrs = 0; ResultsN.WinsStat.pRplPnrs = 0; ResultsN.NoOfWorkerThds = 1; ResultsN.pAddVersMaps = NULL;
Status = WinsStatusNew(g_hBind, WINSINTF_E_CONFIG_ALL_MAPS, &ResultsN); if( Status is RPC_S_PROCNUM_OUT_OF_RANGE ) { //Try old API
Results.WinsStat.NoOfPnrs = 0; Results.WinsStat.pRplPnrs = 0; Status = WinsStatus(g_hBind, WINSINTF_E_CONFIG_ALL_MAPS, &Results); fNew = FALSE; } if( Status isnot NO_ERROR ) goto ErrorReturn;
if( fNew ) { dwIpCount = ResultsN.NoOfOwners; } else { dwIpCount = Results.NoOfOwners; } }
if( fFile is TRUE ) { pFile = _wfopen(wcFile, L"a+"); if( pFile is NULL ) { DisplayMessage(g_hModule, EMSG_WINS_FILEOPEN_FAILED, wcFile); fOpenFile = FALSE; } else { fOpenFile = TRUE; } } for( dw=0; dw<dwIpCount; dw++ ) { LPSTR pszLastName = NULL; DWORD dwLastNameLen = 0; DWORD dwDesired = 0; BOOL fDone = FALSE;
if( fHeader is FALSE ) { DisplayMessage(g_hModule, MSG_WINS_RECORD_DESC);
DisplayMessage(g_hModule, MSG_WINS_RECORD_TABLE);
if( fOpenFile ) { DumpMessage(g_hModule, pFile, FMSG_WINS_RECORDS_TABLE); } DisplayMessage(g_hModule, WINS_FORMAT_LINE);
fHeader = TRUE; }
WinsAdd.Len = 4; WinsAdd.Type = 0;
if( fAll ) { if( fNew ) { WinsAdd.IPAdd = ResultsN.pAddVersMaps[dw].Add.IPAdd; } else { WinsAdd.IPAdd = Results.AddVersMaps[dw].Add.IPAdd; } } else { WinsAdd.IPAdd = pdwIp[dw]; }
DisplayMessage(g_hModule, MSG_SRVR_RETRIEVE_DATABASE, IpAddressToString(WinsAdd.IPAdd)); fOnce = FALSE; Status = NO_ERROR; dwTotal = 0; fDone = FALSE;
while( Status is NO_ERROR and dwTotal < NoOfRecsDesired and fDone is FALSE ) { dwDesired = ( NoOfRecsDesired - dwTotal > 500 ) ? 500 : (NoOfRecsDesired - dwTotal); if( Recs.pRow ) { WinsFreeMem(Recs.pRow); Recs.pRow = NULL; } memset( &Recs, 0x00, sizeof(WINSINTF_RECS_T));
Status = WinsGetDbRecsByName(g_hBind, &WinsAdd, dwStart, fOnce ? pszLastName: NULL, dwLastNameLen, dwDesired, TypeOfRec, &Recs); if( Status isnot NO_ERROR ) { if( fOnce is FALSE ) { DisplayMessage(g_hModule, EMSG_SRVR_RETRIEVEDB_FAILED, IpAddressToString(WinsAdd.IPAdd));
DisplayErrorMessage(EMSG_SRVR_ERROR_MESSAGE, Status); fError = TRUE; } else if ( Status isnot ERROR_REC_NON_EXISTENT ) { DisplayMessage(g_hModule, EMSG_SRVR_RETRIEVEDB_FAILED, IpAddressToString(WinsAdd.IPAdd));
DisplayErrorMessage(EMSG_SRVR_ERROR_MESSAGE, Status); fError = TRUE; }
Status = NO_ERROR;
break; }
fOnce = TRUE;
dwTotal += Recs.NoOfRecs;
if( dwDesired > Recs.NoOfRecs ) { fDone = TRUE; } pRow = Recs.pRow;
if( Recs.NoOfRecs is 0 ) { DisplayMessage(g_hModule, MSG_WINS_NO_RECORDS); } else { WCHAR Name[21] = {L'\0'}; WCHAR Type[2] = {L'\0'}; WCHAR State[2] = {L'\0'}; WCHAR Version[9] = {L'\0'}; WCHAR Group[2] = {L'\0'}; WCHAR IPAdd[MAX_IP_STRING_LEN+1] = {L'\0'}; WCHAR Buffer[16] = {L'\0'}; DWORD dwState = WINS_STATE_ACTIVE; DWORD dwType = WINS_TYPE_STATIC; DWORD dwGroup = WINS_GROUP_UNIQUE; DWORD dwTempLen = 0; struct tm* time = NULL; int iType = 1; for( j=0; j<Recs.NoOfRecs; j++ ) { LPWSTR pwszGroup = NULL, pwszStatic = NULL, pwszType = NULL, pwszState = NULL; WCHAR wszGroup[100] = {L'\0'}, wszType[100] = {L'\0'}, wszState[100] = {L'\0'};
CHAR chEndChar = (CHAR)0x00;
DWORD dwGrouplen = 0, dwTypelen = 0, dwStatelen = 0; memset( Name, 0x00, 21*sizeof(WCHAR)); if( j is Recs.NoOfRecs - 1 ) { if( pszLastName ) { WinsFreeMemory(pszLastName); pszLastName = NULL; } pszLastName = WinsAllocateMemory(strlen(pRow->pName)+2); if(pszLastName is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } memset(pszLastName, 0x00, strlen(pRow->pName)+2);
strcpy(pszLastName, pRow->pName); //1B records detected at the boundary, swap 1st and
//16th char
if( pszLastName[15] == 0x1B ) { CHAR ch = pszLastName[15]; pszLastName[15] = pszLastName[0]; pszLastName[0] = ch; }
strcat(pszLastName, "\x01"); dwLastNameLen = pRow->NameLen+1; } if( pRow->NameLen > 16 ) i = 15; else i = pRow->NameLen; chEndChar = pRow->pName[i];
pRow->pName[i] = (CHAR)0x20; //pRow->pName[16] = '\0';
if( fFilter is TRUE ) { if( chFilter isnot chEndChar ) { pRow++; continue; } } pwszTemp = WinsOemToUnicode(pRow->pName, NULL); if( pwszTemp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } dwTempLen = wcslen(pwszTemp); dwTempLen = (dwTempLen>16) ? 16 : dwTempLen;
wcsncpy(Name, pwszTemp, dwTempLen); WinsFreeMemory(pwszTemp); pwszTemp = NULL;
for( i=dwTempLen; i<16; i++ ) Name[i] = L' '; Name[15] = L'['; i=0;
WinsHexToString(Name+16, (LPBYTE)&chEndChar, 1); Name[18] = L'h'; Name[19] = L']'; Name[20] = L'\0';
if( pRow->fStatic ) { wcscpy(Type, L"S"); pwszType = L"STATIC "; dwType = WINS_TYPE_STATIC; } else { wcscpy(Type, L"D"); pwszType = L"DYNAMIC"; dwType = WINS_TYPE_DYNAMIC;
}
if( pRow->State_e is WINSINTF_E_ACTIVE ) { wcscpy(State, L"A"); pwszState = L"ACTIVE "; dwState = WINS_STATE_ACTIVE; }
else if( pRow->State_e is WINSINTF_E_RELEASED ) { wcscpy(State, L"R"); pwszState = L"RELEASED "; dwState = WINS_STATE_RELEASED; } else { wcscpy(State, L"T"); pwszState = L"TOMBSTONE"; dwState = WINS_STATE_TOMBSTONE; }
if( pRow->TypOfRec_e is WINSINTF_E_UNIQUE ) { wcscpy(Group,L"U"); pwszGroup = L"UNIQUE "; dwGroup = WINS_GROUP_UNIQUE; } else if( pRow->TypOfRec_e is WINSINTF_E_NORM_GROUP ) { wcscpy(Group,L"N"); pwszGroup = L"GROUP "; dwGroup = WINS_GROUP_GROUP; } else if( pRow->TypOfRec_e is WINSINTF_E_SPEC_GROUP ) { if( pRow->pName[15] is 0x1C ) { wcscpy(Group, L"D"); pwszGroup = L"DOMAIN NAME "; dwGroup = WINS_GROUP_DOMAIN; } else { wcscpy(Group,L"I"); pwszGroup = L"INTERNET GROUP"; dwGroup = WINS_GROUP_INTERNET; } } else { wcscpy(Group,L"M"); pwszGroup = L"MULTIHOMED "; dwGroup = WINS_GROUP_MULTIHOMED; } dwStatelen = LoadStringW(g_hModule, dwState, wszState, sizeof(wszState)/sizeof(WCHAR));
dwGrouplen = LoadStringW(g_hModule, dwGroup, wszGroup, sizeof(wszGroup)/sizeof(WCHAR));
dwTypelen = LoadStringW(g_hModule, dwType, wszType, sizeof(wszType)/sizeof(WCHAR));
memset(Version, L'\0', 9); _itow((int)pRow->VersNo.LowPart, Buffer, 16); wcsncpy(Version, Buffer, wcslen(Buffer)>8?8:wcslen(Buffer));
for( i=wcslen(Version); i<9; i++ ) Version[i] = L' '; Version[8] = L'\0';
pwszTime = GetDateTimeString(pRow->TimeStamp, TRUE, &iType);
if ( pRow->TypOfRec_e is WINSINTF_E_UNIQUE or pRow->TypOfRec_e is WINSINTF_E_NORM_GROUP ) {
wcscpy(IPAdd, IpAddressToString(pRow->Add.IPAdd));
for( i=wcslen(IPAdd); i<MAX_IP_STRING_LEN; i++ ) IPAdd[i] = L' '; IPAdd[MAX_IP_STRING_LEN] = L'\0'; DisplayMessage(g_hModule, MSG_WINS_RECORD_ENTRY, Name, Type, State, Version, Group, IPAdd, iType ? wszInfinite : pwszTime); if( fOpenFile ) { DumpMessage(g_hModule, pFile, FMSG_WINS_RECORD_ENTRY, Name, ( dwTypelen > 0 ) ? wszType : pwszType, ( dwStatelen > 0 ) ? wszState : pwszState, Version, ( dwGrouplen > 0 ) ? wszGroup : pwszGroup, iType ? wszInfinite : pwszTime, IPAdd, IpAddressToString(WinsAdd.IPAdd));
}
} else //spec. grp or multihomed
{ DWORD ind; BOOL fFirst = FALSE; for ( ind=0; ind < pRow->NoOfAdds ; /*no third expr*/ ) { struct in_addr InAddr1, InAddr2; LPWSTR pwszTempAddr = NULL; InAddr1.s_addr = htonl( (pRow->pAdd + ind++)->IPAdd); InAddr2.s_addr = htonl( (pRow->pAdd + ind++)->IPAdd);
pwszTempAddr = WinsOemToUnicode(inet_ntoa(InAddr2), NULL);
if( pwszTempAddr is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
wcscpy(IPAdd, pwszTempAddr);
WinsFreeMemory(pwszTempAddr); pwszTempAddr = NULL;
for( i=wcslen(IPAdd); i<MAX_IP_STRING_LEN; i++ ) IPAdd[i] = L' ';
if( fFirst is FALSE ) {
fFirst = TRUE; DisplayMessage(g_hModule, MSG_WINS_RECORD_ENTRY, Name, Type, State, Version, Group, IPAdd, iType ? wszInfinite : pwszTime); if( fOpenFile ) { pwszTempAddr = WinsOemToUnicode(inet_ntoa(InAddr1), NULL); if( pwszTempAddr is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } DumpMessage(g_hModule, pFile, FMSG_WINS_RECORD_ENTRY, Name, ( dwTypelen > 0 ) ? wszType : pwszType, ( dwStatelen > 0 ) ? wszState : pwszState, Version, ( dwGrouplen > 0 ) ? wszGroup : pwszGroup, iType ? wszInfinite : pwszTime, IPAdd, pwszTempAddr); WinsFreeMemory(pwszTempAddr); pwszTempAddr = NULL;
} } else {
pwszTempAddr = WinsOemToUnicode(inet_ntoa(InAddr2), NULL); if( pwszTempAddr is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
DisplayMessage(g_hModule, MSG_WINS_RECORD_IPADDRESS, pwszTempAddr);
WinsFreeMemory(pwszTempAddr); pwszTempAddr = NULL;
if( fOpenFile ) { pwszTempAddr = WinsOemToUnicode(inet_ntoa(InAddr1), NULL); if( pwszTempAddr is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } DumpMessage(g_hModule, pFile, FMSG_WINS_RECORD_IPADDRESS, IPAdd, pwszTempAddr);
WinsFreeMemory(pwszTempAddr); pwszTempAddr = NULL; DumpMessage(g_hModule, pFile, WINS_FORMAT_LINE); } DisplayMessage(g_hModule, WINS_FORMAT_LINE);
} } }
pRow++; dwRecCount++; if( pwszTime ) { WinsFreeMemory(pwszTime); pwszTime = NULL; } }
} }
DisplayMessage(g_hModule, MSG_WINS_RECORDS_RETRIEVED, IpAddressToString(WinsAdd.IPAdd), dwTotal); }
DisplayMessage(g_hModule, WINS_FORMAT_LINE); if( fFilter ) { DisplayMessage(g_hModule, MSG_SRVR_FILTER_RECCOUNT, wszFilter, dwRecCount); if( fOpenFile ) { DumpMessage(g_hModule, pFile, MSG_SRVR_FILTER_RECCOUNT, wszFilter, dwRecCount); } } else { DisplayMessage(g_hModule, MSG_SRVR_TOTAL_RECCOUNT, dwRecCount);
if( fOpenFile ) { DumpMessage(g_hModule, pFile, MSG_SRVR_TOTAL_RECCOUNT, dwRecCount); }
}
CommonReturn: if( Status is NO_ERROR and fError is FALSE ) { DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); } else if( fError is TRUE ) { DisplayMessage(g_hModule, EMSG_WINS_RETRIEVEDB_PARTIAL); } if( fOpenFile ) { fclose(pFile); pFile = NULL; } if( Recs.pRow ) { WinsFreeMem(Recs.pRow); Recs.pRow = NULL; }
if( pwszTime ) { WinsFreeMemory(pwszTime); pwszTime = NULL; }
if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; } if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; } return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_SHOW_DATABASE, Status); goto CommonReturn;}
DWORDHandleSrvrShowDomain( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Displays the domain master browser recordsArguments : All aguments are passes as array of wide char strings in ppwcArguments. NONE.Return Value: Returns the status of the operation.
--*/
{ DWORD Status = NO_ERROR; DWORD i = 0, j=0; WINSINTF_BROWSER_NAMES_T Names; PWINSINTF_BROWSER_INFO_T pInfo = NULL; PWINSINTF_BROWSER_INFO_T pInfoSv = NULL; WCHAR wcName[273] = {L'\0'}, wcCount[20] = {L'\0'};
for(i=0; i<273; i++ ) wcName[i] = L' ';
for(i=0; i<20; i++ ) wcCount[i] = L' ';
wcCount[19] = L'\0';
Names.EntriesRead = 0; Names.pInfo = NULL; Status = WinsGetBrowserNames(&g_BindData, &Names); if (Status is NO_ERROR ) { DisplayMessage(g_hModule, MSG_WINS_DOMAIN_COUNT, Names.EntriesRead); DisplayMessage(g_hModule, WINS_FORMAT_LINE);
DisplayMessage(g_hModule, MSG_WINS_DOMAIN_TABLE);
pInfoSv = pInfo = Names.pInfo; for(i=0; i < Names.EntriesRead; i++) { LPWSTR pwcTemp = NULL; LPSTR pcTemp = NULL;
_itow((int)i, wcCount+3, 10); for( j=wcslen(wcCount); j<19; j++ ) wcCount[j] = L' '; wcCount[19] = L'\0';
pwcTemp = WinsOemToUnicode(pInfo->pName, NULL); if( pwcTemp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
wcsncpy(wcName+3, pwcTemp, (15>wcslen(pwcTemp))? wcslen(pwcTemp) : 15); wcName[18] = L'[';
if( strlen(pInfo->pName ) > 15 ) pcTemp = pInfo->pName + 15; else pcTemp = pInfo->pName + strlen(pInfo->pName);
WinsHexToString(wcName+19, (LPBYTE)pcTemp, 1); wcName[21] = L'h'; wcName[22] = L']'; if( strlen(pInfo->pName)>16) { wcName[23] = L'.'; wcscpy(wcName+24, pwcTemp+17); wcName[wcslen(wcName)] = L'\0'; } else wcName[23] = L'\0';
DisplayMessage(g_hModule, MSG_WINS_DOMAIN_ENTRY, wcCount, wcName); if( pwcTemp ) { WinsFreeMemory(pwcTemp); pwcTemp = NULL; } pInfo++; } WinsFreeMem(pInfoSv); pInfoSv = NULL; if( Status is NO_ERROR ) DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); } else goto ErrorReturn;
CommonReturn: if( pInfoSv ) { WinsFreeMem(pInfoSv); pInfoSv = NULL; } return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_SHOW_DOMAIN, Status); goto CommonReturn;
}
DWORDHandleSrvrShowInfo( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Displays server propertiesArguments : All aguments are passes as array of wide char strings in ppwcArguments. NONE.Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; HKEY hServer = NULL, hParameter = NULL, hCCRoot = NULL; LPWSTR pTemp = NULL;
WCHAR wcData[256] = {L'\0'}; DWORD dwType = REG_SZ, dwLen = 256*sizeof(WCHAR), dwData = 0, dwLow = 0; LPWSTR pwszDayString = NULL, pwszTimeString = NULL;
if( wcslen(g_ServerNetBiosName) > 0 ) { pTemp = g_ServerNetBiosName; }
Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer);
if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegOpenKeyEx(hServer, PARAMETER, 0, KEY_READ, &hParameter);
if( Status isnot NO_ERROR ) goto ErrorReturn;
DisplayMessage(g_hModule, WINS_FORMAT_LINE); DisplayMessage(g_hModule, MSG_WINS_DATABASE_BACKUPPARAM);
Status = RegQueryValueEx(hParameter, WINSCNF_BACKUP_DIR_PATH_NM, 0, &dwType, (LPBYTE)wcData, &dwLen); if( Status is NO_ERROR ) { DisplayMessage(g_hModule, MSG_WINS_DATABASE_BACKUPDIR, wcData); } else { DisplayMessage(g_hModule, MSG_WINS_DATABASE_BACKUPDIR, wszUnknown); }
dwLen = sizeof(DWORD); dwType = REG_DWORD;
Status = RegQueryValueEx(hParameter, WINSCNF_DO_BACKUP_ON_TERM_NM, 0, &dwType, (LPBYTE)&dwData, &dwLen);
if( Status is NO_ERROR ) { DisplayMessage(g_hModule, MSG_WINS_DATABASE_BACKUPONTERM, dwData ? wszEnable : wszDisable); } else { DisplayMessage(g_hModule, MSG_WINS_DATABASE_BACKUPONTERM, wszUnknown); }
DisplayMessage(g_hModule, WINS_FORMAT_LINE); DisplayMessage(g_hModule, MSG_WINS_NAMERECORD_SETTINGS);
{
WINSINTF_RESULTS_T Results = {0}; WINSINTF_RESULTS_NEW_T ResultsN = {0}; BOOL fNew = TRUE;
ResultsN.WinsStat.NoOfPnrs = 0; ResultsN.WinsStat.pRplPnrs = NULL; ResultsN.NoOfWorkerThds = 1;
Status = WinsStatusNew(g_hBind, WINSINTF_E_CONFIG, &ResultsN);
if( Status is RPC_S_PROCNUM_OUT_OF_RANGE ) { //Try old API
Results.WinsStat.NoOfPnrs = 0; Results.WinsStat.pRplPnrs = 0; Status = WinsStatus(g_hBind, WINSINTF_E_CONFIG, &Results); fNew = FALSE; } if( Status is NO_ERROR ) { if( fNew ) { LPWSTR pwszDayString = NULL;
pwszDayString = MakeDayTimeString(ResultsN.RefreshInterval);
DisplayMessage(g_hModule, MSG_WINS_NAMERECORD_REFRESHINTVL, pwszDayString); WinsFreeMemory(pwszDayString); pwszDayString = NULL;
pwszDayString = MakeDayTimeString(ResultsN.TombstoneInterval); DisplayMessage(g_hModule, MSG_WINS_NAMERECORD_TOMBSTONEINTVL, pwszDayString); WinsFreeMemory(pwszDayString); pwszDayString = NULL;
pwszDayString = MakeDayTimeString(ResultsN.TombstoneTimeout); DisplayMessage(g_hModule, MSG_WINS_NAMERECORD_TOMBSTONETMOUT, pwszDayString); WinsFreeMemory(pwszDayString); pwszDayString = NULL;
pwszDayString = MakeDayTimeString(ResultsN.VerifyInterval); DisplayMessage(g_hModule, MSG_WINS_NAMERECORD_VERIFYINTVL, pwszDayString); WinsFreeMemory(pwszDayString); pwszDayString = NULL;
if( ResultsN.WinsStat.pRplPnrs) { WinsFreeMem(ResultsN.WinsStat.pRplPnrs); ResultsN.WinsStat.pRplPnrs = NULL; } } else { LPWSTR pwszDayString = NULL;
pwszDayString = MakeDayTimeString(Results.RefreshInterval);
DisplayMessage(g_hModule, MSG_WINS_NAMERECORD_REFRESHINTVL, pwszDayString); WinsFreeMemory(pwszDayString); pwszDayString = NULL;
pwszDayString = MakeDayTimeString(Results.TombstoneInterval); DisplayMessage(g_hModule, MSG_WINS_NAMERECORD_TOMBSTONEINTVL, pwszDayString); WinsFreeMemory(pwszDayString); pwszDayString = NULL;
pwszDayString = MakeDayTimeString(Results.TombstoneTimeout); DisplayMessage(g_hModule, MSG_WINS_NAMERECORD_TOMBSTONETMOUT, pwszDayString); WinsFreeMemory(pwszDayString); pwszDayString = NULL;
pwszDayString = MakeDayTimeString(Results.VerifyInterval); DisplayMessage(g_hModule, MSG_WINS_NAMERECORD_VERIFYINTVL, pwszDayString); WinsFreeMemory(pwszDayString); pwszDayString = NULL;
if( Results.WinsStat.pRplPnrs) { WinsFreeMem(Results.WinsStat.pRplPnrs); Results.WinsStat.pRplPnrs = NULL; } }
} else { DisplayMessage(g_hModule, MSG_WINS_NAMERECORD_REFRESHINTVL, wszUnknown); DisplayMessage(g_hModule, MSG_WINS_NAMERECORD_TOMBSTONEINTVL, wszUnknown);
DisplayMessage(g_hModule, MSG_WINS_NAMERECORD_TOMBSTONETMOUT, wszUnknown);
DisplayMessage(g_hModule, MSG_WINS_NAMERECORD_VERIFYINTVL, wszUnknown);
}
}
DisplayMessage(g_hModule, WINS_FORMAT_LINE); DisplayMessage(g_hModule, MSG_WINS_DBCC_PARAM);
Status = RegOpenKeyEx(hParameter, CC, 0, KEY_READ, &hCCRoot);
if( Status isnot NO_ERROR ) { DisplayMessage(g_hModule, MSG_WINS_DBCC_STATE, wszDisable); } else { DisplayMessage(g_hModule, MSG_WINS_DBCC_STATE, wszEnable);
dwType = REG_DWORD; dwData = 0; dwLen = sizeof(DWORD); Status = RegQueryValueEx(hCCRoot, WINSCNF_CC_MAX_RECS_AAT_NM, 0, &dwType, (LPBYTE)&dwData, &dwLen);
if( Status is NO_ERROR ) { WCHAR Buffer[20] = {L'\0'}; _itow(dwData, Buffer, 10); DisplayMessage(g_hModule, MSG_WINS_DBCC_MAXCOUNT, Buffer); } else { DisplayMessage(g_hModule, MSG_WINS_DBCC_MAXCOUNT, wszUnknown); }
dwType = REG_DWORD; dwData = 0; dwLen = sizeof(DWORD); Status = RegQueryValueEx(hCCRoot, WINSCNF_CC_USE_RPL_PNRS_NM, 0, &dwType, (LPBYTE)&dwData, &dwLen); if( Status is NO_ERROR ) { DisplayMessage(g_hModule, MSG_WINS_DBCC_CHECKAGAINST, dwData ? wszRandom : wszOwner); } else { DisplayMessage(g_hModule, MSG_WINS_DBCC_CHECKAGAINST, wszUnknown); }
dwType = REG_DWORD; dwData = 0; dwLen = sizeof(DWORD);
Status = RegQueryValueEx(hCCRoot, WINSCNF_CC_INTVL_NM, 0, &dwType, (LPBYTE)&dwData, &dwLen); if( Status is NO_ERROR ) { WCHAR Buffer[5] = {L'\0'}; _itow(dwData/(60*60), Buffer, 10); DisplayMessage(g_hModule, MSG_WINS_DBCC_CHECKEVERY, Buffer); } else { DisplayMessage(g_hModule, MSG_WINS_DBCC_CHECKEVERY, wszUnknown); }
memset(wcData, 0x00, 256*sizeof(WCHAR));
dwType = REG_SZ; dwLen = 256*sizeof(WCHAR);
Status = RegQueryValueEx(hCCRoot, WINSCNF_SP_TIME_NM, 0, &dwType, (LPBYTE)wcData, &dwLen); if( Status is NO_ERROR ) { DisplayMessage(g_hModule, MSG_WINS_DBCC_STARTAT, wcData); } else { DisplayMessage(g_hModule, MSG_WINS_DBCC_STARTAT, wszUnknown); } }
DisplayMessage(g_hModule, WINS_FORMAT_LINE); DisplayMessage(g_hModule, MSG_WINS_LOGGING_PARAM);
dwType = REG_DWORD; dwData = 0; dwLen = sizeof(DWORD);
Status = RegQueryValueEx(hParameter, WINSCNF_LOG_FLAG_NM, 0, &dwType, (LPBYTE)&dwData, &dwLen);
if( Status is NO_ERROR ) { DisplayMessage(g_hModule, MSG_WINS_LOGGING_FLAG, dwData ? wszEnable : wszDisable); } else { DisplayMessage(g_hModule, MSG_WINS_LOGGING_FLAG, wszUnknown); }
dwType = REG_DWORD; dwData = 0; dwLen = sizeof(DWORD);
Status = RegQueryValueEx(hParameter, WINSCNF_LOG_DETAILED_EVTS_NM, 0, &dwType, (LPBYTE)&dwData, &dwLen);
if( Status is NO_ERROR ) { DisplayMessage(g_hModule, MSG_WINS_LOGGING_DETAILS, dwData ? wszEnable : wszDisable); } else { DisplayMessage(g_hModule, MSG_WINS_LOGGING_DETAILS, wszUnknown); }
DisplayMessage(g_hModule, WINS_FORMAT_LINE); DisplayMessage(g_hModule, MSG_WINS_BURSTHNDL_PARAM);
dwType = REG_DWORD; dwData = 0; dwLen = sizeof(DWORD);
Status = RegQueryValueEx(hParameter, WINSCNF_BURST_HANDLING_NM, 0, &dwType, (LPBYTE)&dwData, &dwLen);
if( Status is NO_ERROR ) { DisplayMessage(g_hModule, MSG_WINS_BURSTHNDL_STATE, dwData ? wszEnable : wszDisable);
if( dwData > 0 ) { dwType = REG_DWORD; dwData = 0; dwLen = sizeof(DWORD);
Status = RegQueryValueEx(hParameter, WINSCNF_BURST_QUE_SIZE_NM, 0, &dwType, (LPBYTE)&dwData, &dwLen);
if( Status is NO_ERROR ) { WCHAR Buffer[10] = {L'\0'}; _itow(dwData, Buffer, 10); DisplayMessage(g_hModule, MSG_WINS_BURSTHNDL_SIZE, Buffer); } else { DisplayMessage(g_hModule, MSG_WINS_BURSTHNDL_SIZE, wszUnknown); } } } else { DisplayMessage(g_hModule, MSG_WINS_BURSTHNDL_STATE, wszUnknown); }
DisplayMessage(g_hModule, WINS_FORMAT_LINE);
dwType = REG_DWORD; dwData = 0; dwLen = sizeof(DWORD);
Status = RegQueryValueEx(hParameter, WINSCNF_INIT_VERSNO_VAL_HW_NM, NULL, &dwType, (LPBYTE)&dwData, &dwLen); dwLow = 0;
Status = RegQueryValueEx(hParameter, WINSCNF_INIT_VERSNO_VAL_LW_NM, NULL, &dwType, (LPBYTE)&dwLow, &dwLen);
if(Status is NO_ERROR ) { wsprintf(wcData, L" %x , %x", dwData, dwLow); DisplayMessage(g_hModule, MSG_SRVR_START_VERSION, wcData); }
DisplayMessage(g_hModule, WINS_FORMAT_LINE);CommonReturn: if( Status is NO_ERROR ) DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS);
if( hCCRoot ) { RegCloseKey(hCCRoot); hCCRoot = NULL; }
if( hParameter ) { RegCloseKey(hParameter); hParameter = NULL; }
if( hServer ) { RegCloseKey(hServer); hServer = NULL; }
return Status;
ErrorReturn: DisplayErrorMessage(EMSG_SRVR_SHOW_INFO, Status); goto CommonReturn;
}
DWORDHandleSrvrShowPartner( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Displays the list of Partners optionally based on the partner type.Arguments : All aguments are passes as array of wide char strings in ppwcArguments. Optional : Partner type - Pull or Push or Both(default)Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount, dwCount; TAG_TYPE pttTags[] = { {WINS_TOKEN_TYPE, FALSE, FALSE} };
PDWORD pdwTagType = NULL, pdwTagNum = NULL; DWORD p = 0; typedef enum {all=0, pull, push, both}eType;
eType Type = all;
BOOL fPush = TRUE, fPull = TRUE; HKEY hServer = NULL, hPullRoot = NULL, hPushRoot = NULL; WCHAR wcServerIpAdd[MAX_IP_STRING_LEN] = {L'\0'}; WCHAR wcServer[256] = {L'\0'}; DWORD dwLen = 0; typedef struct _Server_List { WCHAR wcServerIpAddress[MAX_IP_STRING_LEN + 1]; WCHAR wcServerName[1024]; eType etype; }Server_List, *PServer_List; PServer_List pServerList = NULL; LPWSTR pwszServerList = NULL;
LPWSTR pTemp = NULL; dwCount = 0;
if( dwArgCount >= dwCurrentIndex + 1 ) { dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum); if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
for( j=0; j<dwTagCount; j++ ) { switch(pdwTagType[j]) { case 0: { WCHAR wc = L'\0';
if( IsPureNumeric(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) is FALSE or wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) > 1 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } wc = ppwcArguments[dwCurrentIndex+pdwTagNum[j]][0];
if( wc is L'0' ) { Type = all; } if( wc is L'1' ) { Type = pull; break; } else if( wc is L'2' ) { Type = push; break; } else if ( wc is L'3' ) { Type = both; break; } else { Status = ERROR_INVALID_PARAMETER; break; } break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } } }
if( wcslen(g_ServerNetBiosName) > 0 ) { pTemp = g_ServerNetBiosName; }
Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer); if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegOpenKeyEx(hServer, PULLROOT, 0, KEY_READ, &hPullRoot); if( Status isnot ERROR_FILE_NOT_FOUND and Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegOpenKeyEx(hServer, PUSHROOT, 0, KEY_READ, &hPushRoot); if( Status isnot ERROR_FILE_NOT_FOUND and Status isnot NO_ERROR ) goto ErrorReturn; if( hPullRoot is NULL and hPushRoot is NULL ) { goto ErrorReturn; }
Status = NO_ERROR;
{ DWORD dwSubkey = 0; HKEY hKey = NULL;
if (hPullRoot != NULL) { Status = RegQueryInfoKey(hPullRoot, NULL, NULL, NULL, &dwSubkey, NULL, NULL, NULL, NULL, NULL, NULL, NULL);
if (Status isnot NO_ERROR) { goto ErrorReturn; } }
if( dwSubkey > 0 ) { pServerList = WinsAllocateMemory(dwSubkey*sizeof(Server_List)); if( pServerList is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
for( j=0; j<dwSubkey; j++ ) { DWORD dwLen = (MAX_IP_STRING_LEN + 1)*sizeof(WCHAR); DWORD dwType = REG_SZ; Status = RegEnumKeyEx(hPullRoot, j, pServerList[j].wcServerIpAddress, &dwLen, NULL, NULL, NULL, NULL); if( Status isnot NO_ERROR ) { if( Status is ERROR_FILE_NOT_FOUND ) { continue; } else { goto ErrorReturn; } } Status = RegOpenKeyEx(hPullRoot, pServerList[j].wcServerIpAddress, 0, KEY_READ, &hKey); if( Status isnot NO_ERROR ) { if( Status is ERROR_FILE_NOT_FOUND ) { continue; } else { goto ErrorReturn; } }
dwLen = 1024*sizeof(WCHAR);
Status = RegQueryValueEx(hKey, L"NetBIOSName", 0, &dwType, (LPBYTE)pServerList[j].wcServerName, &dwLen); if( Status isnot NO_ERROR ) { if( Status is ERROR_FILE_NOT_FOUND ) { wcscpy(pServerList[j].wcServerName, wszUnknown); } else { goto ErrorReturn; } }
pServerList[j].etype = pull; RegCloseKey(hKey); hKey = NULL; } }
i = dwSubkey; dwCount = i; Status = NO_ERROR; dwSubkey = 0;
if (hPushRoot != NULL) { Status = RegQueryInfoKey(hPushRoot, NULL, NULL, NULL, &dwSubkey, NULL, NULL, NULL, NULL, NULL, NULL, NULL);
if (Status isnot NO_ERROR) { goto ErrorReturn; } }
if( dwSubkey > 0 ) { PServer_List pTempList = NULL;
if( pServerList ) pTempList = pServerList; pServerList= WinsAllocateMemory((dwSubkey+i)*sizeof(Server_List)); if( pServerList is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } memcpy(pServerList, pTempList, i*sizeof(Server_List)); if( pTempList ) { WinsFreeMemory(pTempList); pTempList = NULL; }
p = 0; for( j=0; j<dwSubkey; j++ ) { DWORD dwLen = (MAX_IP_STRING_LEN + 1)*sizeof(WCHAR); DWORD dwType = REG_SZ; WCHAR wcIpTemp[MAX_IP_STRING_LEN+1] = {L'\0'}; DWORD k = 0; BOOL fFind = FALSE; Status = RegEnumKeyEx(hPushRoot, j, wcIpTemp, &dwLen, NULL, NULL, NULL, NULL); if( Status isnot NO_ERROR ) { if( Status is ERROR_FILE_NOT_FOUND ) { Status = NO_ERROR; continue; } else { goto ErrorReturn; } } for( k=0; k<i; k++ ) { if( _wcsicmp(pServerList[k].wcServerIpAddress, wcIpTemp) is 0 ) { pServerList[k].etype = all; fFind = TRUE; break; } }
if( fFind is FALSE ) { wcscpy(pServerList[i+p].wcServerIpAddress, wcIpTemp); Status = RegOpenKeyEx(hPushRoot, pServerList[i+p].wcServerIpAddress, 0, KEY_READ, &hKey); if( Status isnot NO_ERROR ) { if( Status is ERROR_FILE_NOT_FOUND ) { Status = NO_ERROR; continue; } else { goto ErrorReturn; } }
dwLen = 1024*sizeof(WCHAR); Status = RegQueryValueEx(hKey, L"NetBIOSName", 0, &dwType, (LPBYTE)pServerList[i+p].wcServerName, &dwLen); if( Status isnot NO_ERROR ) { if( Status is ERROR_FILE_NOT_FOUND ) { wcscpy(pServerList[i+p].wcServerName, wszUnknown); Status = NO_ERROR; } else { goto ErrorReturn; } } pServerList[i+p].etype = push; RegCloseKey(hKey); hKey = NULL; p++; dwCount++; } else continue; } } }
if( dwCount <= 0 ) { DisplayMessage(g_hModule, MSG_WINS_NO_PARTNER); } else { DisplayMessage(g_hModule, WINS_FORMAT_LINE); DisplayMessage(g_hModule, MSG_WINS_PARTNER_COUNT, dwCount);
DisplayMessage(g_hModule, WINS_FORMAT_LINE); DisplayMessage(g_hModule, MSG_WINS_PARTNERLIST_TABLE);
for( j=0; j<dwCount; j++) { WCHAR wcServer[32] = {L'\0'}; WCHAR wcIp[25] = {L'\0'}; DWORD dwServerLen = 0; DWORD k = 0; for( k=0; k<31; k++ ) wcServer[k] = L' '; wcServer[31] = L'\0';
for( k=0; k<24; k++ ) wcIp[k] = L' '; wcIp[24] = L'\0'; dwServerLen = MIN( 24, wcslen(pServerList[j].wcServerName) ); switch(Type) { case all: default: {
memcpy(wcServer+3, pServerList[j].wcServerName, dwServerLen*sizeof(WCHAR) ); memcpy(wcIp+3, pServerList[j].wcServerIpAddress, wcslen(pServerList[j].wcServerIpAddress)*sizeof(WCHAR)); DisplayMessage(g_hModule, MSG_WINS_PARTNERLIST_ENTRY, wcServer, wcIp, pServerList[j].etype is all ? wszPushpull: (pServerList[j].etype is pull) ? wszPull : wszPush); break; } case pull: { if( pServerList[j].etype is pull or pServerList[j].etype is all ) { memcpy(wcServer+3, pServerList[j].wcServerName, dwServerLen*sizeof(WCHAR)); memcpy(wcIp+3, pServerList[j].wcServerIpAddress, wcslen(pServerList[j].wcServerIpAddress)*sizeof(WCHAR));
DisplayMessage(g_hModule, MSG_WINS_PARTNERLIST_ENTRY, wcServer, wcIp, pServerList[j].etype is all ? wszPushpull : wszPull); } break; } case push: { if( pServerList[j].etype is push or pServerList[j].etype is all ) { memcpy(wcServer+3, pServerList[j].wcServerName, dwServerLen*sizeof(WCHAR)); memcpy(wcIp+3, pServerList[j].wcServerIpAddress, wcslen(pServerList[j].wcServerIpAddress)*sizeof(WCHAR));
DisplayMessage(g_hModule, MSG_WINS_PARTNERLIST_ENTRY, wcServer, wcIp, pServerList[j].etype is all ? wszPushpull : wszPush); } break; } case both: { if( pServerList[j].etype is all ) { memcpy(wcServer+3, pServerList[j].wcServerName, dwServerLen*sizeof(WCHAR)); memcpy(wcIp+3, pServerList[j].wcServerIpAddress, wcslen(pServerList[j].wcServerIpAddress)*sizeof(WCHAR));
DisplayMessage(g_hModule, MSG_WINS_PARTNERLIST_ENTRY, wcServer, wcIp, wszPushpull);
} }
} } } DisplayMessage(g_hModule, WINS_FORMAT_LINE);CommonReturn: if( Status is NO_ERROR ) DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS);
if( hPullRoot ) { RegCloseKey(hPullRoot); hPullRoot = NULL; }
if( hPushRoot ) { RegCloseKey(hPushRoot); hPushRoot = NULL; }
if( hServer ) { RegCloseKey(hServer); hServer = NULL; }
if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; } if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; } return Status;ErrorReturn:
if( Status is ERROR_FILE_NOT_FOUND ) { DisplayMessage(g_hModule, MSG_WINS_NO_PARTNER); Status = NO_ERROR; } else { DisplayErrorMessage(EMSG_SRVR_SHOW_PARTNER, Status); } goto CommonReturn;
}
DWORDHandleSrvrShowReccount( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Displays the record count based on the version(optionally)Arguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : Address of the server whose records to be counted Optional : Version range. Max and Min version both in the format {high,low}Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount = 0; TAG_TYPE pttTags[] = { {WINS_TOKEN_SERVER, TRUE, FALSE}, {WINS_TOKEN_MAXVER, FALSE, FALSE}, {WINS_TOKEN_MINVER, FALSE, FALSE}, }; WINSINTF_VERS_NO_T MinVer={0}, MaxVer={0}; WINSINTF_ADD_T WinsAdd = {0}; WINSINTF_RECS_T Recs = {0}; WCHAR wcServerIpAdd[MAX_IP_STRING_LEN+1] = {L'\0'}; LPWSTR pwszTemp = NULL; LPDWORD pdwTagType = NULL, pdwTagNum = NULL;
if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, HLP_SRVR_SHOW_RECCOUNT_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum);
if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
for( j=0; j<dwTagCount; j++ ) { switch( pdwTagType[j] ) { case 0: { struct hostent * lpHostEnt = NULL; CHAR cAddr[16]; BYTE pbAdd[4]; char szAdd[4]; int k = 0, l=0; DWORD dwLen, nLen = 0; CHAR *pTemp = NULL; CHAR *pNetBios = NULL;
if( IsIpAddress( ppwcArguments[dwCurrentIndex+pdwTagNum[j]] ) ) { DWORD dwIp = inet_addr(WinsUnicodeToAnsi(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL)); lpHostEnt = gethostbyaddr((char *)&dwIp, 4, AF_INET); if(lpHostEnt isnot NULL )//Valid IP Address
{ wcscpy(wcServerIpAdd, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); break; } else { Status = WSAGetLastError(); goto ErrorReturn; } }
if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) > 2 and _wcsnicmp(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], L"\\\\", 2) is 0 ) k = 2;
lpHostEnt = gethostbyname(WinsUnicodeToAnsi(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]+k, NULL)); if( lpHostEnt is NULL ) { DisplayMessage(g_hModule, EMSG_WINS_INVALID_COMPUTER_NAME, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]+k); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
memcpy(pbAdd, lpHostEnt->h_addr_list[0], 4); for( l=0;l<4; l++) { _itoa((int)pbAdd[l], szAdd, 10); memcpy(cAddr+nLen, szAdd, strlen(szAdd)); nLen += strlen(szAdd); *(cAddr+nLen) = '.'; nLen++;
} *(cAddr+nLen-1) = '\0'; { LPWSTR pstr = WinsAnsiToUnicode(cAddr, NULL); if (pstr != NULL) { wcscpy(wcServerIpAdd, pstr); WinsFreeMemory(pstr); } else { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } } break; } case 1: { Status = GetVersionData(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], &MaxVer); if( Status isnot NO_ERROR ) goto ErrorReturn; break; } case 2: { Status = GetVersionData(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], &MinVer); if( Status isnot NO_ERROR ) goto ErrorReturn; break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } } WinsAdd.IPAdd = StringToIpAddress(wcServerIpAdd); WinsAdd.Len = 4; WinsAdd.Type = 0;
Status = WinsGetDbRecs(g_hBind, &WinsAdd, MinVer, MaxVer, &Recs);
if( Status isnot NO_ERROR ) goto ErrorReturn; DisplayMessage(g_hModule, MSG_WINS_RECORDS_COUNT_OWNER, wcServerIpAdd, Recs.TotalNoOfRecs);
DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS);
CommonReturn: if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; }
if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; }
if( Recs.pRow ) { WinsFreeMem(Recs.pRow); Recs.pRow = NULL; } return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_SHOW_RECCOUNT, Status); goto CommonReturn;}
DWORDHandleSrvrShowRecbyversion( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Displays records based on Version range, filtered by 16th charArguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : Address of the server whose records to be displayed Optional : Version range. Max and Min version both in the format {high,low}, 16th character, Name etcReturn Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwTagCount, dwNumArgs; TAG_TYPE pttTags[] = { {WINS_TOKEN_SERVER, TRUE, FALSE}, {WINS_TOKEN_MAXVER, FALSE, FALSE}, {WINS_TOKEN_MINVER, FALSE, FALSE}, {WINS_TOKEN_NAME, FALSE, FALSE}, {WINS_TOKEN_ENDCHAR, FALSE, FALSE}, {WINS_TOKEN_CASE, FALSE, FALSE}, }; WCHAR wcName[17] = {L'\0'}; LPWSTR pwcScope = NULL; CHAR ch16thChar = 0x00; DWORD dwNameLen = 0; WINSINTF_VERS_NO_T MinVer={0}, MaxVer={0}; WINSINTF_ADD_T WinsAdd = {0}; WINSINTF_RECS_T Recs = {0}; WCHAR wcServerIpAdd[MAX_IP_STRING_LEN+1] = {L'\0'}; WCHAR wcFile[MAX_PATH] = {L'\0'}; LPWSTR pwszTemp = NULL; BOOL fEndChar = FALSE, fCase = FALSE, fFile = FALSE, fName = FALSE; LPDWORD pdwTagType = NULL, pdwTagNum = NULL; LPSTR pszTempAddr = NULL; LPSTR lpName = NULL;
if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, HLP_SRVR_SHOW_RECBYVERSION_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum);
if( Status isnot NO_ERROR ) goto ErrorReturn;
for( j=0; j<dwTagCount; j++ ) { switch( pdwTagType[j] ) { case 0: { if( IsIpAddress( ppwcArguments[dwCurrentIndex+pdwTagNum[j]] ) ) { wcscpy(wcServerIpAdd, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); } else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } break; } case 1: { Status = GetVersionData(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], &MaxVer); if( Status isnot NO_ERROR ) goto ErrorReturn; break; } case 2: { Status = GetVersionData(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], &MinVer); if( Status isnot NO_ERROR ) goto ErrorReturn; break; } case 3: { wcsncpy(wcName, ppwcArguments[dwCurrentIndex+pdwTagNum[j]], 15); fName = TRUE; break; } case 4: { DWORD dwLen = 0, k=0; fEndChar = TRUE; dwLen = wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); if( dwLen > 2 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } for( k=0; k<dwLen; k++ ) { WCHAR wc = ppwcArguments[dwCurrentIndex+pdwTagNum[j]][k]; if( towlower(wc) >= L'a' and towlower(wc) <= L'z' ) { if( towlower(wc) > L'f' ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } } ch16thChar = StringToHexA(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); break; } case 5: { WCHAR wc = L'\0'; if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) isnot 1 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } wc = ppwcArguments[dwCurrentIndex+pdwTagNum[j]][0]; if( wc is L'0' ) { fCase = FALSE; } else if( wc is L'1' ) { fCase = TRUE; } else { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } break;
} default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } } for( j=0; j<sizeof(pttTags)/sizeof(TAG_TYPE); j++ ) { if( pttTags[j].dwRequired is TRUE and pttTags[j].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } if( fName ) { //Process the name option if present
lpName = WinsUnicodeToOem(wcName, NULL); if( lpName is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwNameLen = strlen(lpName);
if( dwNameLen >= 16 ) { DisplayMessage(g_hModule, EMSG_SRVR_INVALID_NETBIOS_NAME); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
if( fEndChar) { LPSTR pTemp = lpName; lpName = WinsAllocateMemory(17); if( lpName is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } strcpy(lpName, pTemp); for( i=strlen(pTemp); i<16; i++ ) { lpName[i] = ' '; } lpName[15] = ch16thChar; lpName[16] = '\0';
WinsFreeMemory(pTemp); dwNameLen = 16; } }
WinsAdd.IPAdd = StringToIpAddress(wcServerIpAdd); WinsAdd.Len = 4; WinsAdd.Type = 0;
pszTempAddr = WinsUnicodeToOem(wcServerIpAdd, NULL);
if( pszTempAddr is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
Status = GetDbRecs(MinVer, MaxVer, &WinsAdd, pszTempAddr, fName, lpName, dwNameLen, FALSE, 0, FALSE, fCase, fFile, fFile ? wcFile : NULL);
WinsFreeMemory(pszTempAddr); pszTempAddr = NULL;
if( lpName ) { WinsFreeMemory(lpName); lpName = NULL; } if( Status isnot NO_ERROR ) goto ErrorReturn; DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS);CommonReturn: if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; }
if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; }
return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_SHOW_RECBYVERSION, Status); goto CommonReturn;}
DWORDHandleSrvrShowName( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Displays detail information for a particular name recordsArguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : Name of the records Optional : 16th character and ScopeReturn Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount = 0; TAG_TYPE pttTags[] = { {WINS_TOKEN_NAME, TRUE, FALSE}, {WINS_TOKEN_ENDCHAR, FALSE, FALSE}, {WINS_TOKEN_SCOPE, FALSE, FALSE} }; PDWORD pdwTagNum = NULL, pdwTagType = NULL;
WCHAR wszName[MAX_STRING_LEN+4] = {L'\0'}; BOOL fEndChar = FALSE; CHAR ch16thChar = 0x00; BOOL fScope = FALSE; WCHAR wszScope[MAX_STRING_LEN] = {L'\0'}; DWORD dwStrLen = 0; LPWSTR pwszTemp = NULL; WINSINTF_RECORD_ACTION_T RecAction = {0}; PWINSINTF_RECORD_ACTION_T pRecAction = NULL; LPWSTR pwszGroup = NULL, pwszState = NULL, pwszType = NULL; CHAR chEndChar = (CHAR)0x00;
WCHAR wszGroup[50] = {L'\0'}, wszState[50] = {L'\0'}, wszType[50] = {L'\0'}; DWORD dwTempLen = 0;
struct tm* time = NULL; DWORD dwGroup = WINS_GROUP_UNIQUE; DWORD dwState = WINS_STATE_ACTIVE; DWORD dwType = WINS_TYPE_STATIC; DWORD dwGrouplen = 0, dwStatelen = 0, dwTypelen = 0; LPSTR pszTemp = NULL; LPWSTR pwszTime = NULL; int iType = 1;
memset(&RecAction, 0x00, sizeof(WINSINTF_RECORD_ACTION_T)); RecAction.fStatic = TRUE;
if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, HLP_SRVR_SHOW_NAME_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dwNumArgs = dwArgCount - dwCurrentIndex;
pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum); if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
i = dwTagCount; for( j=0; j<i; j++ ) { switch(pdwTagType[j]) { case 0: { DWORD dwLen = wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); if( dwLen is 0 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dwLen = ( 255 < dwLen ) ? 255 : dwLen; memcpy(wszName, ppwcArguments[dwCurrentIndex+pdwTagNum[j]], dwLen*sizeof(WCHAR)); wszName[dwLen] = L'\0'; break; } case 1: { DWORD dwLen = 0, k=0; fEndChar = TRUE; dwLen = wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); if( dwLen > 2 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } for( k=0; k<dwLen; k++ ) { WCHAR wc = ppwcArguments[dwCurrentIndex+pdwTagNum[j]][k]; if( towlower(wc) >= L'a' and towlower(wc) <= L'z' ) { if( towlower(wc) > L'f' ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } } ch16thChar = StringToHexA(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); break; } case 2: { DWORD dwLen = wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); fScope = TRUE; if( dwLen is 0 ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } dwLen = ( 255 < dwLen ) ? 255 : dwLen; memcpy(wszScope, ppwcArguments[dwCurrentIndex+pdwTagNum[j]], dwLen*sizeof(WCHAR)); wszScope[dwLen] = L'\0'; break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } } _wcsupr(wszName); _wcsupr(wszScope); wszName[16] = L'\0';
pszTemp = WinsUnicodeToOem(wszName, NULL);
if( pszTemp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwStrLen = strlen(pszTemp);
if( dwStrLen >= 16 ) { DisplayMessage(g_hModule, EMSG_SRVR_INVALID_NETBIOS_NAME); Status = ERROR_INVALID_PARAMETER; WinsFreeMemory(pszTemp); pszTemp = NULL; goto ErrorReturn; }
RecAction.pName = WinsAllocateMemory(273);
if( RecAction.pName is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
strncpy(RecAction.pName, pszTemp, 16); WinsFreeMemory(pszTemp); pszTemp = NULL;
for( i = dwStrLen; i<16; i++ ) { RecAction.pName[i] = ' '; }
if( fEndChar ) { RecAction.pName[15] = ch16thChar; } if( fEndChar and ch16thChar is 0x00 ) RecAction.pName[15] = 0x00; RecAction.pName[16] = '\0';
dwStrLen = 16;
if( fScope ) { DWORD dwLen;
RecAction.pName[dwStrLen] = '.'; pszTemp = WinsUnicodeToOem(wszScope, NULL); if( pszTemp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwLen = strlen(pszTemp); dwLen = ( 255 - dwStrLen < dwLen ) ? 255 - dwStrLen : dwLen; strncpy(RecAction.pName + dwStrLen + 1, pszTemp, dwLen);
WinsFreeMemory(pszTemp); pszTemp = NULL;
RecAction.pName[dwLen + dwStrLen + 1] = '\0'; if( fEndChar and ch16thChar is 0x00 ) dwStrLen = 16+dwLen+1; else dwStrLen = strlen(RecAction.pName); } else {
RecAction.pName[dwStrLen] = '\0'; }
RecAction.NameLen = dwStrLen;
RecAction.Cmd_e = WINSINTF_E_QUERY;
RecAction.OwnerId = StringToIpAddress(g_ServerIpAddressUnicodeString); RecAction.NameLen = dwStrLen; pRecAction = &RecAction;
Status = WinsRecordAction(g_hBind, &pRecAction);
if( Status isnot NO_ERROR ) goto ErrorReturn;
RecAction.pName[RecAction.NameLen] = L'\0';
memset(wszName, 0x00, MAX_STRING_LEN*sizeof(WCHAR)); if( pRecAction->NameLen >= 16 ) { chEndChar = pRecAction->pName[15]; pRecAction->pName[15] = 0x00; } else { chEndChar = pRecAction->pName[pRecAction->NameLen]; pRecAction->pName[pRecAction->NameLen] = 0x00; } pwszTemp = WinsOemToUnicode(pRecAction->pName, NULL); if( pwszTemp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } wcscpy(wszName, pwszTemp);
WinsFreeMemory(pwszTemp); pwszTemp = NULL;
for( i=wcslen(wszName); i<16; i++ ) { wszName[i] = L' '; }
wszName[15] = L'['; WinsHexToString(wszName+16, (LPBYTE)&chEndChar, 1); wszName[18] = L'h'; wszName[19] = L']';
if( pRecAction->NameLen > 16 ) { pwszTemp = WinsOemToUnicode(pRecAction->pName+16, NULL); if( pwszTemp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
wcscpy(wszName + 20 , pwszTemp); WinsFreeMemory(pwszTemp); pwszTemp = NULL; wszName[wcslen(wszName)] = L'\0'; } else { wszName[20] = L'\0'; } if( pRecAction->pName[15] is 0x1C ) { pwszGroup = L"DOMAIN NAME"; dwGroup = WINS_GROUP_DOMAIN; } else if( pRecAction->TypOfRec_e is WINSINTF_E_UNIQUE ) { pwszGroup = L"UNIQUE"; dwGroup = WINS_GROUP_UNIQUE; } else if( pRecAction->TypOfRec_e is WINSINTF_E_NORM_GROUP ) { pwszGroup = L"GROUP"; dwGroup = WINS_GROUP_GROUP; } else if( pRecAction->TypOfRec_e is WINSINTF_E_SPEC_GROUP ) { pwszGroup = L"INTERNET GROUP"; dwGroup = WINS_GROUP_INTERNET; } else { pwszGroup = L"MULTIHOMED"; dwGroup = WINS_GROUP_MULTIHOMED; }
//Load the group string
{ dwGrouplen = LoadStringW(g_hModule, dwGroup, wszGroup, sizeof(wszGroup)/sizeof(WCHAR));
if( dwGrouplen is 0 ) wcscpy(wszGroup, pwszGroup); }
pwszTime = GetDateTimeString(pRecAction->TimeStamp, FALSE, &iType); if( pRecAction->State_e is WINSINTF_E_ACTIVE ) { pwszState = L"ACTIVE"; dwState = WINS_STATE_ACTIVE; } else if( pRecAction->State_e is WINSINTF_E_RELEASED ) { dwState = WINS_STATE_RELEASED; pwszState = L"RELEASED"; } else { dwState = WINS_STATE_TOMBSTONE; pwszState = L"TOMBSTONE"; }
//Load the State string
{ dwStatelen = LoadStringW(g_hModule, dwState, wszState, sizeof(wszState)/sizeof(WCHAR));
if( dwStatelen is 0 ) wcscpy(wszState, pwszState); }
if( pRecAction->fStatic ) { dwType = WINS_TYPE_STATIC; pwszType = L"STATIC"; } else { dwType = WINS_TYPE_DYNAMIC; pwszType = L"DYNAMIC"; }
//Load the State string
{ dwTypelen = LoadStringW(g_hModule, dwType, wszType, sizeof(wszType)/sizeof(WCHAR));
if( dwTypelen is 0 ) wcscpy(wszType, pwszType); } DisplayMessage( g_hModule, MSG_WINS_DISPLAY_NAME, wszName, pRecAction->NodeTyp, wszState, iType ? wszInfinite : pwszTime, wszGroup, pRecAction->VersNo.HighPart, pRecAction->VersNo.LowPart, wszType); if( pwszTime ) { WinsFreeMemory(pwszTime); pwszTime = NULL; }
if ( ( pRecAction->pName[15] isnot 0x1C and pRecAction->pName[15] isnot 0x1E ) and ( pRecAction->TypOfRec_e is WINSINTF_E_UNIQUE or pRecAction->TypOfRec_e is WINSINTF_E_NORM_GROUP ) ) { DisplayMessage(g_hModule, MSG_WINS_IPADDRESS_STRING, IpAddressToString(pRecAction->Add.IPAdd)); } else { for (i=0; i<pRecAction->NoOfAdds; ) { DisplayMessage(g_hModule, MSG_WINS_OWNER_ADDRESS, IpAddressToString((pRecAction->pAdd + i++)->IPAdd)); DisplayMessage(g_hModule, MSG_WINS_MEMBER_ADDRESS, IpAddressToString((pRecAction->pAdd + i++)->IPAdd)); } }
//If UNIQUE, look for 0x00 and 0x03 records also
if( ( pRecAction->TypOfRec_e is WINSINTF_E_UNIQUE or pRecAction->TypOfRec_e is WINSINTF_E_MULTIHOMED ) and fEndChar is FALSE ) { DWORD dwNameLen = RecAction.NameLen;
DisplayMessage(g_hModule, WINS_FORMAT_LINE);
RecAction.pName[15] = 0x00; RecAction.NameLen = 16; pRecAction = &RecAction;
Status = WinsRecordAction(g_hBind, &pRecAction);
if( Status is NO_ERROR ) { RecAction.pName[RecAction.NameLen] = L'\0';
memset(wszName, 0x00, MAX_STRING_LEN*sizeof(WCHAR)); pwszTemp = WinsOemToUnicode(pRecAction->pName, NULL); if( pwszTemp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } wcscpy(wszName, pwszTemp); WinsFreeMemory(pwszTemp); pwszTemp = NULL;
for( i=wcslen(wszName); i<16; i++ ) { wszName[i] = L' '; }
for( i=wcslen(wszName)+3; j>=15; j-- ) wszName[j-1] = wszName[j-4]; wszName[15] = L'['; WinsHexToString(wszName+16, (LPBYTE)(pRecAction->pName+15), 1); wszName[18] = L'h'; wszName[19] = L']';
if( pRecAction->NameLen > 16 ) { pwszTemp = WinsOemToUnicode(pRecAction->pName+16, NULL); if( pwszTemp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
wcscpy(wszName + 20 , pwszTemp); WinsFreeMemory(pwszTemp); pwszTemp = NULL; wszName[wcslen(wszName)] = L'\0'; } else { wszName[20] = L'\0'; }
iType = 1; pwszTime = GetDateTimeString(pRecAction->TimeStamp, FALSE, &iType);
DisplayMessage( g_hModule, MSG_WINS_DISPLAY_NAME, wszName, pRecAction->NodeTyp, wszState, iType ? wszInfinite : pwszTime, wszGroup, pRecAction->VersNo.HighPart, pRecAction->VersNo.LowPart, wszType); if( pwszTime ) { WinsFreeMemory(pwszTime); pwszTime = NULL; }
if( pRecAction->TypOfRec_e is WINSINTF_E_UNIQUE ) { DisplayMessage(g_hModule, MSG_WINS_IPADDRESS_STRING, IpAddressToString(pRecAction->Add.IPAdd)); } else { for (i=0; i<pRecAction->NoOfAdds; ) { DisplayMessage(g_hModule, MSG_WINS_OWNER_ADDRESS, IpAddressToString((pRecAction->pAdd + i++)->IPAdd)); DisplayMessage(g_hModule, MSG_WINS_MEMBER_ADDRESS, IpAddressToString((pRecAction->pAdd + i++)->IPAdd)); } } } DisplayMessage(g_hModule, WINS_FORMAT_LINE);
//Now Look for 0x03 record
if( fScope ) { DWORD dwLen; dwStrLen = 16; RecAction.pName[dwStrLen] = '.'; pszTemp = WinsUnicodeToOem(wszScope, NULL); if( pszTemp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } dwLen = strlen(pszTemp); dwLen = ( 255 - dwStrLen < dwLen ) ? 255 - dwStrLen : dwLen; strncpy(RecAction.pName + dwStrLen + 1, pszTemp, dwLen); WinsFreeMemory(pszTemp); pszTemp = NULL;
RecAction.pName[dwLen + dwStrLen + 1] = '\0'; if( fEndChar and ch16thChar is 0x00 ) dwStrLen = 16+dwLen+1; else dwStrLen = strlen(RecAction.pName); } else {
RecAction.pName[dwStrLen] = '\0'; }
RecAction.pName[15] = 0x03; RecAction.NameLen = dwNameLen;
pRecAction = &RecAction; Status = WinsRecordAction(g_hBind, &pRecAction);
if( Status is NO_ERROR ) { CHAR chEndChar = pRecAction->pName[15];
RecAction.pName[RecAction.NameLen] = L'\0';
memset(wszName, 0x00, MAX_STRING_LEN*sizeof(WCHAR));
if( pRecAction->NameLen >= 16 ) { chEndChar = pRecAction->pName[15]; pRecAction->pName[15] = 0x00; } else { chEndChar = pRecAction->pName[pRecAction->NameLen]; pRecAction->pName[pRecAction->NameLen] = 0x00; } pwszTemp = WinsOemToUnicode(pRecAction->pName, NULL); if( pwszTemp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } wcscpy(wszName, pwszTemp); WinsFreeMemory(pwszTemp); pwszTemp = NULL;
for( i = wcslen(wszName); i < 16; i++ ) { wszName[i] = L' '; }
for( i=wcslen(wszName)+3; j>=15; j-- ) wszName[j-1] = wszName[j-4];
wszName[15] = L'['; WinsHexToString(wszName+16, (LPBYTE)&chEndChar, 1); wszName[18] = L'h'; wszName[19] = L']';
if( pRecAction->NameLen > 16 ) { pwszTemp = WinsOemToUnicode(pRecAction->pName+16, NULL); if( pwszTemp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } wcscpy(wszName + 20 , pwszTemp); WinsFreeMemory(pwszTemp); pwszTemp = NULL; wszName[wcslen(wszName)] = L'\0'; } else { wszName[20] = L'\0'; } iType = 1; pwszTime = GetDateTimeString(pRecAction->TimeStamp, FALSE, &iType);
DisplayMessage(g_hModule, MSG_WINS_DISPLAY_NAME, wszName, pRecAction->NodeTyp, wszState, iType ? wszInfinite : pwszTime, wszGroup, pRecAction->VersNo.HighPart, pRecAction->VersNo.LowPart, wszType); if( pwszTime ) { WinsFreeMemory(pwszTime); pwszTime = NULL; }
if( pRecAction->TypOfRec_e is WINSINTF_E_UNIQUE ) { DisplayMessage(g_hModule, MSG_WINS_IPADDRESS_STRING, IpAddressToString(pRecAction->Add.IPAdd)); } else { for (i=0; i<pRecAction->NoOfAdds; ) { DisplayMessage(g_hModule, MSG_WINS_OWNER_ADDRESS, IpAddressToString((pRecAction->pAdd + i++)->IPAdd)); DisplayMessage(g_hModule, MSG_WINS_MEMBER_ADDRESS, IpAddressToString((pRecAction->pAdd + i++)->IPAdd)); } } } }
CommonReturn: if( Status is NO_ERROR ) DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS);
if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; }
if( pszTemp ) { WinsFreeMemory(pszTemp); pszTemp = NULL; }
if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; } if( RecAction.pName ) { WinsFreeMemory(RecAction.pName); RecAction.pName = NULL; } return Status;ErrorReturn: DisplayErrorMessage(EMSG_SRVR_SHOW_NAME, Status); goto CommonReturn;}
DWORDHandleSrvrShowServer( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Displays the current WINS serverArguments : All aguments are passes as array of wide char strings in ppwcArguments. NONE.Return Value: Returns the status of the operation.
--*/{ DisplayMessage(g_hModule, MSG_WINS_SERVER_NAME, g_ServerNameUnicode, g_ServerIpAddressUnicodeString); return NO_ERROR; }
DWORDHandleSrvrShowStatistics( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Displays the Server statisticsArguments : All aguments are passes as array of wide char strings in ppwcArguments. NONE.Return Value: Returns the status of the operation.
--*/{
DWORD Status = NO_ERROR; WINSINTF_RESULTS_T Results = {0}; WINSINTF_RESULTS_NEW_T ResultsN = {0}; WCHAR IpAddress[33] = {L'\0'}; WCHAR RepCount[25] = {L'\0'}; WCHAR Buffer[8] = {L'\0'}; BOOL fNew = TRUE; DWORD i = 0, k = 0;
ResultsN.WinsStat.NoOfPnrs = 0; ResultsN.WinsStat.pRplPnrs = NULL; ResultsN.pAddVersMaps = NULL;
Status = WinsStatusNew(g_hBind, WINSINTF_E_STAT, &ResultsN); if( Status is RPC_S_PROCNUM_OUT_OF_RANGE ) { //Try old API
Results.WinsStat.NoOfPnrs = 0; Results.WinsStat.pRplPnrs = 0; Status = WinsStatus(g_hBind, WINSINTF_E_CONFIG_ALL_MAPS, &Results); fNew = FALSE; } if( Status isnot NO_ERROR ) { DisplayErrorMessage(EMSG_SRVR_SHOW_STATISTICS, Status); }
if( fNew ) { DisplayMessage(g_hModule, MSG_WINS_TIMESTAMP, TMSTN.WinsStartTime.wMonth, TMSTN.WinsStartTime.wDay, TMSTN.WinsStartTime.wYear, TMSTN.WinsStartTime.wHour, TMSTN.WinsStartTime.wMinute, TMSTN.WinsStartTime.wSecond );
DisplayMessage(g_hModule, MSG_WINS_LAST_INIT, TIME_ARGSN(LastInitDbTime));
DisplayMessage(g_hModule, MSG_WINS_PLANNED_SCV, TIME_ARGSN(LastPScvTime)); DisplayMessage(g_hModule, MSG_WINS_TRIGGERED_SCV, TIME_ARGSN(LastATScvTime));
DisplayMessage(g_hModule, MSG_WINS_TOMBSTONE_SCV, TIME_ARGSN(LastTombScvTime));
DisplayMessage(g_hModule, MSG_WINS_REPLICA_VERIFICATION, TIME_ARGSN(LastVerifyScvTime));
DisplayMessage(g_hModule, MSG_WINS_PLANNED_REPLICATION, TMSTN.LastPRplTime.wMonth, TMSTN.LastPRplTime.wDay, TMSTN.LastPRplTime.wYear, TMSTN.LastPRplTime.wHour, TMSTN.LastPRplTime.wMinute, TMSTN.LastPRplTime.wSecond );
DisplayMessage(g_hModule, MSG_WINS_TRIGGERED_REPLICATION, TMSTN.LastATRplTime.wMonth, TMSTN.LastATRplTime.wDay, TMSTN.LastATRplTime.wYear, TMSTN.LastATRplTime.wHour, TMSTN.LastATRplTime.wMinute, TMSTN.LastATRplTime.wSecond );
DisplayMessage(g_hModule, MSG_WINS_RESET_COUNTER, TIME_ARGSN(CounterResetTime));
DisplayMessage(g_hModule, WINS_FORMAT_LINE); DisplayMessage(g_hModule, MSG_WINS_COUNTER_INFORMATION, ResultsN.WinsStat.Counters.NoOfUniqueReg, ResultsN.WinsStat.Counters.NoOfGroupReg, ResultsN.WinsStat.Counters.NoOfSuccQueries, ResultsN.WinsStat.Counters.NoOfFailQueries, ResultsN.WinsStat.Counters.NoOfUniqueRef, ResultsN.WinsStat.Counters.NoOfGroupRef, ResultsN.WinsStat.Counters.NoOfSuccRel, ResultsN.WinsStat.Counters.NoOfFailRel, ResultsN.WinsStat.Counters.NoOfUniqueCnf, ResultsN.WinsStat.Counters.NoOfGroupCnf ); if (ResultsN.WinsStat.NoOfPnrs) { DisplayMessage(g_hModule, MSG_WINS_PARTNER_TABLE);
for (i =0; i < ResultsN.WinsStat.NoOfPnrs; i++) { LPTSTR pstr;
for(k=0; k<32; k++) { IpAddress[k] = L' '; } IpAddress[32] = L'\0'; for(k=0; k<24; k++) RepCount[k] = L' '; RepCount[24] = L'\0'; pstr = IpAddressToString((ResultsN.WinsStat.pRplPnrs + i)->Add.IPAdd); if (pstr == NULL) { Status = ERROR_NOT_ENOUGH_MEMORY; break; } wcscpy(IpAddress+8, pstr); IpAddress[wcslen(IpAddress)] = L' '; _itow((ResultsN.WinsStat.pRplPnrs + i)->NoOfRpls, Buffer, 10); wcscpy(RepCount+8, Buffer); RepCount[wcslen(RepCount)] = L' ';
DisplayMessage(g_hModule, MSG_WINS_PARTNER_INFO, IpAddress, RepCount, (ResultsN.WinsStat.pRplPnrs + i)->NoOfCommFails ); } }
WinsFreeMem(ResultsN.pAddVersMaps); ResultsN.pAddVersMaps = NULL; WinsFreeMem(ResultsN.WinsStat.pRplPnrs); ResultsN.WinsStat.pRplPnrs = NULL; } else { DisplayMessage(g_hModule, MSG_WINS_TIMESTAMP, TMST.WinsStartTime.wMonth, TMST.WinsStartTime.wDay, TMST.WinsStartTime.wYear, TMST.WinsStartTime.wHour, TMST.WinsStartTime.wMinute, TMST.WinsStartTime.wSecond );
DisplayMessage(g_hModule, MSG_WINS_LAST_INIT, TIME_ARGS(LastInitDbTime));
DisplayMessage(g_hModule, MSG_WINS_PLANNED_SCV, TIME_ARGS(LastPScvTime)); DisplayMessage(g_hModule, MSG_WINS_TRIGGERED_SCV, TIME_ARGS(LastATScvTime));
DisplayMessage(g_hModule, MSG_WINS_TOMBSTONE_SCV, TIME_ARGS(LastTombScvTime));
DisplayMessage(g_hModule, MSG_WINS_REPLICA_VERIFICATION, TIME_ARGS(LastVerifyScvTime));
DisplayMessage(g_hModule, MSG_WINS_PLANNED_REPLICATION, TMST.LastPRplTime.wMonth, TMST.LastPRplTime.wDay, TMST.LastPRplTime.wYear, TMST.LastPRplTime.wHour, TMST.LastPRplTime.wMinute, TMST.LastPRplTime.wSecond );
DisplayMessage(g_hModule, MSG_WINS_TRIGGERED_REPLICATION, TMST.LastATRplTime.wMonth, TMST.LastATRplTime.wDay, TMST.LastATRplTime.wYear, TMST.LastATRplTime.wHour, TMST.LastATRplTime.wMinute, TMST.LastATRplTime.wSecond );
DisplayMessage(g_hModule, MSG_WINS_RESET_COUNTER, TIME_ARGS(CounterResetTime));
DisplayMessage(g_hModule, WINS_FORMAT_LINE); DisplayMessage(g_hModule, MSG_WINS_COUNTER_INFORMATION, Results.WinsStat.Counters.NoOfUniqueReg, Results.WinsStat.Counters.NoOfGroupReg, Results.WinsStat.Counters.NoOfSuccQueries, Results.WinsStat.Counters.NoOfFailQueries, Results.WinsStat.Counters.NoOfUniqueRef, Results.WinsStat.Counters.NoOfGroupRef, Results.WinsStat.Counters.NoOfSuccRel, Results.WinsStat.Counters.NoOfFailRel, Results.WinsStat.Counters.NoOfUniqueCnf, Results.WinsStat.Counters.NoOfGroupCnf ); if (Results.WinsStat.NoOfPnrs) { DisplayMessage(g_hModule, MSG_WINS_PARTNER_TABLE);
for (i =0; i < Results.WinsStat.NoOfPnrs; i++) {
for(k=0; k<32; k++) { IpAddress[k] = L' '; } IpAddress[32] = L'\0'; for(k=0; k<24; k++) RepCount[k] = L' '; RepCount[24] = L'\0'; wcscpy(IpAddress+8, IpAddressToString((Results.WinsStat.pRplPnrs + i)->Add.IPAdd)); IpAddress[wcslen(IpAddress)] = L' '; _itow((Results.WinsStat.pRplPnrs + i)->NoOfRpls, Buffer, 10); wcscpy(RepCount+8, Buffer); RepCount[wcslen(RepCount)] = L' ';
DisplayMessage(g_hModule, MSG_WINS_PARTNER_INFO, IpAddress, RepCount, (Results.WinsStat.pRplPnrs + i)->NoOfCommFails ); } }
WinsFreeMem(Results.WinsStat.pRplPnrs); Results.WinsStat.pRplPnrs = NULL;
}
DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); return Status; }
DWORDHandleSrvrShowVersion( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Displas the current version counter value.Arguments : All aguments are passes as array of wide char strings in ppwcArguments. NONE.Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; WINSINTF_RESULTS_T Results;
Results.AddVersMaps[0].Add.Len = 4; Results.AddVersMaps[0].Add.Type = 0; Results.AddVersMaps[0].Add.IPAdd = StringToIpAddress(g_ServerIpAddressUnicodeString); Results.WinsStat.NoOfPnrs = 0; Results.WinsStat.pRplPnrs = NULL; Status = WinsStatus(g_hBind, WINSINTF_E_ADDVERSMAP, &Results); if( Status isnot NO_ERROR ) { DisplayErrorMessage(EMSG_SRVR_SHOW_VERSION, Status); return Status; }
DisplayMessage(g_hModule, MSG_WINS_VERSION_INFO, g_ServerIpAddressUnicodeString, Results.AddVersMaps[0].VersNo.HighPart, Results.AddVersMaps[0].VersNo.LowPart);
return Status; }
DWORDHandleSrvrShowVersionmap( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Displays the version mappingArguments : All aguments are passes as array of wide char strings in ppwcArguments. NONE.Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; WINSINTF_RESULTS_T Results = {0}; WINSINTF_RESULTS_NEW_T ResultsN = {0}; LPSTR pszIp = NULL;
ResultsN.WinsStat.NoOfPnrs = 0; ResultsN.WinsStat.pRplPnrs = NULL; pszIp = WinsUnicodeToOem(g_ServerIpAddressUnicodeString, NULL); if( pszIp is NULL ) { DisplayMessage(g_hModule, EMSG_WINS_OUT_OF_MEMORY); Status = ERROR_NOT_ENOUGH_MEMORY; return Status; }
Status = GetStatus(TRUE, (LPVOID)&ResultsN, TRUE, TRUE, pszIp); if( Status is RPC_S_PROCNUM_OUT_OF_RANGE ) { //Try old API
Results.WinsStat.NoOfPnrs = 0; Results.WinsStat.pRplPnrs = 0; Status = GetStatus(TRUE, (LPVOID)&Results, FALSE, TRUE, pszIp); }
WinsFreeMemory(pszIp); pszIp = NULL; if( ResultsN.pAddVersMaps ) { WinsFreeMem(ResultsN.pAddVersMaps); ResultsN.pAddVersMaps = NULL; }
if( Status isnot NO_ERROR ) { DisplayErrorMessage(EMSG_SRVR_SHOW_VERSIONMAP, Status); return Status; } DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS); return Status;
}
DWORDHandleSrvrShowPartnerproperties( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Displays the default partner propertiesArguments : All aguments are passes as array of wide char strings in ppwcArguments. NONE.Return Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; HKEY hServer = NULL, hDefaultPull = NULL, hDefaultPush = NULL, hPullRoot = NULL, hPushRoot = NULL, hPartner = NULL, hParameter = NULL; LPWSTR pTemp = NULL, pServerList = NULL; WCHAR wcBuffer[255] = {L'\0'};
DWORD dwRplWCnfPnrs = 0, dwMigrate = 0, dwData = 0, i = 0, dwPersonaMode = 0, dwSelfFndPnrs = 0;
DWORD dwType = REG_DWORD; DWORD dwSize = sizeof(DWORD); LPBYTE pbData = NULL;
if( wcslen(g_ServerNetBiosName) > 0 ) { pTemp = g_ServerNetBiosName; }
Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer); if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegOpenKeyEx(hServer, PARAMETER, 0, KEY_READ, &hParameter);
if( Status isnot NO_ERROR ) { goto ErrorReturn; }
DisplayMessage(g_hModule, WINS_FORMAT_LINE);
Status = RegQueryValueEx(hParameter, WINSCNF_RPL_ONLY_W_CNF_PNRS_NM, 0, &dwType, (LPBYTE)&dwRplWCnfPnrs, &dwSize); if( Status is NO_ERROR ) { DisplayMessage(g_hModule, MSG_SRVR_REPLICATE_STATE, dwRplWCnfPnrs ? wszEnable : wszDisable); } else { DisplayMessage(g_hModule, MSG_SRVR_REPLICATE_STATE, wszUnknown); }
dwMigrate = 0; dwSize = sizeof(DWORD);
Status = RegQueryValueEx(hParameter, WINSCNF_MIGRATION_ON_NM, 0, &dwType, (LPBYTE)&dwMigrate, &dwSize); if( Status is NO_ERROR ) { DisplayMessage(g_hModule, MSG_SRVR_MIGRATE_STATE, dwMigrate ? wszEnable : wszDisable); } else { DisplayMessage(g_hModule, MSG_SRVR_MIGRATE_STATE, wszUnknown); }
DisplayMessage(g_hModule, WINS_FORMAT_LINE); DisplayMessage(g_hModule, MSG_SRVR_AUTOCONFIGURE);
Status = RegQueryValueEx(hParameter, WINSCNF_USE_SELF_FND_PNRS_NM, 0, &dwType, (LPBYTE)&dwSelfFndPnrs, &dwSize); if( Status isnot NO_ERROR ) { DisplayMessage(g_hModule, MSG_SRVR_SELFFINDPNRS_STATE, wszUnknown); } else { DisplayMessage(g_hModule, MSG_SRVR_SELFFINDPNRS_STATE, dwSelfFndPnrs ? wszEnable : wszDisable); if( dwSelfFndPnrs > 0 ) { dwData = 0; dwSize = sizeof(DWORD); Status = RegQueryValueEx(hParameter, WINSCNF_MCAST_INTVL_NM, 0, &dwType, (LPBYTE)&dwData, &dwSize); if(Status is NO_ERROR ) { LPWSTR pwszTime = MakeTimeString(dwData); if( pwszTime is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } DisplayMessage(g_hModule, MSG_SRVR_MCAST_INTERVAL, pwszTime); WinsFreeMemory(pwszTime); pwszTime = NULL;
} else { DisplayMessage(g_hModule, MSG_SRVR_MCAST_INTERVAL, wszUnknown); }
dwData = 0; dwSize = sizeof(DWORD); Status = RegQueryValueEx(hParameter, WINSCNF_MCAST_TTL_NM, 0, &dwType, (LPBYTE)&dwData, &dwSize); if( Status is NO_ERROR ) { WCHAR Buffer[10] = {L'\0'}; _itow(dwData, Buffer, 10); DisplayMessage(g_hModule, MSG_SRVR_MCAST_TTL, Buffer); } else { DisplayMessage(g_hModule, MSG_SRVR_MCAST_TTL, wszUnknown); }
} }
//Display PNG Servers
DisplayMessage(g_hModule, WINS_FORMAT_LINE);
Status = RegOpenKeyEx(hServer, PARTNERROOT, 0, KEY_READ, &hPartner); if( Status isnot NO_ERROR ) goto ErrorReturn;
dwSize = sizeof(DWORD); Status = RegQueryValueEx(hPartner, TEXT(WINSCNF_PERSONA_MODE_NM), 0, &dwType, (LPVOID)&dwPersonaMode, &dwSize);
DisplayMessage(g_hModule, dwPersonaMode == PERSMODE_NON_GRATA ? MSG_WINS_PNGSERVER_TABLE : MSG_WINS_PGSERVER_TABLE);
dwSize = 0; Status = RegQueryValueEx(hPartner, dwPersonaMode == PERSMODE_NON_GRATA ? PNGSERVER : PGSERVER, 0, &dwType, NULL, &dwSize);
if( dwSize <= 2 ) { DisplayMessage(g_hModule, dwPersonaMode == PERSMODE_NON_GRATA ? MSG_WINS_NO_PNGSERVER : MSG_WINS_NO_PGSERVER); }
else { pbData = WinsAllocateMemory(dwSize);
if( pbData is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } Status = RegQueryValueEx(hPartner, dwPersonaMode == PERSMODE_NON_GRATA ? PNGSERVER : PGSERVER, 0, &dwType, pbData, &dwSize);
if( Status isnot NO_ERROR ) goto ErrorReturn;
pServerList = (LPWSTR)pbData;
i = 0; while(TRUE) { if ( i+1 < dwSize/sizeof(WCHAR) ) { if( pServerList[i] is L'\0' and pServerList[i+1] isnot L'\0' ) { pServerList[i] = L','; } i++; } else break; }
pTemp = wcstok(pServerList, L",");
while(pTemp isnot NULL ) { DisplayMessage(g_hModule, dwPersonaMode == PERSMODE_NON_GRATA ? MSG_WINS_PNGSERVER_ENTRY : MSG_WINS_PGSERVER_ENTRY, pTemp); pTemp = wcstok(NULL, L","); } }
Status = RegOpenKeyEx(hServer, DEFAULTPULL, 0, KEY_READ, &hDefaultPull); if( Status is ERROR_FILE_NOT_FOUND ) { DisplayMessage(g_hModule, MSG_NO_DEFAULT_PULL); } else if( Status isnot NO_ERROR ) { goto ErrorReturn; }
Status = RegOpenKeyEx(hServer, PULLROOT, 0, KEY_READ, &hPullRoot);
if( Status is ERROR_FILE_NOT_FOUND ) { DisplayMessage(g_hModule, MSG_NO_DEFAULT_PULL); } else if( Status isnot NO_ERROR ) goto ErrorReturn;
DisplayMessage(g_hModule, WINS_FORMAT_LINE); DisplayMessage(g_hModule, MSG_SRVR_PULL_INFO);
dwData = 0; dwSize = sizeof(DWORD);
if( hPullRoot isnot NULL ) { Status = RegQueryValueEx(hPullRoot, PERSISTENCE, 0, &dwType, (LPBYTE)&dwData, &dwSize); if( Status is NO_ERROR ) { DisplayMessage(g_hModule, MSG_SRVR_PULL_PERSISTENCE_STATE, dwData ? wszEnable : wszDisable); } else { DisplayMessage(g_hModule, MSG_SRVR_PULL_PERSISTENCE_STATE, wszUnknown); }
dwData = 0; dwSize = sizeof(DWORD);
Status = RegQueryValueEx(hPullRoot, WINSCNF_INIT_TIME_RPL_NM, 0, &dwType, (LPBYTE)&dwData, &dwSize); if( Status is NO_ERROR ) { DisplayMessage(g_hModule, MSG_SRVR_PULL_INITTIMEREPL_STATE, dwData ? wszEnable : wszDisable); } else { DisplayMessage(g_hModule, MSG_SRVR_PULL_INITTIMEREPL_STATE, wszUnknown); } }
dwData = 0; dwSize = 255;
if( hDefaultPull isnot NULL ) { Status = RegQueryValueEx(hDefaultPull, WINSCNF_SP_TIME_NM, 0, &dwType, (LPBYTE)wcBuffer, &dwSize); if( Status is NO_ERROR ) { DisplayMessage(g_hModule, MSG_SRVR_PULL_STARTTIME, wcBuffer); } else { DisplayMessage(g_hModule, MSG_SRVR_PULL_STARTTIME, wszUnknown); } dwSize = sizeof(DWORD); dwData = 0;
Status = RegQueryValueEx(hDefaultPull, WINSCNF_RPL_INTERVAL_NM, 0, &dwType, (LPBYTE)&dwData, &dwSize);
if( Status is NO_ERROR ) { LPWSTR pwszDayString = MakeDayTimeString(dwData); DisplayMessage(g_hModule, MSG_WINS_PULL_REPLINTERVAL, MakeDayTimeString(dwData)); WinsFreeMemory(pwszDayString); pwszDayString = NULL; } else { DisplayMessage(g_hModule, MSG_WINS_PULL_REPLINTERVAL, wszUnknown); } }
dwData = 0; dwSize = sizeof(DWORD);
if( hPullRoot isnot NULL ) { Status = RegQueryValueEx(hPullRoot, WINSCNF_RETRY_COUNT_NM, 0, &dwType, (LPBYTE)&dwData, &dwSize); if( Status is NO_ERROR ) { WCHAR Buffer[20]; _itow(dwData, Buffer, 10);
DisplayMessage(g_hModule, MSG_WINS_PULL_RETRYCOUNT, Buffer);
} else { DisplayMessage(g_hModule, MSG_WINS_PULL_RETRYCOUNT, wszUnknown); } }
Status = RegOpenKeyEx(hServer, DEFAULTPUSH, 0, KEY_READ, &hDefaultPush);
if( Status is ERROR_FILE_NOT_FOUND ) { DisplayMessage(g_hModule, MSG_NO_DEFAULT_PUSH); } else if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegOpenKeyEx(hServer, PUSHROOT, 0, KEY_READ, &hPushRoot); if( Status is ERROR_FILE_NOT_FOUND ) { DisplayMessage(g_hModule, MSG_NO_DEFAULT_PUSH); } else if( Status isnot NO_ERROR ) goto ErrorReturn;
DisplayMessage(g_hModule, WINS_FORMAT_LINE); DisplayMessage(g_hModule, MSG_SRVR_PUSH_INFO);
dwData = 0; dwSize = sizeof(DWORD); if( hPushRoot isnot NULL ) { Status = RegQueryValueEx(hPushRoot, PERSISTENCE, 0, &dwType, (LPBYTE)&dwData, &dwSize); if( Status is NO_ERROR ) { DisplayMessage(g_hModule, MSG_SRVR_PUSH_PERSISTENCE_STATE, dwData ? wszEnable : wszDisable); } else { DisplayMessage(g_hModule, MSG_SRVR_PUSH_PERSISTENCE_STATE, wszUnknown); }
dwData = 0; dwSize = sizeof(DWORD); Status = RegQueryValueEx(hPushRoot, WINSCNF_INIT_TIME_RPL_NM, 0, &dwType, (LPBYTE)&dwData, &dwSize);
if( Status is NO_ERROR ) { DisplayMessage(g_hModule, MSG_SRVR_PUSH_INITTIMEREPL_STATE, dwData ? wszEnable : wszDisable); } else { DisplayMessage(g_hModule, MSG_SRVR_PUSH_INITTIMEREPL_STATE, wszUnknown); }
dwData = 0; dwSize = sizeof(DWORD); Status = RegQueryValueEx(hPushRoot, WINSCNF_ADDCHG_TRIGGER_NM, 0, &dwType, (LPBYTE)&dwData, &dwSize);
if( Status is NO_ERROR ) { DisplayMessage(g_hModule, MSG_SRVR_PUSH_ONADDCHANGE, dwData ? wszEnable : wszDisable); } else { DisplayMessage(g_hModule, MSG_SRVR_PUSH_ONADDCHANGE, wszUnknown); } } dwData = 0; dwSize = sizeof(DWORD); Status = NO_ERROR;
if( hDefaultPush isnot NULL ) { Status = RegQueryValueEx(hDefaultPush, WINSCNF_UPDATE_COUNT_NM, 0, &dwType, (LPBYTE)&dwData, &dwSize); if( Status is NO_ERROR ) { WCHAR Buffer[10] = {L'\0'}; _itow(dwData, Buffer, 10);
DisplayMessage(g_hModule, MSG_SRVR_PUSH_UPDATECOUNT, Buffer); } else { DisplayMessage(g_hModule, MSG_SRVR_PUSH_UPDATECOUNT, wszUnknown); } }
CommonReturn: if( Status is NO_ERROR ) DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS);
if( hPushRoot ) { RegCloseKey(hPushRoot); hPushRoot = NULL; }
if( hPullRoot ) { RegCloseKey(hPullRoot); hPullRoot = NULL; }
if( hDefaultPush ) { RegCloseKey(hDefaultPush); hDefaultPush = NULL; }
if( hDefaultPull ) { RegCloseKey(hDefaultPull); hDefaultPull = NULL; }
if( hParameter ) { RegCloseKey(hParameter); hParameter = NULL; }
if( hPartner ) { RegCloseKey(hPartner); hPartner = NULL; }
if( hServer ) { RegCloseKey(hServer); hServer = NULL; }
if( pbData ) { WinsFreeMemory(pbData); pbData = NULL; }
return Status;
ErrorReturn: if( Status is ERROR_FILE_NOT_FOUND ) { DisplayMessage(g_hModule, MSG_WINS_NO_PARTNER); } else { DisplayErrorMessage(EMSG_SRVR_SHOW_PARTNERPROPERTIES, Status); } goto CommonReturn;
}
DWORDHandleSrvrShowPullpartnerproperties( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Displays the properties for a particular Pull partnerArguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : Pull partner addressReturn Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount; TAG_TYPE pttTags[] = { {WINS_TOKEN_SERVER, TRUE, FALSE}, }; PDWORD pdwTagType = NULL, pdwTagNum = NULL;
WCHAR wcServerIpAdd[MAX_IP_STRING_LEN+1] = {L'\0'}; HKEY hPullRoot = NULL, hPullServer = NULL, hServer = NULL; BOOL fPush = FALSE;
LPWSTR pTemp = NULL;
if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, HLP_SRVR_SHOW_PULLPARTNERPROPERTIES_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dwNumArgs = dwArgCount - dwCurrentIndex; pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum);
if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
for( j=0; j<dwTagCount; j++ ) { switch(pdwTagType[j]) { case 0: { struct hostent * lpHostEnt = NULL; CHAR cAddr[16]; BYTE pbAdd[4]; char szAdd[4]; int k = 0, l=0; DWORD dwLen, nLen = 0; CHAR *pTemp = NULL; CHAR *pNetBios = NULL;
if( IsIpAddress( ppwcArguments[dwCurrentIndex+pdwTagNum[j]] ) ) { DWORD dwIp = inet_addr(WinsUnicodeToAnsi(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL)); lpHostEnt = gethostbyaddr((char *)&dwIp, 4, AF_INET); if(lpHostEnt isnot NULL )//Valid IP Address
{ wcscpy(wcServerIpAdd, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); break; } else { Status = ERROR_INVALID_IPADDRESS; goto ErrorReturn; } }
if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) > 2 and _wcsnicmp(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], L"\\\\", 2) is 0 ) k = 2;
lpHostEnt = gethostbyname(WinsUnicodeToAnsi(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]+k, NULL)); if( lpHostEnt is NULL ) { DisplayMessage(g_hModule, EMSG_WINS_INVALID_COMPUTER_NAME, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]+k); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
memcpy(pbAdd, lpHostEnt->h_addr_list[0], 4); for( l=0;l<4; l++) { _itoa((int)pbAdd[l], szAdd, 10); memcpy(cAddr+nLen, szAdd, strlen(szAdd)); nLen += strlen(szAdd); *(cAddr+nLen) = '.'; nLen++;
} *(cAddr+nLen-1) = '\0'; { LPWSTR pstr = WinsAnsiToUnicode(cAddr, NULL);
if (pstr != NULL) { wcscpy(wcServerIpAdd, pstr); WinsFreeMemory(pstr); } else { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } } break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } }
if( wcslen(g_ServerNetBiosName) > 0 ) { pTemp = g_ServerNetBiosName; }
Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer); if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegOpenKeyEx(hServer, PULLROOT, 0, KEY_READ, &hPullRoot);
if( Status isnot NO_ERROR ) { goto ErrorReturn; }
DisplayMessage(g_hModule, WINS_FORMAT_LINE);
//Now check if the desired server is in the list of pull server.
Status = RegOpenKeyEx(hPullRoot, wcServerIpAdd, 0, KEY_READ, &hPullServer);
if( Status isnot NO_ERROR ) { DisplayMessage(g_hModule, EMSG_WINS_INVALID_PULLPARTNER, wcServerIpAdd, g_ServerIpAddressUnicodeString); goto CommonReturn; }
//Check if it is also a push partner or not.
{ DWORD dwKeyLen = 0; HKEY hPushServer = NULL; dwKeyLen = wcslen(PUSHROOT)+ wcslen(L"\\") + wcslen(wcServerIpAdd); pTemp = WinsAllocateMemory((dwKeyLen+1)*sizeof(WCHAR)); if(pTemp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } memset(pTemp, 0x00, (dwKeyLen+1)*sizeof(WCHAR)); wcscat(pTemp,PUSHROOT); wcscat(pTemp, L"\\"); wcscat(pTemp, wcServerIpAdd);
Status = RegOpenKeyEx(hServer, pTemp, 0, KEY_READ, &hPushServer); if( Status isnot NO_ERROR ) { fPush = FALSE; } else { fPush = TRUE; } WinsFreeMemory(pTemp); pTemp = NULL; if( hPushServer ) { RegCloseKey(hPushServer); hPushServer = NULL; } }
//Now look for required parameters to display
{ WCHAR wcData[256] = {L'\0'}; DWORD dwData = 0; DWORD dwDatalen = 256; DWORD dwType = REG_SZ;
Status = RegQueryValueEx(hPullServer, L"NetBIOSName", 0, &dwType, (LPBYTE)wcData, &dwDatalen); if( Status isnot NO_ERROR ) { DisplayMessage(g_hModule, MSG_WINS_PULLPARTNER_INFO, wcServerIpAdd, wszUnknown, (fPush is TRUE) ? wszPushpull : wszPull); } else { DisplayMessage(g_hModule, MSG_WINS_PULLPARTNER_INFO, wcServerIpAdd, wcData, (fPush is TRUE) ? wszPushpull : wszPull); }
dwDatalen = sizeof(DWORD); dwType = REG_DWORD;
Status = RegQueryValueEx(hPullServer, PERSISTENCE, 0, &dwType, (LPBYTE)&dwData, &dwDatalen); if( Status is NO_ERROR ) { DisplayMessage(g_hModule, MSG_SRVR_PULL_PERSISTENCE_STATE, dwData ? wszEnable : wszDisable); } else { DisplayMessage(g_hModule, MSG_SRVR_PULL_PERSISTENCE_STATE, wszUnknown); }
dwType = REG_SZ; dwDatalen = 256; Status = RegQueryValueEx(hPullServer, WINSCNF_SP_TIME_NM, 0, &dwType, (LPBYTE)wcData, &dwDatalen); if( Status is NO_ERROR ) { DisplayMessage(g_hModule, MSG_SRVR_PULL_STARTTIME, wcData); } else { DisplayMessage(g_hModule, MSG_SRVR_PULL_STARTTIME, L"00:00:00"); } dwDatalen = sizeof(DWORD); dwType = REG_DWORD;
Status = RegQueryValueEx(hPullServer, WINSCNF_RPL_INTERVAL_NM, 0, &dwType, (LPBYTE)&dwData, &dwDatalen);
if( Status is NO_ERROR ) { LPWSTR pwszDayString = MakeDayTimeString(dwData); DisplayMessage(g_hModule, MSG_WINS_PULL_REPLINTERVAL, pwszDayString); WinsFreeMemory(pwszDayString); pwszDayString = NULL; } else { DisplayMessage(g_hModule, MSG_WINS_PULL_REPLINTERVAL, L"00:00:00"); } }
DisplayMessage(g_hModule, WINS_FORMAT_LINE);CommonReturn: if( Status is NO_ERROR ) DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS);
if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; }
if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; }
if( hPullServer ) { RegCloseKey(hPullServer); hPullServer = NULL; }
if( hPullRoot ) { RegCloseKey(hPullRoot); hPullRoot = NULL; }
if( hServer ) { RegCloseKey(hServer); hServer = NULL; }
return Status;
ErrorReturn: DisplayErrorMessage(EMSG_SRVR_SHOW_PULLPARTNERPROPERTIES, Status); goto CommonReturn;
}
DWORDHandleSrvrShowPushpartnerproperties( IN LPCWSTR pwszMachine, IN OUT LPWSTR *ppwcArguments, IN DWORD dwCurrentIndex, IN DWORD dwArgCount, IN DWORD dwFlags, IN LPCVOID pvData, OUT BOOL *pbDone )/*++
Routine Description : Displays the properties for a particular Push partnerArguments : All aguments are passes as array of wide char strings in ppwcArguments. Compulsory : Push partner addressReturn Value: Returns the status of the operation.
--*/{ DWORD Status = NO_ERROR; DWORD i, j, dwNumArgs, dwTagCount; TAG_TYPE pttTags[] = { {WINS_TOKEN_SERVER, TRUE, FALSE}, }; PDWORD pdwTagType = NULL, pdwTagNum = NULL;
WCHAR wcServerIpAdd[MAX_IP_STRING_LEN+1] = {L'\0'}; HKEY hPushRoot = NULL, hPushServer = NULL, hServer = NULL; BOOL fPull = FALSE;
LPWSTR pTemp = NULL;
if( dwArgCount < dwCurrentIndex + 1 ) { DisplayMessage(g_hModule, HLP_SRVR_SHOW_PUSHPARTNERPROPERTIES_EX); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
dwNumArgs = dwArgCount - dwCurrentIndex; pdwTagType = WinsAllocateMemory(dwNumArgs*sizeof(DWORD)); pdwTagNum = WinsAllocateMemory(dwNumArgs*sizeof(DWORD));
if( pdwTagType is NULL or pdwTagNum is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
dwTagCount = sizeof(pttTags)/sizeof(TAG_TYPE);
Status = PreProcessCommand(ppwcArguments, dwArgCount, dwCurrentIndex, pttTags, &dwTagCount, pdwTagType, pdwTagNum);
if( Status isnot NO_ERROR ) goto ErrorReturn;
for( i=0; i<sizeof(pttTags)/sizeof(TAG_TYPE); i++ ) { if( pttTags[i].dwRequired is TRUE and pttTags[i].bPresent is FALSE ) { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } }
for( j=0; j<dwTagCount; j++ ) { switch(pdwTagType[j]) { case 0: { struct hostent * lpHostEnt = NULL; CHAR cAddr[16]; BYTE pbAdd[4]; char szAdd[4]; int k = 0, l=0; DWORD dwLen, nLen = 0; CHAR *pTemp = NULL; CHAR *pNetBios = NULL;
if( IsIpAddress( ppwcArguments[dwCurrentIndex+pdwTagNum[j]] ) ) { DWORD dwIp = inet_addr(WinsUnicodeToAnsi(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], NULL)); lpHostEnt = gethostbyaddr((char *)&dwIp, 4, AF_INET); if(lpHostEnt isnot NULL )//Valid IP Address
{ wcscpy(wcServerIpAdd, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]); break; } else { Status = ERROR_INVALID_IPADDRESS; goto ErrorReturn; } }
if( wcslen(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]) > 2 and _wcsnicmp(ppwcArguments[dwCurrentIndex+pdwTagNum[j]], L"\\\\", 2) is 0 ) k = 2; lpHostEnt = gethostbyname(WinsUnicodeToAnsi(ppwcArguments[dwCurrentIndex+pdwTagNum[j]]+k, NULL)); if( lpHostEnt is NULL ) { DisplayMessage(g_hModule, EMSG_WINS_INVALID_COMPUTER_NAME, ppwcArguments[dwCurrentIndex+pdwTagNum[j]]+k); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; }
memcpy(pbAdd, lpHostEnt->h_addr_list[0], 4); for( l=0;l<4; l++) { _itoa((int)pbAdd[l], szAdd, 10); memcpy(cAddr+nLen, szAdd, strlen(szAdd)); nLen += strlen(szAdd); *(cAddr+nLen) = '.'; nLen++;
} *(cAddr+nLen-1) = '\0'; { LPWSTR pstr = WinsAnsiToUnicode(cAddr, NULL);
if (pstr != NULL) { wcscpy(wcServerIpAdd, pstr); WinsFreeMemory(pstr); } else { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } } break; } default: { Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } } }
if( wcslen(g_ServerNetBiosName) > 0 ) { pTemp = g_ServerNetBiosName; }
Status = RegConnectRegistry(pTemp, HKEY_LOCAL_MACHINE, &hServer); if( Status isnot NO_ERROR ) goto ErrorReturn;
Status = RegOpenKeyEx(hServer, PUSHROOT, 0, KEY_READ, &hPushRoot);
if( Status isnot NO_ERROR ) { goto ErrorReturn; }
DisplayMessage(g_hModule, WINS_FORMAT_LINE);
//Now check if the desired server is in the list of push server.
Status = RegOpenKeyEx(hPushRoot, wcServerIpAdd, 0, KEY_READ, &hPushServer);
if( Status isnot NO_ERROR ) { DisplayMessage(g_hModule, EMSG_WINS_INVALID_PUSHPARTNER, wcServerIpAdd, g_ServerIpAddressUnicodeString); goto CommonReturn; }
//Check if it is also a pull partner or not.
{ DWORD dwKeyLen = 0; HKEY hPullServer = NULL; dwKeyLen = wcslen(PULLROOT)+ wcslen(L"\\") + wcslen(wcServerIpAdd); pTemp = WinsAllocateMemory((dwKeyLen+1)*sizeof(WCHAR)); if(pTemp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; } memset(pTemp, 0x00, (dwKeyLen+1)*sizeof(WCHAR)); wcscat(pTemp,PULLROOT); wcscat(pTemp, L"\\"); wcscat(pTemp, wcServerIpAdd);
Status = RegOpenKeyEx(hServer, pTemp, 0, KEY_READ, &hPullServer); if( Status isnot NO_ERROR ) { fPull = FALSE; } else { fPull = TRUE; } WinsFreeMemory(pTemp); pTemp = NULL; if( hPullServer ) { RegCloseKey(hPullServer); hPullServer = NULL; } }
//Now look for required parameters to display
{ WCHAR wcData[256] = {L'\0'}; DWORD dwData = 0; DWORD dwDatalen = 256; DWORD dwType = REG_SZ;
Status = RegQueryValueEx(hPushServer, L"NetBIOSName", 0, &dwType, (LPBYTE)wcData, &dwDatalen); if( Status isnot NO_ERROR ) { DisplayMessage(g_hModule, MSG_WINS_PUSHPARTNER_INFO, wcServerIpAdd, wszUnknown, (fPull is TRUE) ? wszPushpull : wszPush); } else { DisplayMessage(g_hModule, MSG_WINS_PUSHPARTNER_INFO, wcServerIpAdd, wcData, (fPull is TRUE) ? wszPushpull : wszPush); }
dwDatalen = sizeof(DWORD); dwType = REG_DWORD;
Status = RegQueryValueEx(hPushServer, PERSISTENCE, 0, &dwType, (LPBYTE)&dwData, &dwDatalen); if( Status is NO_ERROR ) { DisplayMessage(g_hModule, MSG_SRVR_PUSH_PERSISTENCE_STATE, dwData ? wszEnable : wszDisable); } else { DisplayMessage(g_hModule, MSG_SRVR_PUSH_PERSISTENCE_STATE, wszUnknown); }
dwDatalen = sizeof(DWORD); dwType = REG_DWORD;
Status = RegQueryValueEx(hPushServer, WINSCNF_UPDATE_COUNT_NM, 0, &dwType, (LPBYTE)&dwData, &dwDatalen);
if( Status is NO_ERROR ) { WCHAR Buffer[10] = {L'\0'}; _itow(dwData, Buffer, 10);
DisplayMessage(g_hModule, MSG_SRVR_PUSH_UPDATECOUNT, Buffer); } else { DisplayMessage(g_hModule, MSG_SRVR_PUSH_UPDATECOUNT, wszUnknown); } }
DisplayMessage(g_hModule, WINS_FORMAT_LINE);
CommonReturn: if( Status is NO_ERROR ) DisplayMessage(g_hModule, EMSG_WINS_ERROR_SUCCESS);
if( hPushServer ) { RegCloseKey(hPushServer); hPushServer = NULL; }
if( hPushRoot ) { RegCloseKey(hPushRoot); hPushRoot = NULL; }
if( hServer ) { RegCloseKey(hServer); hServer = NULL; } if( pdwTagType ) { WinsFreeMemory(pdwTagType); pdwTagType = NULL; } if( pdwTagNum ) { WinsFreeMemory(pdwTagNum); pdwTagNum = NULL; }
return Status;
ErrorReturn: DisplayErrorMessage(EMSG_SRVR_SHOW_PUSHPARTNERPROPERTIES, Status); goto CommonReturn;
}
DWORDGetVersionData( LPWSTR pwszVers, WINSINTF_VERS_NO_T *Version ){ LPWSTR pTemp = NULL; LPWSTR pwcBuffer = NULL; DWORD dwLen = 0; LPWSTR pwszToken=L",-."; if( ( pwszVers is NULL ) or ( IsBadStringPtr(pwszVers, MAX_STRING_LEN) is TRUE ) ) { return ERROR_INVALID_PARAMETER; }
dwLen = wcslen(pwszVers);
if( dwLen<2 ) { return ERROR_INVALID_PARAMETER; } pwcBuffer = WinsAllocateMemory((dwLen+1)*sizeof(WCHAR));
if( pwcBuffer is NULL ) { return ERROR_NOT_ENOUGH_MEMORY; }
wcscpy(pwcBuffer, pwszVers); if( pwcBuffer[0] isnot L'{' or pwcBuffer[dwLen-1] isnot L'}' ) { WinsFreeMemory(pwcBuffer); pwcBuffer = NULL; return ERROR_INVALID_PARAMETER; } pwcBuffer[dwLen-1] = L'\0'; pTemp = wcstok(pwcBuffer+1, pwszToken); if( pTemp is NULL ) { WinsFreeMemory(pwcBuffer); pwcBuffer = NULL; return ERROR_INVALID_PARAMETER; } Version->HighPart = wcstoul(pTemp, NULL, 16); pTemp = wcstok(NULL, pwszToken); if( pTemp is NULL ) { WinsFreeMemory(pwcBuffer); pwcBuffer = NULL; return ERROR_INVALID_PARAMETER; } Version->LowPart = wcstoul(pTemp, NULL, 16); WinsFreeMemory(pwcBuffer); pwcBuffer = NULL; return NO_ERROR;}
DWORDPreProcessCommand( IN LPWSTR *ppwcArguments, IN DWORD dwArgCount, IN DWORD dwCurrentIndex, IN OUT PTAG_TYPE pttTags, IN OUT PDWORD pdwTagCount, OUT PDWORD pdwTagType, OUT PDWORD pdwTagNum){ DWORD Status = NO_ERROR; DWORD i, j, dwTag = 0; LPWSTR pwszTemp = NULL;
if( pdwTagType is NULL or pdwTagNum is NULL or pttTags is NULL or pdwTagCount is NULL ) { return ERROR_INVALID_PARAMETER; }
dwTag = *pdwTagCount; i = 0; if( wcsstr(ppwcArguments[dwCurrentIndex], NETSH_ARG_DELIMITER) isnot NULL ) {
LPWSTR pwszTag = NULL; while( IsBadStringPtr(ppwcArguments[dwCurrentIndex+i], MAX_STRING_LEN) is FALSE ) { pwszTag = NULL; if( dwArgCount <= dwCurrentIndex + i ) break; if( wcslen(ppwcArguments[dwCurrentIndex+i]) is 0 ) break;
pwszTemp = WinsAllocateMemory((wcslen(ppwcArguments[dwCurrentIndex+i])+1)*sizeof(WCHAR));
if( pwszTemp is NULL ) { Status = ERROR_NOT_ENOUGH_MEMORY; goto ErrorReturn; }
wcscpy(pwszTemp, ppwcArguments[dwCurrentIndex+i]); if( wcsstr(ppwcArguments[dwCurrentIndex+i], NETSH_ARG_DELIMITER ) is NULL ) { DisplayMessage(g_hModule, EMSG_SRVR_NOT_TAGGED, i+1); i++; continue; } pwszTag = wcstok(pwszTemp, NETSH_ARG_DELIMITER); for( j=0; j<dwTag; j++ ) { if( FALSE is MatchToken(pwszTag, pttTags[j].pwszTag) ) { continue; } else { if( pttTags[j].bPresent is TRUE ) { Status = ERROR_INVALID_PARAMETER; WinsFreeMemory(pwszTemp); pwszTemp = NULL; goto ErrorReturn; } else { LPWSTR pwszVal = wcstok(NULL, NETSH_ARG_DELIMITER); if( pwszVal is NULL ) { wcscpy(ppwcArguments[dwCurrentIndex+i], L""); } else { wcscpy(ppwcArguments[dwCurrentIndex+i], pwszVal); } pttTags[j].bPresent = TRUE; pdwTagType[i] = j; pdwTagNum[i] = i; break; } } } if( pwszTemp ) { WinsFreeMemory(pwszTemp); pwszTemp = NULL; } i++; } } else { while( IsBadStringPtr(ppwcArguments[dwCurrentIndex+i], MAX_STRING_LEN) is FALSE ) { if( wcsstr(ppwcArguments[dwCurrentIndex+i], NETSH_ARG_DELIMITER) isnot NULL ) { DisplayMessage(g_hModule, EMSG_SRVR_NOT_UNTAGGED, i+1); Status = ERROR_INVALID_PARAMETER; goto ErrorReturn; } if( dwArgCount <= dwCurrentIndex + i ) break; // if( wcslen(ppwcArguments[dwCurrentIndex+i]) is 0 )
// break;
if( dwTag <= i ) { break; } pdwTagType[i] = i; pdwTagNum[i] = i; pttTags[i].bPresent = TRUE; i++; } } *pdwTagCount = i;CommonReturn: if( pwszTemp ) { WinsFreeMemory(pwszTemp); pwszTemp = NULL; } return Status;ErrorReturn: goto CommonReturn;}
DWORDGetStatus( BOOL fPrint, LPVOID pResultsA, BOOL fNew, BOOL fShort, LPCSTR pStartIp ){ DWORD Status, i; struct in_addr InAddr; PWINSINTF_RESULTS_T pResults = pResultsA; PWINSINTF_RESULTS_NEW_T pResultsN = pResultsA; PWINSINTF_ADD_VERS_MAP_T pAddVersMaps; DWORD NoOfOwners; WCHAR IpAddress[21] = {L'\0'}; WCHAR OwnerId[15] = {L'\0'}; WCHAR Buffer[5] = {L'\0'}; LPWSTR pwszDay = NULL; handle_t BindHdl; WINSINTF_BIND_DATA_T BindData = {0};
BindData.fTcpIp = TRUE; BindData.pServerAdd = (LPBYTE)WinsOemToUnicode(pStartIp, NULL);
if( BindData.pServerAdd is NULL ) { return ERROR_NOT_ENOUGH_MEMORY; } DisplayMessage(g_hModule, MSG_SRVR_MAPTABLE_HEADER, BindData.pServerAdd);
BindHdl = WinsBind(&BindData);
if( BindHdl is NULL ) { DisplayMessage(g_hModule, EMSG_WINS_BIND_FAILED, BindData.pServerAdd); return NO_ERROR; }
if (!fNew) { Status = WinsStatus(BindHdl, WINSINTF_E_CONFIG, pResultsA); }
else { pResultsN->pAddVersMaps = NULL; Status = WinsStatusNew(BindHdl, WINSINTF_E_CONFIG, pResultsN); } if( Status isnot NO_ERROR ) { WinsUnbind(&BindData, BindHdl); return Status; }
if( fShort is TRUE ) { DisplayMessage(g_hModule, MSG_WINS_NAMERECORD_SETTINGS); if( fNew ) { pwszDay = MakeDayTimeString(pResultsN->RefreshInterval); } else { pwszDay = MakeDayTimeString(pResults->RefreshInterval); }
DisplayMessage(g_hModule, MSG_WINS_NAMERECORD_REFRESHINTVL, pwszDay); WinsFreeMemory(pwszDay); pwszDay = NULL;
if( fNew ) { pwszDay = MakeDayTimeString(pResultsN->TombstoneInterval); } else { pwszDay = MakeDayTimeString(pResults->TombstoneInterval); }
DisplayMessage(g_hModule, MSG_WINS_NAMERECORD_TOMBSTONEINTVL, pwszDay); WinsFreeMemory(pwszDay); pwszDay = NULL;
if( fNew ) { pwszDay = MakeDayTimeString(pResultsN->TombstoneTimeout); } else { pwszDay = MakeDayTimeString(pResults->TombstoneTimeout); }
DisplayMessage(g_hModule, MSG_WINS_NAMERECORD_TOMBSTONETMOUT, pwszDay);
WinsFreeMemory(pwszDay); pwszDay = NULL;
if( fNew ) { pwszDay = MakeDayTimeString(pResultsN->VerifyInterval); } else { pwszDay = MakeDayTimeString(pResults->VerifyInterval); } DisplayMessage(g_hModule, MSG_WINS_NAMERECORD_VERIFYINTVL, pwszDay);
WinsFreeMemory(pwszDay); pwszDay = NULL;
DisplayMessage(g_hModule,WINS_FORMAT_LINE);
if (!fNew) { DisplayMessage(g_hModule, MSG_WINS_PRIORITY_CLASS, pResults->WinsPriorityClass == NORMAL_PRIORITY_CLASS ? wszNormal : wszHigh); DisplayMessage(g_hModule, MSG_WINS_WORKER_THREAD, pResults->NoOfWorkerThds);
pAddVersMaps = pResults->AddVersMaps; NoOfOwners = pResults->NoOfOwners; } else { DisplayMessage(g_hModule, MSG_WINS_PRIORITY_CLASS, pResultsN->WinsPriorityClass == NORMAL_PRIORITY_CLASS ? wszNormal : wszHigh); DisplayMessage(g_hModule, MSG_WINS_WORKER_THREAD, pResultsN->NoOfWorkerThds);
pAddVersMaps = pResultsN->pAddVersMaps; NoOfOwners = pResultsN->NoOfOwners;
}
DisplayMessage(g_hModule,WINS_FORMAT_LINE);
if (NoOfOwners isnot 0) { DisplayMessage(g_hModule, MSG_WINS_OWNER_TABLE);
for ( i= 0; i < NoOfOwners; i++, pAddVersMaps++) { DWORD j=0; for(j=0; j<20; j++) { IpAddress[j] = L' '; } IpAddress[20] = L'\0'; for(j=0; j<14; j++) { OwnerId[j] = L' '; } OwnerId[14] = L'\0';
_itow(i, Buffer, 10); wcscpy(OwnerId+4, Buffer); for(j=wcslen(OwnerId); j<14; j++) OwnerId[j] = L' '; OwnerId[14] = L'\0';
wcscpy(IpAddress+2, IpAddressToString(pAddVersMaps->Add.IPAdd)); for(j=wcslen(IpAddress); j<20; j++) IpAddress[j] = L' '; IpAddress[20] = L'\0';
if (fNew) { if( pAddVersMaps->VersNo.HighPart is MAXLONG and pAddVersMaps->VersNo.LowPart is MAXULONG ) {
DisplayMessage(g_hModule, MSG_WINS_OWNER_INFO_MAX, OwnerId, IpAddress, wszDeleted); continue; }
if (fShort && pAddVersMaps->VersNo.QuadPart == 0) { continue; }
DisplayMessage(g_hModule, MSG_WINS_OWNER_INFO, OwnerId, IpAddress, pAddVersMaps->VersNo.HighPart, pAddVersMaps->VersNo.LowPart);
}
} } else { DisplayMessage(g_hModule, MSG_WINS_NO_RECORDS); } } WinsUnbind(&BindData, BindHdl);
if( BindData.pServerAdd ) { WinsFreeMemory(BindData.pServerAdd); BindData.pServerAdd = NULL; } return(Status);}
VOIDChkAdd( PWINSINTF_RECORD_ACTION_T pRow, DWORD Add, BOOL fFile, FILE *pFile, DWORD OwnerIP, LPBOOL pfMatch ){
struct in_addr InAddr1, InAddr2; DWORD dwIpAddress = 0; LPWSTR pwszAdd1 = NULL; LPWSTR pwszAdd2 = NULL; LPWSTR pwszOwner = NULL;
BOOL fFirst = FALSE;
if ( pRow->TypOfRec_e is WINSINTF_E_UNIQUE or pRow->TypOfRec_e is WINSINTF_E_NORM_GROUP ) { pwszAdd2 = IpAddressToString(pRow->Add.IPAdd); pwszOwner = IpAddressToString(OwnerIP); if( pwszAdd2 is NULL or pwszOwner is NULL ) { DisplayMessage(g_hModule, EMSG_WINS_OUT_OF_MEMORY);
if( pwszAdd2 ) { WinsFreeMemory(pwszAdd2); pwszAdd2 = NULL; }
if( pwszOwner ) { WinsFreeMemory(pwszOwner); pwszOwner = NULL; } return; }
if (*pfMatch) { if (Add isnot pRow->Add.IPAdd) { WinsFreeMemory(pwszAdd2); pwszAdd2 = NULL; WinsFreeMemory(pwszOwner ); pwszOwner = NULL; *pfMatch = FALSE; return; } } DisplayMessage(g_hModule, MSG_WINS_IPADDRESS_STRING, pwszAdd2 );
if( fFile ) { DumpMessage(g_hModule, pFile, FMSG_WINS_IPADDRESS_STRING, pwszAdd2, pwszOwner); }
WinsFreeMemory(pwszAdd2); pwszAdd2 = NULL; WinsFreeMemory(pwszOwner ); pwszOwner = NULL;
return; } else //spec. grp or multihomed
{ DWORD ind; if (!*pfMatch) { DisplayMessage(g_hModule, MSG_WINS_MEMBER_COUNT, pRow->NoOfAdds/2); }
for ( ind=0; ind < pRow->NoOfAdds ; /*no third expr*/ ) { LPSTR psz1 = NULL; InAddr1.s_addr = htonl( (pRow->pAdd + ind++)->IPAdd); psz1 = inet_ntoa(InAddr1); if( psz1 is NULL ) { continue; }
pwszAdd1 = WinsOemToUnicode(psz1, NULL );
if( pwszAdd1 is NULL ) { DisplayMessage(g_hModule, EMSG_WINS_OUT_OF_MEMORY); return; }
if (!*pfMatch) { DisplayMessage(g_hModule, MSG_WINS_OWNER_ADDRESS, pwszAdd1); }
InAddr2.s_addr = htonl((pRow->pAdd + ind++)->IPAdd); psz1 = inet_ntoa(InAddr2); if( psz1 is NULL ) { if( pwszAdd1 ) { WinsFreeMemory(pwszAdd1); pwszAdd1 = NULL; } continue; }
pwszAdd2 = WinsOemToUnicode(psz1, NULL);
if( pwszAdd2 is NULL ) { DisplayMessage(g_hModule, EMSG_WINS_OUT_OF_MEMORY); if( pwszAdd1 ) { WinsFreeMemory(pwszAdd1); pwszAdd1 = NULL; } return; }
if (!*pfMatch) { DisplayMessage(g_hModule, MSG_WINS_IPADDRESS_STRING, pwszAdd2 ); if( fFile ) { if( fFirst is FALSE ) { DumpMessage(g_hModule, pFile, FMSG_WINS_IPADDRESS_STRING, pwszAdd2, pwszAdd1); fFirst = TRUE; } else { DumpMessage(g_hModule, pFile, FMSG_WINS_IPADDRESS_LIST, pwszAdd2, pwszAdd1); } } } if (*pfMatch) { if (Add isnot (pRow->pAdd + ind - 1)->IPAdd) { *pfMatch = FALSE;
if( pwszAdd1 ) { WinsFreeMemory(pwszAdd1); pwszAdd1 = NULL; } if( pwszAdd2 ) { WinsFreeMemory(pwszAdd2); pwszAdd2 = NULL; } return; } } }
//
// If there were no members to compare with, then
// let us set *pfMatch to FALSE.
//
if (ind == 0) { if (*pfMatch) { *pfMatch = FALSE; } } if( pwszAdd1 ) { WinsFreeMemory(pwszAdd1); pwszAdd1 = NULL; }
if( pwszAdd2 ) { WinsFreeMemory(pwszAdd2); pwszAdd2 = NULL; }
return; }}
DWORDGetDbRecs( WINSINTF_VERS_NO_T LowVersNo, WINSINTF_VERS_NO_T HighVersNo, PWINSINTF_ADD_T pWinsAdd, LPBYTE pTgtAdd, BOOL fSetFilter, LPBYTE pFilterName, DWORD Len, BOOL fAddFilter, DWORD AddFilter, BOOL fCountRec, BOOL fCase, BOOL fFile, LPWSTR pwszFile ){
WINSINTF_RECS_T Recs; DWORD Choice; DWORD Status = WINSINTF_SUCCESS; DWORD TotalCnt = 0; BOOL fMatch; WINSINTF_VERS_NO_T sTmpVersNo; WCHAR Name[20] = {L'\0'}; DWORD dwCount = 0; struct tm* time = NULL; LPWSTR pwszTime = NULL; int iType = 1; FILE *pFile = NULL; BOOL fOpenFile = FALSE; BOOL fHeader = FALSE; LPWSTR pwszGroup = NULL, pwszType = NULL, pwszState = NULL; WCHAR wszGroup[50] = {L'\0'}, wszState[50] = {L'\0'}, wszType[50] = {L'\0'};
DWORD dwGroup = WINS_GROUP_UNIQUE, dwGrouplen = 0, dwState = WINS_STATE_ACTIVE, dwStatelen = 0, dwType = WINS_TYPE_STATIC, dwTypelen = 0;
sTmpVersNo.LowPart = 1; sTmpVersNo.HighPart = 0;
if( fFile ) { pFile = _wfopen(pwszFile, L"a+"); if( pFile is NULL ) { fOpenFile = FALSE; DisplayMessage(g_hModule, EMSG_WINS_FILEOPEN_FAILED, pwszFile); } else { fOpenFile = TRUE; }
} while (TRUE) { LPWSTR pwszTempTgt = NULL;
Recs.pRow = NULL; Status = WinsGetDbRecs(g_hBind, pWinsAdd, LowVersNo, HighVersNo, &Recs);
if (fCountRec) { DisplayMessage(g_hModule, MSG_WINS_RECORDS_COUNT, Recs.TotalNoOfRecs);
break; } if (Status is WINSINTF_SUCCESS) { if (Recs.NoOfRecs > 0) { DWORD i, k; PWINSINTF_RECORD_ACTION_T pRow = Recs.pRow; TotalCnt += Recs.NoOfRecs;
if (!fSetFilter) { DisplayMessage(g_hModule, MSG_WINS_RECORDS_RETRIEVED, IpAddressToString(pWinsAdd->IPAdd), Recs.NoOfRecs); }
for (i=0; i<Recs.NoOfRecs; i++) {
if (fAddFilter) { //
// The address filter was specfied
// If the address matches, then
// fMatch will be TRUE after the
// function returns.
//
fMatch = TRUE; ChkAdd( pRow, AddFilter, fOpenFile, pFile, pWinsAdd->IPAdd, &fMatch ); } else { fMatch = FALSE; }
//
// If the address matched or if no filter
// was specified or if there was a name
// filter and the names matched, print
// out the details
//
if( fCase ) { if( fMatch or fSetFilter is FALSE or ( fAddFilter is FALSE and IsBadStringPtrA(pRow->pName, 256) is FALSE and strncmp(pRow->pName, pFilterName, (pRow->NameLen>Len)?Len:pRow->NameLen) is 0 ) ) { WCHAR Name[256] = {L'\0'}; CHAR chEndChar = 0x00; LPWSTR pwszTemp = NULL; DWORD dwTempLen = 0; DisplayMessage(g_hModule, MSG_WINS_RECORD_LINE); if( fOpenFile and g_fHeader is FALSE ) { DumpMessage(g_hModule, pFile, FMSG_WINS_RECORD_TABLE); g_fHeader = TRUE; } chEndChar = (CHAR)pRow->pName[15]; pRow->pName[15] = '\0';
pwszTemp = WinsOemToUnicode(pRow->pName, NULL);
if( pwszTemp is NULL ) { DisplayMessage(g_hModule, EMSG_WINS_OUT_OF_MEMORY); Status = WINSINTF_FAILURE; break; } dwTempLen = ( 16 > wcslen(pwszTemp) ) ? wcslen(pwszTemp) : 16;
wcsncpy(Name, pwszTemp, dwTempLen); WinsFreeMemory(pwszTemp); pwszTemp = NULL; for( k=dwTempLen; k<15; k++ ) { Name[k] = L' '; }
Name[15] = L'['; WinsHexToString(Name+16, (LPBYTE)&chEndChar, 1); Name[18] = L'h'; Name[19] = L']'; Name[20] = L'\0'; if( IsBadStringPtrA(pRow->pName+16, 240) is FALSE ) { pwszTemp = WinsOemToUnicode(pRow->pName+16, NULL); if( pwszTemp is NULL ) { DisplayMessage(g_hModule, EMSG_WINS_OUT_OF_MEMORY); Status = WINSINTF_FAILURE; break; } dwTempLen = ( 240 > wcslen(pwszTemp) ) ? wcslen(pwszTemp) : 240; wcsncpy(Name+20, pwszTemp, dwTempLen); WinsFreeMemory(pwszTemp); pwszTemp = NULL;
} Name[wcslen(Name)] = L'\0';
if( chEndChar is 0x1C ) { dwGroup = WINS_GROUP_DOMAIN; pwszGroup = L"DOMAIN NAME "; }
else if( pRow->TypOfRec_e is WINSINTF_E_UNIQUE ) { pwszGroup = L"UNIQUE "; dwGroup = WINS_GROUP_UNIQUE; } else if( pRow->TypOfRec_e is WINSINTF_E_NORM_GROUP ) { pwszGroup = L"GROUP "; dwGroup = WINS_GROUP_GROUP; } else if( pRow->TypOfRec_e is WINSINTF_E_SPEC_GROUP ) { pwszGroup = L"INTERNET GROUP"; dwGroup = WINS_GROUP_INTERNET; } else { pwszGroup = L"MULTIHOMED "; dwGroup = WINS_GROUP_MULTIHOMED; }
//Load the group string
{ dwGrouplen = LoadStringW(g_hModule, dwGroup, wszGroup, sizeof(wszGroup)/sizeof(WCHAR));
if( dwGrouplen is 0 ) wcscpy(wszGroup, pwszGroup); }
if( pRow->State_e is WINSINTF_E_ACTIVE ) { pwszState = L"ACTIVE"; dwState = WINS_STATE_ACTIVE; } else if( pRow->State_e is WINSINTF_E_RELEASED ) { dwState = WINS_STATE_RELEASED; pwszState = L"RELEASED"; } else { dwState = WINS_STATE_TOMBSTONE; pwszState = L"TOMBSTONE"; }
//Load the State string
{ dwStatelen = LoadStringW(g_hModule, dwState, wszState, sizeof(wszState)/sizeof(WCHAR));
if( dwStatelen is 0 ) wcscpy(wszState, pwszState); }
if( pRow->fStatic ) { dwType = WINS_TYPE_STATIC; pwszType = L"STATIC"; } else { dwType = WINS_TYPE_DYNAMIC; pwszType = L"DYNAMIC"; }
//Load the State string
{ dwTypelen = LoadStringW(g_hModule, dwType, wszType, sizeof(wszType)/sizeof(WCHAR));
if( dwTypelen is 0 ) wcscpy(wszType, pwszType); }
iType = 1; pwszTime = GetDateTimeString(pRow->TimeStamp, FALSE, &iType); DisplayMessage(g_hModule, MSG_WINS_RECORD_INFO, Name, wszType, wszState, pRow->VersNo.HighPart, pRow->VersNo.LowPart, pRow->NodeTyp, wszGroup, iType ? wszInfinite : pwszTime); if( fOpenFile ) { DumpMessage(g_hModule, pFile, FMSG_WINS_RECORDS_INFO, Name, wszType, wszState, pRow->VersNo.HighPart, pRow->VersNo.LowPart, pRow->NodeTyp, wszGroup, iType ? wszInfinite : pwszTime); }
if( pwszTime ) { WinsFreeMemory(pwszTime); pwszTime = NULL; } fMatch = FALSE;
ChkAdd( pRow, AddFilter, fOpenFile, pFile, pWinsAdd->IPAdd, &fMatch );
DisplayMessage(g_hModule, MSG_WINS_RECORD_LINE); dwCount++; } } else { if( fMatch or fSetFilter is FALSE or ( fAddFilter is FALSE and IsBadStringPtrA(pRow->pName, 256) is FALSE and _strnicmp(pRow->pName, pFilterName, (pRow->NameLen>Len)?Len:pRow->NameLen) is 0 ) ) { WCHAR Name[256] = {L'\0'}; CHAR chEndChar = 0x00; LPWSTR pwszTemp = NULL; DWORD dwTempLen = 0;
DisplayMessage(g_hModule, MSG_WINS_RECORD_LINE); if( fOpenFile and g_fHeader is FALSE ) { DumpMessage(g_hModule, pFile, FMSG_WINS_RECORD_TABLE); g_fHeader = TRUE; } chEndChar = (CHAR)pRow->pName[15]; pRow->pName[15] = '\0';
pwszTemp = WinsOemToUnicode(pRow->pName, NULL);
if( pwszTemp is NULL ) { DisplayMessage(g_hModule, EMSG_WINS_OUT_OF_MEMORY); Status = WINSINTF_FAILURE; break; } dwTempLen = ( 16 > wcslen(pwszTemp) ) ? wcslen(pwszTemp) : 16;
wcsncpy(Name, pwszTemp, dwTempLen); WinsFreeMemory(pwszTemp); pwszTemp = NULL; for( k=dwTempLen; k<15; k++ ) { Name[k] = L' '; }
Name[15] = L'['; WinsHexToString(Name+16, (LPBYTE)&chEndChar, 1); Name[18] = L'h'; Name[19] = L']'; Name[20] = L'\0'; if( IsBadStringPtrA(pRow->pName+16, 240) is FALSE ) { pwszTemp = WinsOemToUnicode(pRow->pName+16, NULL); if( pwszTemp is NULL ) { DisplayMessage(g_hModule, EMSG_WINS_OUT_OF_MEMORY); Status = WINSINTF_FAILURE; break; } dwTempLen = ( 240 > wcslen(pwszTemp) ) ? wcslen(pwszTemp) : 240; wcsncpy(Name+20, pwszTemp, dwTempLen); WinsFreeMemory(pwszTemp); pwszTemp = NULL;
} Name[wcslen(Name)] = L'\0';
if( chEndChar is 0x1C ) { dwGroup = WINS_GROUP_DOMAIN; pwszGroup = L"DOMAIN NAME "; } else if( pRow->TypOfRec_e is WINSINTF_E_UNIQUE ) { pwszGroup = L"UNIQUE "; dwGroup = WINS_GROUP_UNIQUE; } else if( pRow->TypOfRec_e is WINSINTF_E_NORM_GROUP ) { pwszGroup = L"GROUP "; dwGroup = WINS_GROUP_GROUP; } else if( pRow->TypOfRec_e is WINSINTF_E_SPEC_GROUP ) { pwszGroup = L"INTERNET GROUP"; dwGroup = WINS_GROUP_INTERNET; } else { pwszGroup = L"MULTIHOMED "; dwGroup = WINS_GROUP_MULTIHOMED; }
//Load the group string
{ dwGrouplen = LoadStringW(g_hModule, dwGroup, wszGroup, sizeof(wszGroup)/sizeof(WCHAR));
if( dwGrouplen is 0 ) wcscpy(wszGroup, pwszGroup); }
if( pRow->State_e is WINSINTF_E_ACTIVE ) { pwszState = L"ACTIVE"; dwState = WINS_STATE_ACTIVE; } else if( pRow->State_e is WINSINTF_E_RELEASED ) { dwState = WINS_STATE_RELEASED; pwszState = L"RELEASED"; } else { dwState = WINS_STATE_TOMBSTONE; pwszState = L"TOMBSTONE"; }
//Load the State string
{ dwStatelen = LoadStringW(g_hModule, dwState, wszState, sizeof(wszState)/sizeof(WCHAR));
if( dwStatelen is 0 ) wcscpy(wszState, pwszState); }
if( pRow->fStatic ) { dwType = WINS_TYPE_STATIC; pwszType = L"STATIC"; } else { dwType = WINS_TYPE_DYNAMIC; pwszType = L"DYNAMIC"; }
//Load the State string
{ dwTypelen = LoadStringW(g_hModule, dwType, wszType, sizeof(wszType)/sizeof(WCHAR));
if( dwTypelen is 0 ) wcscpy(wszType, pwszType); } iType = 1; pwszTime = GetDateTimeString(pRow->TimeStamp, FALSE, &iType);
DisplayMessage(g_hModule, MSG_WINS_RECORD_INFO, Name, wszType, wszState, pRow->VersNo.HighPart, pRow->VersNo.LowPart, pRow->NodeTyp, wszGroup, iType ? wszInfinite : pwszTime);
if( fOpenFile ) { DumpMessage(g_hModule, pFile, FMSG_WINS_RECORDS_INFO, Name, wszType, wszState, pRow->VersNo.HighPart, pRow->VersNo.LowPart, pRow->NodeTyp, wszGroup, iType ? wszInfinite : pwszTime); } if( pwszTime ) { WinsFreeMemory(pwszTime); pwszTime = NULL; } fMatch = FALSE;
ChkAdd( pRow, AddFilter, fOpenFile, pFile, pWinsAdd->IPAdd, &fMatch );
DisplayMessage(g_hModule, MSG_WINS_RECORD_LINE); dwCount++; }
} pRow++; }// end of for (all recs)
if (Status != WINSINTF_SUCCESS) break;
//
// If a range was chosen and records
// retrieved are == the limit of 100
// and if the Max vers no retrieved
// is less than that specified, ask
// user if he wishes to continue
//
if (!fSetFilter) { DisplayMessage(g_hModule, MSG_WINS_RECORDS_SEARCHED, Recs.NoOfRecs); } if ( Recs.NoOfRecs < Recs.TotalNoOfRecs and LiLtr((--pRow)->VersNo,HighVersNo) ) { LowVersNo.QuadPart = LiAdd(pRow->VersNo, sTmpVersNo); continue; }
DisplayMessage(g_hModule, MSG_WINS_SEARCHDB_COUNT, TotalCnt);
break; } pwszTempTgt = WinsOemToUnicode(pTgtAdd, NULL);
if( pwszTempTgt is NULL ) { DisplayMessage(g_hModule, EMSG_WINS_OUT_OF_MEMORY); Status = WINSINTF_FAILURE; break; }
DisplayMessage(g_hModule, MSG_WINS_NO_RECORD, pwszTempTgt); WinsFreeMemory(pwszTempTgt); pwszTempTgt = NULL;
} else { pwszTempTgt = WinsOemToUnicode(pTgtAdd, NULL);
if( pwszTempTgt is NULL ) { DisplayMessage(g_hModule, EMSG_WINS_OUT_OF_MEMORY); break; }
DisplayMessage(g_hModule, MSG_WINS_NO_RECORD, pwszTempTgt);
WinsFreeMemory(pwszTempTgt); pwszTempTgt = NULL;
} break;
} // while (TRUE)
DisplayMessage(g_hModule, MSG_SRVR_RECORD_MATCH, dwCount);
if( fOpenFile is TRUE ) { fclose(pFile); pFile = NULL; }
g_dwSearchCount += dwCount;
if (Recs.pRow != NULL) { WinsFreeMem(Recs.pRow); } return(Status);} // GetDbRecs
LPWSTRGetDateTimeString(DWORD_PTR TimeStamp, BOOL fShort, int *piType ){
DWORD Status = NO_ERROR, dwTime = 0; int iType = 1;
LPWSTR pwszTime = NULL; if( TimeStamp is INFINITE_EXPIRATION ) { iType = 1; } else { Status = FormatDateTimeString(TimeStamp, fShort, NULL, &dwTime);
if( Status is NO_ERROR ) { pwszTime = WinsAllocateMemory((dwTime+1)*sizeof(WCHAR));
if( pwszTime ) { dwTime++;
Status = FormatDateTimeString(TimeStamp, fShort, pwszTime, &dwTime);
} if( Status is NO_ERROR ) { iType = 0; } } }
*piType = iType; return pwszTime;}
|
