123 lines
2.3 KiB
C
123 lines
2.3 KiB
C
|
///**************************************************************
|
||
|
|
/// Microsoft LAN Manager *
|
||
|
|
/// Copyright(c) Microsoft Corp., 1990 *
|
||
|
|
///**************************************************************
|
||
|
|
//
|
||
|
|
// This program is designed to do functional testing on the following
|
||
|
|
// APIs:
|
||
|
|
// NetUserModalsGet
|
||
|
|
// NetUserModalsSet
|
||
|
|
//
|
||
|
|
// This test can be run independently of other tests.
|
||
|
|
//
|
||
|
|
//
|
||
|
|
|
||
|
|
#include <nt.h> // TIME definition
|
||
|
|
#include <ntrtl.h> // TIME definition
|
||
|
|
#include <nturtl.h> // TIME definition
|
||
|
|
#define NOMINMAX // Avoid redefinition of min and max in stdlib.h
|
||
|
|
#include <windef.h>
|
||
|
|
#include <winbase.h>
|
||
|
|
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <string.h>
|
||
|
|
#include <lmcons.h>
|
||
|
|
#include <lmapibuf.h>
|
||
|
|
#include <netlib.h>
|
||
|
|
#include <netdebug.h>
|
||
|
|
#include <lmaccess.h>
|
||
|
|
#include <lmerr.h>
|
||
|
|
#include <ntsam.h>
|
||
|
|
|
||
|
|
#include "accessp.h"
|
||
|
|
#include "netlogon.h"
|
||
|
|
#include "logonp.h"
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// SetRole()
|
||
|
|
//
|
||
|
|
|
||
|
|
void
|
||
|
|
SetRole(
|
||
|
|
DWORD Role
|
||
|
|
)
|
||
|
|
{
|
||
|
|
DWORD err;
|
||
|
|
PUSER_MODALS_INFO_1 um1p;
|
||
|
|
USER_MODALS_INFO_1006 um1006;
|
||
|
|
|
||
|
|
//
|
||
|
|
// setup data for update
|
||
|
|
//
|
||
|
|
um1006.usrmod1006_role = Role;
|
||
|
|
|
||
|
|
if (err = NetUserModalsSet(NULL, 1006, (LPBYTE)&um1006, NULL)) {
|
||
|
|
|
||
|
|
printf("NetUserModalsSet failed %d \n", err);
|
||
|
|
return;
|
||
|
|
|
||
|
|
} else {
|
||
|
|
|
||
|
|
//
|
||
|
|
// verify set data
|
||
|
|
//
|
||
|
|
|
||
|
|
if (err = NetUserModalsGet(NULL, 1, (LPBYTE *) &um1p)) {
|
||
|
|
|
||
|
|
printf("NetUserModalsGet failed %d \n", err);
|
||
|
|
return;
|
||
|
|
|
||
|
|
} else {
|
||
|
|
|
||
|
|
//
|
||
|
|
// verify initial settings
|
||
|
|
//
|
||
|
|
|
||
|
|
if( um1p->usrmod1_role != Role ) {
|
||
|
|
printf("Verify ROLE failed \n");
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
printf("SamRole set successfully");
|
||
|
|
}
|
||
|
|
|
||
|
|
NetApiBufferFree( um1p );
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
void __cdecl
|
||
|
|
main(argc, argv)
|
||
|
|
int argc;
|
||
|
|
char **argv;
|
||
|
|
{
|
||
|
|
DWORD Role;
|
||
|
|
|
||
|
|
if( argc < 2 ) {
|
||
|
|
printf("Usage : SamRole [ Primary | Backup ] \n" );
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
if(_stricmp( argv[1], "Primary" ) == 0) {
|
||
|
|
|
||
|
|
Role = UAS_ROLE_PRIMARY;
|
||
|
|
|
||
|
|
} else if( _stricmp(argv[1], "Backup") == 0) {
|
||
|
|
|
||
|
|
Role = UAS_ROLE_BACKUP;
|
||
|
|
|
||
|
|
} else {
|
||
|
|
|
||
|
|
printf("Usage : SamRole [ Primary | Backup ] \n" );
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
SetRole(Role);
|
||
|
|
|
||
|
|
}
|
||
|
|
|