plugins/org.eclipse.actf.model.ui.editors.ie/src/org/eclipse/actf/model/internal/ui/editors/ie/win32/RegistryUtil.java

/*******************************************************************************
 * Copyright (c) 2007 IBM Corporation and Others
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *    Takashi ITOH - initial API and implementation
 *******************************************************************************/

package org.eclipse.actf.model.internal.ui.editors.ie.win32;

import org.eclipse.swt.internal.win32.OS;
import org.eclipse.swt.internal.win32.TCHAR;

public class RegistryUtil {

	public static final int REG_NONE = 0, REG_SZ = 1, REG_EXPAND_SZ = 2,
			REG_BINARY = 3, REG_DWORD = 4, REG_DWORD_BIG_ENDIAN = 5,
			REG_LINK = 6, REG_MULTI_SZ = 7, REG_RESOURCE_LIST = 8,
			REG_FULL_RESOURCE_DESCRIPTOR = 9,
			REG_RESOURCE_REQUIREMENTS_LIST = 10, REG_QWORD = 11;

	public static final int KEY_READ = 0x20019;

	public static final String IE_SETTINGS_KEY = "Software\\Microsoft\\Internet Explorer\\Settings", //$NON-NLS-1$
			IE_ANCHOR_COLOR = "Anchor Color", //$NON-NLS-1$
			IE_ANCHOR_COLOR_VISITED = "Anchor Color Visited"; //$NON-NLS-1$

	public static int getIERegistryInt(String valueString) {
		return getRegistryInt(OS.HKEY_CURRENT_USER,
				RegistryUtil.IE_SETTINGS_KEY, valueString);
	}

	public static String getIERegistryString(String valueString) {
		return getRegistryString(OS.HKEY_CURRENT_USER,
				RegistryUtil.IE_SETTINGS_KEY, valueString);
	}

	public static int getRegistryInt(int hKeyParent, String subKeyString,
			String valueString) {
		int hKey = open(hKeyParent, subKeyString, 0, KEY_READ);
		if (0 != hKey) {
			try {
				return getIntegerValue(hKey, valueString);
			} finally {
				close(hKey);
			}
		}
		return 0;
	}

	public static String getRegistryString(int hKeyParent, String subKeyString,
			String valueString) {
		int hKey = open(hKeyParent, subKeyString, 0, KEY_READ);
		if (0 != hKey) {
			try {
				return getStringValue(hKey, valueString);
			} finally {
				close(hKey);
			}
		}
		return null;
	}

	private static int open(int hKeyParent, String subKeyString, int ulOptions,
			int samDesired) {
		TCHAR keyName = new TCHAR(0, subKeyString, true);
		int[] phKey = new int[1];
		if (0 == OS.RegOpenKeyEx(hKeyParent, keyName, ulOptions, samDesired,
				phKey)) {
			return phKey[0];
		}
		return 0;
	}

	private static int close(int hKey) {
		return OS.RegCloseKey(hKey);
	}

	private static int getIntegerValue(int hKey, String valueString) {
		TCHAR valueName = new TCHAR(0, valueString, true);
		int[] pcbData = new int[] { 4 };
		int[] pType = new int[1];
		int[] pData = new int[1];
		if (0 == OS.RegQueryValueEx(hKey, valueName, 0, pType, pData, pcbData)) {
			if (REG_DWORD == pType[0]) {
				return pData[0];
			}
		}
		return 0;
	}

	private static String getStringValue(int hKey, String valueString) {
		TCHAR valueName = new TCHAR(0, valueString, true);
		int[] pcbData = new int[1];
		int[] pType = new int[1];
		if (0 == OS.RegQueryValueEx(hKey, valueName, 0, pType, (TCHAR) null,
				pcbData)) {
			if (REG_SZ == pType[0]) {
				TCHAR buff = new TCHAR(0, pcbData[0] / TCHAR.sizeof);
				if (0 == OS.RegQueryValueEx(hKey, valueName, 0, pType, buff,
						pcbData)) {
					return buff.toString(0, buff.strlen());
				}
			}
		}
		return null;
	}

}