com.scudata.common.ODBCUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of esproc Show documentation
SPL(Structured Process Language) A programming language specially for structured data computing.
There is a newer version: 20241126
package com.scudata.common;

import java.lang.reflect.Method;
import java.util.ArrayList;

/**
 * ODBC??????
 *
 */
public final class ODBCUtil {
	/**
	 * ????Դ????????
	 */
	/** ?û?????Դ???? */
	public final static int USER_DSN = 0x00000001;
	/** ϵͳ????Դ???? */
	public final static int SYS_DSN = 0x00000002;

	/**
	 * ODBCע??·??
	 */
	private final static String Rel_Reg_Path = "Software\\ODBC\\ODBC.INI\\ODBC Data Sources";

	/**
	 * ????????Դ?????б?
	 * 
	 * @param type
	 *            ָ??ȡ?û?????Դ???ƻ???ϵͳ????Դ???ƣ?????ȫ??(USER_DSN|SYS_DSN).
	 * @return ????Դ?????б?
	 */
	public static ArrayList getDataSourcesName(int type) {
		ArrayList array = new ArrayList();

		if ((type & USER_DSN) == USER_DSN) {
			int[] ret = WinRegisterUtil.windowsRegOpenKeyEx(
					WinRegisterUtil.HKEY_CURRENT_USER, Rel_Reg_Path,
					WinRegisterUtil.KEY_READ);
			if (ret[WinRegisterUtil.ERROR_CODE] == WinRegisterUtil.ERROR_SUCCESS) {
				enumValue(ret[WinRegisterUtil.NATIVE_HANDLE], array);
				WinRegisterUtil
						.windowsRegCloseKey(ret[WinRegisterUtil.NATIVE_HANDLE]);
			}
		}

		if ((type & SYS_DSN) == SYS_DSN) {
			int[] ret = WinRegisterUtil.windowsRegOpenKeyEx(
					WinRegisterUtil.HKEY_LOCAL_MACHINE, Rel_Reg_Path,
					WinRegisterUtil.KEY_READ);
			if (ret[WinRegisterUtil.ERROR_CODE] == WinRegisterUtil.ERROR_SUCCESS) {
				enumValue(ret[WinRegisterUtil.NATIVE_HANDLE], array);
				WinRegisterUtil
						.windowsRegCloseKey(ret[WinRegisterUtil.NATIVE_HANDLE]);
			}
		}
		return array;
	}

	/**
	 * ö??hKey?µ?ֵ
	 * 
	 * @param hKey
	 *            ??
	 * @param array
	 *            ????
	 */
	private static void enumValue(int hKey, ArrayList array) {
		int i = 0;
		while (true) {
			String retValue = WinRegisterUtil.windowsRegEnumValue(hKey, i++);
			if (retValue == null) {
				break;
			}
			array.add(retValue);
		}
	}
}

/**
 * Windowsע?Ṥ??
 *
 */
final class WinRegisterUtil {
	/**
	 * root key value defined by windows
	 */
	public static final int HKEY_CLASSES_ROOT = 0x80000000;
	public static final int HKEY_CURRENT_USER = 0x80000001;
	public static final int HKEY_LOCAL_MACHINE = 0x80000002;
	public static final int HKEY_USERS = 0x80000003;
	public static final int HKEY_PERFORMANCE_DATA = 0x80000004;
	public static final int HKEY_CURRENT_CONFIG = 0x80000005;

	/**
	 * Constants used to interpret returns of native functions
	 */
	public static final int NATIVE_HANDLE = 0;
	public static final int ERROR_CODE = 1;

	/**
	 * Windows security masks
	 * 
	 */
	public static final int DELETE = 0x10000;
	public static final int KEY_QUERY_VALUE = 1;
	public static final int KEY_SET_VALUE = 2;
	public static final int KEY_CREATE_SUB_KEY = 4;
	public static final int KEY_ENUMERATE_SUB_KEYS = 8;
	public static final int KEY_READ = 0x20019;
	public static final int KEY_WRITE = 0x20006;
	public static final int KEY_ALL_ACCESS = 0xf003f;

	/**
	 * error codes
	 */
	public static final int ERROR_SUCCESS = 0;
	public static final int ERROR_FAILED = -1;

	/**
	 * ??????󳤶?
	 */
	private static final Integer MAX_KEY_LENGHT = new Integer(100);
	/**
	 * ?????
	 */
	private static final Class theClass = getUtilClass();

	/**
	 * ????ȡ?????????
	 * 
	 * @return
	 */
	private static Class getUtilClass() {
		try {
			return Class.forName("java.util.prefs.WindowsPreferences");
		} catch (Exception e) {
			return null;
		}
	}

	/**
	 * ????ִ??WindowsRegOpenKey
	 * 
	 * @param hKey
	 *            ????????????ͷ??HKEY_CURRENT_USER
	 * @param subKey
	 *            ???hKey??·???????磺"Software\\ODBC"
	 * @param securityMask
	 *            ??дȨ?ޣ???????ͷ??KEY_ALL_ACCESS
	 * @return
	 */
	public static int[] windowsRegOpenKeyEx(int hKey, String subKey,
			int securityMask) {
		try {
			Method m = theClass.getDeclaredMethod("WindowsRegOpenKey",
					new Class[] { int.class, byte[].class, int.class });
			m.setAccessible(true);

			Object ret = m.invoke(null, new Object[] { new Integer(hKey),
					stringToByteArray(subKey), new Integer(securityMask) });
			return (int[]) ret;
		} catch (Exception e) {
			System.out.println("Exception windowsRegOpenKeyEx!");
			return new int[] { 0, ERROR_FAILED };
		}
	}

	/**
	 * ?ر?ע???
	 * 
	 * @param hKey
	 *            ??
	 */
	public static void windowsRegCloseKey(int hKey) {
		try {
			Method m = theClass.getDeclaredMethod("WindowsRegCloseKey",
					new Class[] { int.class });
			m.setAccessible(true);

			m.invoke(null, new Object[] { new Integer(hKey) });
		} catch (Exception e) {
			System.out.println("Exception windowsRegCloseKey!");
		}
	}

	/**
	 * ????ִ??WindowsRegQueryValueEx
	 * 
	 * @param hKey
	 *            ?????ľ??
	 * @param key
	 *            ?Ӽ????ַ???????????ʱȡ????
	 * @return
	 */
	public static String windowsRegQueryValueEx(int hKey, String key) {
		try {
			Method m = theClass.getDeclaredMethod("WindowsRegQueryValueEx",
					new Class[] { int.class, byte[].class });
			m.setAccessible(true);

			Object value = m.invoke(null, new Object[] { new Integer(hKey),
					stringToByteArray(key) });
			if (value == null) {
				return null;
			}

			return winByteArrayToString((byte[]) value);
		} catch (Exception e) {
			System.out.println("Exception windowsRegQueryValueEx!");
			return null;
		}
	}

	/**
	 * ö??hKey?µ?ֵ
	 * 
	 * @param hKey
	 * @param valueIndex
	 *            ֵ??ţ???0??ʼ????
	 * @return
	 */
	public static String windowsRegEnumValue(int hKey, int valueIndex) {
		try {
			Method m = theClass.getDeclaredMethod("WindowsRegEnumValue",
					new Class[] { int.class, int.class, int.class });
			m.setAccessible(true);

			Object value = m.invoke(null, new Object[] { new Integer(hKey),
					new Integer(valueIndex), MAX_KEY_LENGHT });

			return winByteArrayToString((byte[]) value);
		} catch (Exception e) {
			System.out.println("Exception windowsRegEnumValue!");
			return null;
		}
	}

	/**
	 * ?ַ???תbyte????
	 * 
	 * @param str
	 * @return
	 */
	private static byte[] stringToByteArray(String str) {
		byte[] result = new byte[str.length() + 1];
		for (int i = 0; i < str.length(); i++) {
			result[i] = (byte) str.charAt(i);
		}
		result[str.length()] = 0;
		return result;
	}

	/**
	 * byte????ת?ַ???
	 * 
	 * @param array
	 * @return
	 */
	private final static String winByteArrayToString(byte[] array) {
		if (array == null) {
			return null;
		}

		int pos = array.length - 1;
		while (pos >= 0) {
			if (array[pos] != '\0') {
				break;
			}
			pos--;
		}

		if (pos >= 0) {
			return new String(array, 0, pos + 1);
		} else {
			return null;
		}
	}
}