public void testConvertSid() {
String sidString = "S-1-1-0"; // Everyone
byte[] sidBytes = Advapi32Util.convertStringSidToSid(sidString);
assertTrue(sidBytes.length > 0);
String convertedSidString = Advapi32Util.convertSidToStringSid(new PSID(sidBytes));
assertEquals(convertedSidString, sidString);
}
public void testRegistryKeyExists() {
assertTrue(Advapi32Util.registryKeyExists(WinReg.HKEY_LOCAL_MACHINE, ""));
assertTrue(Advapi32Util.registryKeyExists(WinReg.HKEY_LOCAL_MACHINE, "Software\\Microsoft"));
assertFalse(
Advapi32Util.registryKeyExists(
WinReg.HKEY_LOCAL_MACHINE, "KeyDoesNotExist\\SubKeyDoesNotExist"));
}
public void testEncryptFile() throws Exception {
File file = createTempFile();
assertEquals(FILE_ENCRYPTABLE, Advapi32Util.fileEncryptionStatus(file));
Advapi32Util.encryptFile(file);
assertEquals(FILE_IS_ENCRYPTED, Advapi32Util.fileEncryptionStatus(file));
file.delete();
}
public void testSetFileSecurityDescriptor() throws Exception {
File file = createTempFile();
SECURITY_DESCRIPTOR_RELATIVE sdr = Advapi32Util.getFileSecurityDescriptor(file, false);
Advapi32Util.setFileSecurityDescriptor(file, sdr, false, true, true, false, true, false);
sdr = Advapi32Util.getFileSecurityDescriptor(file, false);
assertTrue(Advapi32.INSTANCE.IsValidSecurityDescriptor(sdr.getPointer()));
file.delete();
}
public void testGetAccountBySid() {
String accountName = Advapi32Util.getUserName();
Account currentUser = Advapi32Util.getAccountByName(accountName);
Account account = Advapi32Util.getAccountBySid(new PSID(currentUser.sid));
assertEquals(SID_NAME_USE.SidTypeUser, account.accountType);
assertEquals(currentUser.fqn.toLowerCase(), account.fqn.toLowerCase());
assertEquals(currentUser.name.toLowerCase(), account.name.toLowerCase());
assertEquals(currentUser.domain.toLowerCase(), account.domain.toLowerCase());
assertEquals(currentUser.sidString, account.sidString);
}
public void testRegistrySetGetIntValue() {
Advapi32Util.registryCreateKey(WinReg.HKEY_CURRENT_USER, "Software", "JNA");
Advapi32Util.registrySetIntValue(WinReg.HKEY_CURRENT_USER, "Software\\JNA", "IntValue", 42);
assertEquals(
42,
Advapi32Util.registryGetIntValue(WinReg.HKEY_CURRENT_USER, "Software\\JNA", "IntValue"));
assertTrue(
Advapi32Util.registryValueExists(WinReg.HKEY_CURRENT_USER, "Software\\JNA", "IntValue"));
Advapi32Util.registryDeleteKey(WinReg.HKEY_CURRENT_USER, "Software", "JNA");
}
public void testRegistryValueExists() {
assertFalse(
Advapi32Util.registryValueExists(WinReg.HKEY_LOCAL_MACHINE, "Software\\Microsoft", ""));
assertFalse(
Advapi32Util.registryValueExists(
WinReg.HKEY_LOCAL_MACHINE, "Software\\Microsoft", "KeyDoesNotExist"));
assertTrue(
Advapi32Util.registryValueExists(
WinReg.HKEY_LOCAL_MACHINE, "SYSTEM\\CurrentControlSet\\Control", "SystemBootDevice"));
}
public void testRegistryGetEmptyValues() {
HKEY root = WinReg.HKEY_CURRENT_USER;
String keyPath = "Software\\JNA";
Advapi32Util.registryCreateKey(root, "Software", "JNA");
doTestRegistryGetEmptyValues(root, keyPath, WinNT.REG_BINARY);
doTestRegistryGetEmptyValues(root, keyPath, WinNT.REG_EXPAND_SZ);
doTestRegistryGetEmptyValues(root, keyPath, WinNT.REG_MULTI_SZ);
doTestRegistryGetEmptyValues(root, keyPath, WinNT.REG_NONE);
doTestRegistryGetEmptyValues(root, keyPath, WinNT.REG_SZ);
Advapi32Util.registryDeleteKey(root, "Software", "JNA");
}
public void testIsWellKnownSid() {
String everyoneString = "S-1-1-0";
assertTrue(Advapi32Util.isWellKnownSid(everyoneString, WELL_KNOWN_SID_TYPE.WinWorldSid));
assertFalse(
Advapi32Util.isWellKnownSid(
everyoneString, WELL_KNOWN_SID_TYPE.WinAccountAdministratorSid));
byte[] everyoneBytes = Advapi32Util.convertStringSidToSid(everyoneString);
assertTrue(Advapi32Util.isWellKnownSid(everyoneBytes, WELL_KNOWN_SID_TYPE.WinWorldSid));
assertFalse(
Advapi32Util.isWellKnownSid(everyoneBytes, WELL_KNOWN_SID_TYPE.WinAccountAdministratorSid));
}
public void testRegistrySetGetStringValue() {
Advapi32Util.registryCreateKey(WinReg.HKEY_CURRENT_USER, "Software", "JNA");
Advapi32Util.registrySetStringValue(
WinReg.HKEY_CURRENT_USER, "Software\\JNA", "StringValue", "Hello World");
assertEquals(
"Hello World",
Advapi32Util.registryGetStringValue(
WinReg.HKEY_CURRENT_USER, "Software\\JNA", "StringValue"));
assertTrue(
Advapi32Util.registryValueExists(WinReg.HKEY_CURRENT_USER, "Software\\JNA", "StringValue"));
Advapi32Util.registryDeleteKey(WinReg.HKEY_CURRENT_USER, "Software", "JNA");
}
public void testRegistrySetGetExpandableStringValue() {
Advapi32Util.registryCreateKey(WinReg.HKEY_CURRENT_USER, "Software", "JNA");
Advapi32Util.registrySetExpandableStringValue(
WinReg.HKEY_CURRENT_USER, "Software\\JNA", "StringValue", "Temp is %TEMP%");
assertEquals(
"Temp is %TEMP%",
Advapi32Util.registryGetExpandableStringValue(
WinReg.HKEY_CURRENT_USER, "Software\\JNA", "StringValue"));
assertTrue(
Advapi32Util.registryValueExists(WinReg.HKEY_CURRENT_USER, "Software\\JNA", "StringValue"));
Advapi32Util.registryDeleteKey(WinReg.HKEY_CURRENT_USER, "Software", "JNA");
}
public void testDisableEncryption() throws Exception {
File dir = new File(System.getProperty("java.io.tmpdir") + File.separator + System.nanoTime());
dir.mkdir();
assertEquals(FILE_ENCRYPTABLE, Advapi32Util.fileEncryptionStatus(dir));
Advapi32Util.disableEncryption(dir, true);
assertEquals(FILE_DIR_DISALOWED, Advapi32Util.fileEncryptionStatus(dir));
Advapi32Util.disableEncryption(dir, false);
assertEquals(FILE_ENCRYPTABLE, Advapi32Util.fileEncryptionStatus(dir));
for (File file : dir.listFiles()) {
file.delete();
}
dir.delete();
}
public void testExecuteAccess() {
final boolean access =
Advapi32Util.accessCheck(
new File(System.getProperty("java.io.tmpdir")),
Advapi32Util.AccessCheckPermission.EXECUTE);
assertTrue(access);
}
public void testRegistrySetGetBinaryValue() {
byte[] data = {0x00, 0x01, 0x02};
Advapi32Util.registryCreateKey(WinReg.HKEY_CURRENT_USER, "Software", "JNA");
Advapi32Util.registrySetBinaryValue(
WinReg.HKEY_CURRENT_USER, "Software\\JNA", "BinaryValue", data);
byte[] read =
Advapi32Util.registryGetBinaryValue(
WinReg.HKEY_CURRENT_USER, "Software\\JNA", "BinaryValue");
assertEquals(data.length, read.length);
for (int i = 0; i < data.length; i++) {
assertEquals(data[i], read[i]);
}
assertTrue(
Advapi32Util.registryValueExists(WinReg.HKEY_CURRENT_USER, "Software\\JNA", "BinaryValue"));
Advapi32Util.registryDeleteKey(WinReg.HKEY_CURRENT_USER, "Software", "JNA");
}
public void testGetUserAccount() {
USER_INFO_1 userInfo = new USER_INFO_1();
userInfo.usri1_name = "JNANetapi32TestUser";
userInfo.usri1_password = "******";
userInfo.usri1_priv = LMAccess.USER_PRIV_USER;
// ignore test if not able to add user (need to be administrator to do this).
if (LMErr.NERR_Success != Netapi32.INSTANCE.NetUserAdd(null, 1, userInfo, null)) {
return;
}
try {
HANDLEByReference phUser = new HANDLEByReference();
try {
assertTrue(
Advapi32.INSTANCE.LogonUser(
userInfo.usri1_name.toString(),
null,
userInfo.usri1_password.toString(),
WinBase.LOGON32_LOGON_NETWORK,
WinBase.LOGON32_PROVIDER_DEFAULT,
phUser));
Advapi32Util.Account account = Advapi32Util.getTokenAccount(phUser.getValue());
assertTrue(account.name.length() > 0);
assertEquals(userInfo.usri1_name.toString(), account.name);
} finally {
if (phUser.getValue() != WinBase.INVALID_HANDLE_VALUE) {
Kernel32.INSTANCE.CloseHandle(phUser.getValue());
}
}
} finally {
assertEquals(
LMErr.NERR_Success, Netapi32.INSTANCE.NetUserDel(null, userInfo.usri1_name.toString()));
}
}
private void doTestRegistryGetEmptyValues(HKEY root, String keyPath, int valueType) {
String valueName = "EmptyValue";
registrySetEmptyValue(root, keyPath, valueName, valueType);
Map<String, Object> values = Advapi32Util.registryGetValues(root, keyPath);
assertEquals(1, values.size());
assertTrue(values.containsKey(valueName));
}
public void testGetCurrentUserGroups() {
Account[] groups = Advapi32Util.getCurrentUserGroups();
assertTrue(groups.length > 0);
for (Account group : groups) {
assertTrue(group.name.length() > 0);
assertTrue(group.sidString.length() > 0);
assertTrue(group.sid.length > 0);
}
}
public void testRegistryGetCloseKey() {
Advapi32Util.registryCreateKey(WinReg.HKEY_CURRENT_USER, "Software", "JNA");
Advapi32Util.registryCreateKey(WinReg.HKEY_CURRENT_USER, "Software\\JNA", "Key1");
Advapi32Util.registryCreateKey(WinReg.HKEY_CURRENT_USER, "Software\\JNA", "Key2");
HKEYByReference phkKey =
Advapi32Util.registryGetKey(WinReg.HKEY_CURRENT_USER, "Software\\JNA", WinNT.KEY_READ);
String[] subKeys = Advapi32Util.registryGetKeys(phkKey.getValue());
assertEquals(2, subKeys.length);
assertEquals(subKeys[0], "Key1");
assertEquals(subKeys[1], "Key2");
Advapi32Util.registryCloseKey(phkKey.getValue());
Advapi32Util.registryDeleteKey(WinReg.HKEY_CURRENT_USER, "Software\\JNA", "Key1");
Advapi32Util.registryDeleteKey(WinReg.HKEY_CURRENT_USER, "Software\\JNA", "Key2");
Advapi32Util.registryDeleteKey(WinReg.HKEY_CURRENT_USER, "Software", "JNA");
}
public static void main(String[] args) {
junit.textui.TestRunner.run(Advapi32UtilTest.class);
String currentUserName = Advapi32Util.getUserName();
System.out.println("GetUserName: "******" " + group.fqn + " [" + group.sidString + "]");
}
Account accountByName = Advapi32Util.getAccountByName(currentUserName);
System.out.println("AccountByName: " + currentUserName);
System.out.println(" Fqn: " + accountByName.fqn);
System.out.println(" Domain: " + accountByName.domain);
System.out.println(" Sid: " + accountByName.sidString);
Account accountBySid = Advapi32Util.getAccountBySid(new PSID(accountByName.sid));
System.out.println("AccountBySid: " + accountByName.sidString);
System.out.println(" Fqn: " + accountBySid.fqn);
System.out.println(" Name: " + accountBySid.name);
System.out.println(" Domain: " + accountBySid.domain);
}
public void testRegistryGetValues() {
String uu = new String("A" + "\\u00ea" + "\\u00f1" + "\\u00fc" + "C");
Advapi32Util.registryCreateKey(WinReg.HKEY_CURRENT_USER, "Software", "JNA");
Advapi32Util.registrySetIntValue(
WinReg.HKEY_CURRENT_USER, "Software\\JNA", "FourtyTwo" + uu, 42);
Advapi32Util.registrySetStringValue(
WinReg.HKEY_CURRENT_USER, "Software\\JNA", "42" + uu, "FourtyTwo" + uu);
Advapi32Util.registrySetExpandableStringValue(
WinReg.HKEY_CURRENT_USER, "Software\\JNA", "ExpandableString", "%TEMP%");
byte[] dataWritten = {0xD, 0xE, 0xA, 0xD, 0xB, 0xE, 0xE, 0xF};
Advapi32Util.registrySetBinaryValue(
WinReg.HKEY_CURRENT_USER, "Software\\JNA", "DeadBeef", dataWritten);
String[] stringsWritten = {"Hello", "World", "Hello World", uu};
Advapi32Util.registrySetStringArray(
WinReg.HKEY_CURRENT_USER, "Software\\JNA", "StringArray", stringsWritten);
String[] emptyArray = new String[0];
Advapi32Util.registrySetStringArray(
WinReg.HKEY_CURRENT_USER, "Software\\JNA", "EmptyStringArray", emptyArray);
Advapi32Util.registrySetBinaryValue(
WinReg.HKEY_CURRENT_USER, "Software\\JNA", "EmptyBinary", new byte[0]);
TreeMap<String, Object> values =
Advapi32Util.registryGetValues(WinReg.HKEY_CURRENT_USER, "Software\\JNA");
assertEquals(7, values.keySet().size());
assertEquals("FourtyTwo" + uu, values.get("42" + uu));
assertEquals(42, values.get("FourtyTwo" + uu));
assertEquals("%TEMP%", values.get("ExpandableString"));
byte[] dataRead = (byte[]) values.get("DeadBeef");
assertEquals(dataWritten.length, dataRead.length);
for (int i = 0; i < dataWritten.length; i++) {
assertEquals(dataWritten[i], dataRead[i]);
}
String[] stringsRead = (String[]) values.get("StringArray");
assertEquals(stringsWritten.length, stringsRead.length);
for (int i = 0; i < stringsWritten.length; i++) {
assertEquals(stringsWritten[i], stringsRead[i]);
}
stringsRead = (String[]) values.get("EmptyStringArray");
assertEquals(0, stringsRead.length);
Advapi32Util.registryDeleteKey(WinReg.HKEY_CURRENT_USER, "Software", "JNA");
}
public void testBackupEncryptedFile() throws Exception {
// backup an encrypted file
File srcFile = createTempFile();
Advapi32Util.encryptFile(srcFile);
File dest =
new File(
System.getProperty("java.io.tmpdir") + File.separator + "backup" + System.nanoTime());
dest.mkdir();
Advapi32Util.backupEncryptedFile(srcFile, dest);
// simple check to see if a backup file exist
File backupFile = new File(dest.getAbsolutePath() + File.separator + srcFile.getName());
assertTrue(backupFile.exists());
assertEquals(srcFile.length(), backupFile.length());
// backup an encrypted directory
File srcDir =
new File(System.getProperty("java.io.tmpdir") + File.separator + System.nanoTime());
srcDir.mkdir();
Advapi32Util.encryptFile(srcDir);
Advapi32Util.backupEncryptedFile(srcDir, dest);
// Check to see if a backup directory exist
File backupDir = new File(dest.getAbsolutePath() + File.separator + srcDir.getName());
assertTrue(backupDir.exists());
// clean up
srcFile.delete();
for (File file : srcDir.listFiles()) {
file.delete();
}
srcDir.delete();
for (File file : dest.listFiles()) {
file.delete();
}
dest.delete();
}
public void testRegistryCreateKeyDisposition() {
Advapi32Util.registryCreateKey(WinReg.HKEY_CURRENT_USER, "Software", "JNA");
Advapi32Util.registryDeleteKey(WinReg.HKEY_CURRENT_USER, "Software", "JNA");
assertTrue(Advapi32Util.registryCreateKey(WinReg.HKEY_CURRENT_USER, "Software", "JNA"));
assertFalse(Advapi32Util.registryCreateKey(WinReg.HKEY_CURRENT_USER, "Software", "JNA"));
assertTrue(Advapi32Util.registryKeyExists(WinReg.HKEY_CURRENT_USER, "Software\\JNA"));
Advapi32Util.registryDeleteKey(WinReg.HKEY_CURRENT_USER, "Software", "JNA");
}
public void testRegistryGetKeys() {
Advapi32Util.registryCreateKey(WinReg.HKEY_CURRENT_USER, "Software", "JNA");
Advapi32Util.registryCreateKey(WinReg.HKEY_CURRENT_USER, "Software\\JNA", "Key1");
Advapi32Util.registryCreateKey(WinReg.HKEY_CURRENT_USER, "Software\\JNA", "Key2");
String[] subKeys = Advapi32Util.registryGetKeys(WinReg.HKEY_CURRENT_USER, "Software\\JNA");
assertEquals(2, subKeys.length);
assertEquals(subKeys[0], "Key1");
assertEquals(subKeys[1], "Key2");
Advapi32Util.registryDeleteKey(WinReg.HKEY_CURRENT_USER, "Software\\JNA", "Key1");
Advapi32Util.registryDeleteKey(WinReg.HKEY_CURRENT_USER, "Software\\JNA", "Key2");
Advapi32Util.registryDeleteKey(WinReg.HKEY_CURRENT_USER, "Software", "JNA");
}
public void testGetEnvironmentBlock() {
String expected = "KEY=value\0" + "KEY_EMPTY=\0" + "KEY_NUMBER=2\0" + "\0";
// Order is important to kept checking result simple
Map<String, String> mockEnvironment = new TreeMap<String, String>();
mockEnvironment.put("KEY", "value");
mockEnvironment.put("KEY_EMPTY", "");
mockEnvironment.put("KEY_NUMBER", "2");
mockEnvironment.put("KEY_NULL", null);
String block = Advapi32Util.getEnvironmentBlock(mockEnvironment);
assertEquals(
"Environment block must comprise key=value pairs separated by NUL characters",
expected,
block);
}
public void testRegistrySetGetStringArray() {
Advapi32Util.registryCreateKey(WinReg.HKEY_CURRENT_USER, "Software", "JNA");
String[] dataWritten = {"Hello", "World"};
Advapi32Util.registrySetStringArray(
WinReg.HKEY_CURRENT_USER, "Software\\JNA", "MultiStringValue", dataWritten);
assertTrue(
Advapi32Util.registryValueExists(
WinReg.HKEY_CURRENT_USER, "Software\\JNA", "MultiStringValue"));
String[] dataRead =
Advapi32Util.registryGetStringArray(
WinReg.HKEY_CURRENT_USER, "Software\\JNA", "MultiStringValue");
assertEquals(dataWritten.length, dataRead.length);
for (int i = 0; i < dataRead.length; i++) {
assertEquals(dataWritten[i], dataRead[i]);
}
dataWritten = new String[0];
Advapi32Util.registrySetStringArray(
WinReg.HKEY_CURRENT_USER, "Software\\JNA", "EmptyMultiString", dataWritten);
dataRead =
Advapi32Util.registryGetStringArray(
WinReg.HKEY_CURRENT_USER, "Software\\JNA", "EmptyMultiString");
assertEquals(0, dataRead.length);
Advapi32Util.registryDeleteKey(WinReg.HKEY_CURRENT_USER, "Software", "JNA");
}
public void testGetAccountByName() {
String accountName = Advapi32Util.getUserName();
Account account = Advapi32Util.getAccountByName(accountName);
assertEquals(SID_NAME_USE.SidTypeUser, account.accountType);
}
public void testGetUsername() {
String username = Advapi32Util.getUserName();
assertTrue(username.length() > 0);
}
public void testGetAccountSidFromName() {
assertEquals("S-1-1-0", Advapi32Util.getAccountByName("Everyone").sidString);
}
public void testGetAccountNameFromSid() {
assertEquals("Everyone", Advapi32Util.getAccountBySid("S-1-1-0").name);
}
