[개발] Java로 Proxy 변경 Tool 만들기

2016. 6. 3. 09:39

필자의 회사의 경우 보안을 위해 외부 인터넷 망과 연결시 Proxy를 통해야 한다.

그런데, 보안점검을 하다보면 Proxy를 Burpsuite나 Fiddler와 같은 Tool의 포트로 변경해줘야 하는 일이 있는데,

이걸 매번 브라우저의 설정탭을 찾아 들어가 변경하기도 귀찮고,

그렇다고 Cooxie Toolbar같은걸 번거롭게 설치해서 마우스 클릭질 하는 것도 맘에 들지 않아서, Java로 변경해주는 걸 하나 만들어 보았다.

이 방법은 Java를 이용해 Registery 값을 수정하는 방법인데, Chrome과 IE는 적용되지만 FireFox는 따로 Proxy 정보를 저장하고 있어 적용되지 않는다.

소스를 보면, 간단하다.

Proxy.java

package oneclickproxy;

import java.awt.event.WindowAdapter;
import java.awt.event.WindowEvent;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.InputStreamReader;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;

import javax.swing.JDialog;
import javax.swing.JOptionPane;

import ca.beq.util.win32.registry.RegistryKey;
import ca.beq.util.win32.registry.RegistryValue;
import ca.beq.util.win32.registry.RootKey;
import ca.beq.util.win32.registry.ValueType;
import oneclickproxy.WinInet.INTERNET_PER_CONN_OPTION_LIST;
public class proxy {

private static final String REGEDIT_KEY = "Software\\Microsoft\\Windows\\CurrentVersion\\Internet Settings";
	private static final String PROXY_VALUE = "ProxyEnable";
	private static final String AUTOCONFIG_VALUE = "AutoConfigURL";
	private static final String PROXYOVERRIDE_VALUE = "ProxyOverride";
	private static final String PROXYSERVER_VALUE = "ProxyServer";
	private static final String REGISTRY_LIBRARY_PATH = "\\lib\\jRegistryKey.dll";
	private static final String CONFIG_FILEPATH = "\\dat\\config.dat";
	
	static INTERNET_PER_CONN_OPTION_LIST list = new INTERNET_PER_CONN_OPTION_LIST();
	
	public static void main(String[] args) {
		// Load JNI library
//		System.out.println(System.getProperty("java.library.path"));
//		System.out.println(new File(".").getAbsolutePath()+REGISTRY_LIBRARY_PATH);
		RegistryKey.initialize(new File(".").getAbsolutePath()
				+ REGISTRY_LIBRARY_PATH);

proxy proxy = new proxy();
		int before = proxy.getProxyEnable(); // 0 :: enable 1 :: able
		int after = before;
		System.out.println("before :: " + before);
		do {
			proxy.setconfigurl(before);
			after = proxy.setProxyEnable(before);
			System.out.println("mid :: " + after);
		} while (before == after);
		System.out.println("after :: " + after);

JOptionPane op = new JOptionPane(after == 0 ? "���Ͻü�����\n��Ȱ��ȭ�Ǿ����ϴ�."
				: "���Ͻü�����\nȰ��ȭ�Ǿ����ϴ�.", after == 0 ? JOptionPane.ERROR_MESSAGE
				: JOptionPane.INFORMATION_MESSAGE);
		WinInet.INSTANCE.InternetSetOption(null,
				WinInet.INTERNET_OPTION_PER_CONNECTION_OPTION, null, 0);
		WinInet.INSTANCE.InternetSetOption(null,
				WinInet.INTERNET_OPTION_SETTINGS_CHANGED, null, 0);
		WinInet.INSTANCE.InternetSetOption(null,
				WinInet.INTERNET_OPTION_REFRESH, null, 0);
		RegistryKey.initialize(new File(".").getAbsolutePath()
				+ REGISTRY_LIBRARY_PATH);
		final JDialog dialog = op.createDialog("OCP 1.4 made by Y.S.H");
		dialog.setAlwaysOnTop(true); // <-- this line

ScheduledExecutorService sch = Executors.newSingleThreadScheduledExecutor();     
		sch.schedule(new Runnable() {
		    public void run() {
		    	dialog.setVisible(false);
		    	dialog.dispose();
		    }
		}, Integer.parseInt(proxy.get_specificline(3,
				new File(".").getAbsolutePath() + CONFIG_FILEPATH)), TimeUnit.MILLISECONDS);
		
		dialog.setVisible(true);
		dialog.setDefaultCloseOperation(JDialog.DISPOSE_ON_CLOSE);
		dialog.addWindowListener(new WindowAdapter() { 
		    @Override public void windowClosed(WindowEvent e) { 
		      System.exit(0);
		    }
		  });
	    System.exit(0);
	}

private void setconfigurl(int a) {
		// TODO Auto-generated method stub
		RegistryKey key = new RegistryKey(RootKey.HKEY_CURRENT_USER,
				REGEDIT_KEY);
		if (a == 0) {
			if (key.hasValue(AUTOCONFIG_VALUE))
				key.deleteValue(AUTOCONFIG_VALUE);
		} else {
			System.out.println(get_specificline(0,
					new File(".").getAbsolutePath() + CONFIG_FILEPATH));
			RegistryValue value = new RegistryValue(AUTOCONFIG_VALUE,
					ValueType.REG_SZ, get_specificline(0,
							new File(".").getAbsolutePath() + CONFIG_FILEPATH));
			key.setValue(value);
		}
		RegistryValue server_value = new RegistryValue();
		server_value.setName(PROXYSERVER_VALUE);
		server_value.setType(ValueType.REG_SZ);
		server_value.setData(get_specificline(1,
				new File(".").getAbsolutePath() + CONFIG_FILEPATH));
		if (key.hasValue(PROXYSERVER_VALUE)) {
			key.setValue(server_value);
		}
		RegistryValue override_value = new RegistryValue();
		override_value.setName(PROXYOVERRIDE_VALUE);
		override_value.setType(ValueType.REG_SZ);
		override_value.setData(get_specificline(2,
				new File(".").getAbsolutePath() + CONFIG_FILEPATH));

if (key.hasValue(PROXYOVERRIDE_VALUE)) {
			key.setValue(override_value);
		}
	}

private int getProxyEnable() {
		RegistryKey key = new RegistryKey(RootKey.HKEY_CURRENT_USER,
				REGEDIT_KEY);
		RegistryValue value = key.getValue(PROXY_VALUE);

if (key.hasValue(PROXY_VALUE)) {
			return Integer.parseInt(value.getData().toString());
		}
		return 0;
	}

private int setProxyEnable(int enable) {
		RegistryKey key = new RegistryKey(RootKey.HKEY_CURRENT_USER,
				REGEDIT_KEY);
		RegistryValue value = new RegistryValue();
		value.setName(PROXY_VALUE);
		value.setType(ValueType.REG_DWORD_BIG_ENDIAN);
		int after = enable == 0 ? 1 : 0;
		value.setData(after);

if (key.hasValue(PROXY_VALUE)) {
			key.setValue(value);
		}
		return after;
	}

public String get_specificline(int lineNumber, String filename) {
		try {
			FileInputStream fs = new FileInputStream(filename);
			BufferedReader br = new BufferedReader(new InputStreamReader(fs));
			for (int i = 0; i < lineNumber; ++i) {
				br.readLine();
			}
			return br.readLine();
		} catch (Exception e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
		return "FAIL";
	}
}

WinInet.java

package oneclickproxy;

import java.util.Arrays;
import java.util.List;

import com.sun.jna.Native;
import com.sun.jna.Pointer;
import com.sun.jna.Structure;
import com.sun.jna.Union;
import com.sun.jna.win32.StdCallLibrary;
import com.sun.jna.win32.W32APIOptions;

public interface WinInet extends StdCallLibrary {
	WinInet INSTANCE = (WinInet) Native.loadLibrary("wininet", WinInet.class,
			W32APIOptions.UNICODE_OPTIONS);

public static final int INTERNET_PER_CONN_FLAGS = 1;
	public static final int INTERNET_PER_CONN_PROXY_SERVER = 2;
	public static final int INTERNET_PER_CONN_PROXY_BYPASS = 3;
	public static final int INTERNET_PER_CONN_AUTOCONFIG_URL = 4;
	public static final int INTERNET_PER_CONN_AUTODISCOVERY_FLAGS = 5;
	public static final int INTERNET_PER_CONN_AUTOCONFIG_SECONDARY_URL = 6;
	public static final int INTERNET_PER_CONN_AUTOCONFIG_RELOAD_DELAY_MINS = 7;
	public static final int INTERNET_PER_CONN_AUTOCONFIG_LAST_DETECT_TIME = 8;
	public static final int INTERNET_PER_CONN_AUTOCONFIG_LAST_DETECT_URL = 9;

public static final int PROXY_TYPE_DIRECT = 0x00000001; // direct to net
	public static final int PROXY_TYPE_PROXY = 0x00000002; // via named proxy
	public static final int PROXY_TYPE_AUTO_PROXY_URL = 0x00000004; // autoproxy
																	// URL
	public static final int PROXY_TYPE_AUTO_DETECT = 0x00000008; // use
																	// autoproxy
																	// detection

//
	// options manifests for Internet{Query|Set}Option
	//

public static final int INTERNET_OPTION_CALLBACK = 1;
	public static final int INTERNET_OPTION_CONNECT_TIMEOUT = 2;
	public static final int INTERNET_OPTION_CONNECT_RETRIES = 3;
	public static final int INTERNET_OPTION_CONNECT_BACKOFF = 4;
	public static final int INTERNET_OPTION_SEND_TIMEOUT = 5;
	public static final int INTERNET_OPTION_CONTROL_SEND_TIMEOUT = INTERNET_OPTION_SEND_TIMEOUT;
	public static final int INTERNET_OPTION_RECEIVE_TIMEOUT = 6;
	public static final int INTERNET_OPTION_CONTROL_RECEIVE_TIMEOUT = INTERNET_OPTION_RECEIVE_TIMEOUT;
	public static final int INTERNET_OPTION_DATA_SEND_TIMEOUT = 7;
	public static final int INTERNET_OPTION_DATA_RECEIVE_TIMEOUT = 8;
	public static final int INTERNET_OPTION_HANDLE_TYPE = 9;
	public static final int INTERNET_OPTION_LISTEN_TIMEOUT = 11;
	public static final int INTERNET_OPTION_READ_BUFFER_SIZE = 12;
	public static final int INTERNET_OPTION_WRITE_BUFFER_SIZE = 13;

public static final int INTERNET_OPTION_ASYNC_ID = 15;
	public static final int INTERNET_OPTION_ASYNC_PRIORITY = 16;

public static final int INTERNET_OPTION_PARENT_HANDLE = 21;
	public static final int INTERNET_OPTION_KEEP_CONNECTION = 22;
	public static final int INTERNET_OPTION_REQUEST_FLAGS = 23;
	public static final int INTERNET_OPTION_EXTENDED_ERROR = 24;

public static final int INTERNET_OPTION_OFFLINE_MODE = 26;
	public static final int INTERNET_OPTION_CACHE_STREAM_HANDLE = 27;
	public static final int INTERNET_OPTION_USERNAME = 28;
	public static final int INTERNET_OPTION_PASSWORD = 29;
	public static final int INTERNET_OPTION_ASYNC = 30;
	public static final int INTERNET_OPTION_SECURITY_FLAGS = 31;
	public static final int INTERNET_OPTION_SECURITY_CERTIFICATE_STRUCT = 32;
	public static final int INTERNET_OPTION_DATAFILE_NAME = 33;
	public static final int INTERNET_OPTION_URL = 34;
	public static final int INTERNET_OPTION_SECURITY_CERTIFICATE = 35;
	public static final int INTERNET_OPTION_SECURITY_KEY_BITNESS = 36;
	public static final int INTERNET_OPTION_REFRESH = 37;
	public static final int INTERNET_OPTION_PROXY = 38;
	public static final int INTERNET_OPTION_SETTINGS_CHANGED = 39;
	public static final int INTERNET_OPTION_VERSION = 40;
	public static final int INTERNET_OPTION_USER_AGENT = 41;
	public static final int INTERNET_OPTION_END_BROWSER_SESSION = 42;
	public static final int INTERNET_OPTION_PROXY_USERNAME = 43;
	public static final int INTERNET_OPTION_PROXY_PASSWORD = 44;
	public static final int INTERNET_OPTION_CONTEXT_VALUE = 45;
	public static final int INTERNET_OPTION_CONNECT_LIMIT = 46;
	public static final int INTERNET_OPTION_SECURITY_SELECT_CLIENT_CERT = 47;
	public static final int INTERNET_OPTION_POLICY = 48;
	public static final int INTERNET_OPTION_DISCONNECTED_TIMEOUT = 49;
	public static final int INTERNET_OPTION_CONNECTED_STATE = 50;
	public static final int INTERNET_OPTION_IDLE_STATE = 51;
	public static final int INTERNET_OPTION_OFFLINE_SEMANTICS = 52;
	public static final int INTERNET_OPTION_SECONDARY_CACHE_KEY = 53;
	public static final int INTERNET_OPTION_CALLBACK_FILTER = 54;
	public static final int INTERNET_OPTION_CONNECT_TIME = 55;
	public static final int INTERNET_OPTION_SEND_THROUGHPUT = 56;
	public static final int INTERNET_OPTION_RECEIVE_THROUGHPUT = 57;
	public static final int INTERNET_OPTION_REQUEST_PRIORITY = 58;
	public static final int INTERNET_OPTION_HTTP_VERSION = 59;
	public static final int INTERNET_OPTION_RESET_URLCACHE_SESSION = 60;
	public static final int INTERNET_OPTION_ERROR_MASK = 62;
	public static final int INTERNET_OPTION_FROM_CACHE_TIMEOUT = 63;
	public static final int INTERNET_OPTION_BYPASS_EDITED_ENTRY = 64;
	public static final int INTERNET_OPTION_DIAGNOSTIC_SOCKET_INFO = 67;
	public static final int INTERNET_OPTION_CODEPAGE = 68;
	public static final int INTERNET_OPTION_CACHE_TIMESTAMPS = 69;
	public static final int INTERNET_OPTION_DISABLE_AUTODIAL = 70;
	public static final int INTERNET_OPTION_MAX_CONNS_PER_SERVER = 73;
	public static final int INTERNET_OPTION_MAX_CONNS_PER_1_0_SERVER = 74;
	public static final int INTERNET_OPTION_PER_CONNECTION_OPTION = 75;
	public static final int INTERNET_OPTION_DIGEST_AUTH_UNLOAD = 76;
	public static final int INTERNET_OPTION_IGNORE_OFFLINE = 77;
	public static final int INTERNET_OPTION_IDENTITY = 78;
	public static final int INTERNET_OPTION_REMOVE_IDENTITY = 79;
	public static final int INTERNET_OPTION_ALTER_IDENTITY = 80;
	public static final int INTERNET_OPTION_SUPPRESS_BEHAVIOR = 81;
	public static final int INTERNET_OPTION_AUTODIAL_MODE = 82;
	public static final int INTERNET_OPTION_AUTODIAL_CONNECTION = 83;
	public static final int INTERNET_OPTION_CLIENT_CERT_CONTEXT = 84;
	public static final int INTERNET_OPTION_AUTH_FLAGS = 85;
	public static final int INTERNET_OPTION_COOKIES_3RD_PARTY = 86;
	public static final int INTERNET_OPTION_DISABLE_PASSPORT_AUTH = 87;
	public static final int INTERNET_OPTION_SEND_UTF8_SERVERNAME_TO_PROXY = 88;
	public static final int INTERNET_OPTION_EXEMPT_CONNECTION_LIMIT = 89;
	public static final int INTERNET_OPTION_ENABLE_PASSPORT_AUTH = 90;

public static final int INTERNET_OPTION_HIBERNATE_INACTIVE_WORKER_THREADS = 91;
	public static final int INTERNET_OPTION_ACTIVATE_WORKER_THREADS = 92;
	public static final int INTERNET_OPTION_RESTORE_WORKER_THREAD_DEFAULTS = 93;
	public static final int INTERNET_OPTION_SOCKET_SEND_BUFFER_LENGTH = 94;
	public static final int INTERNET_OPTION_PROXY_SETTINGS_CHANGED = 95;

public static final int INTERNET_OPTION_DATAFILE_EXT = 96;

// BOOL InternetSetOption(
	// _In_ HINTERNET hInternet,
	// _In_ DWORD dwOption,
	// _In_ LPVOID lpBuffer,
	// _In_ DWORD dwBufferLength
	// );
	public boolean InternetSetOption(Pointer hInternet, int dwOption,
			Pointer lpBuffer, int dwBufferLength);

// typedef struct _FILETIME {
	// DWORD dwLowDateTime;
	// DWORD dwHighDateTime;
	// } FILETIME, *PFILETIME;
	public static class FILETIME extends Structure {
		public static class ByReference extends FILETIME implements
				Structure.ByReference {
			public ByReference() {
			}

public ByReference(Pointer memory) {
				super(memory);
			}
		}

public FILETIME() {
		}

public FILETIME(Pointer memory) {
			super(memory);
			read();
		}

public int dwLowDateTime;
		public int dwHighDateTime;

@SuppressWarnings("rawtypes")
		@Override
		protected List getFieldOrder() {
			return Arrays.asList(new String[] { "dwLowDateTime",
					"dwHighDateTime" });
		}
	}

// typedef struct {
	// DWORD dwOption;
	// union {
	// DWORD dwValue;
	// LPTSTR pszValue;
	// FILETIME ftValue;
	// } Value;
	// } INTERNET_PER_CONN_OPTION, *LPINTERNET_PER_CONN_OPTION;
	public static class INTERNET_PER_CONN_OPTION extends Structure {
		public static class ByReference extends INTERNET_PER_CONN_OPTION
				implements Structure.ByReference {
			public ByReference() {
			}

public ByReference(Pointer memory) {
				super(memory);
			}
		}

public INTERNET_PER_CONN_OPTION() {
		}

public INTERNET_PER_CONN_OPTION(Pointer memory) {
			super(memory);
			read();
		}

public static class VALUE_UNION extends Union {
			public int dwValue;
			public String pszValue;
			public FILETIME ftValue;
		}

public int dwOption;
		public VALUE_UNION Value = new VALUE_UNION();

@SuppressWarnings("rawtypes")
		@Override
		protected List getFieldOrder() {
			return Arrays.asList(new String[] { "dwOption", "Value" });
		}
	}

// typedef struct {
	// DWORD dwSize;
	// LPTSTR pszConnection;
	// DWORD dwOptionCount;
	// DWORD dwOptionError;
	// LPINTERNET_PER_CONN_OPTION pOptions;
	// } INTERNET_PER_CONN_OPTION_LIST, *LPINTERNET_PER_CONN_OPTION_LIST;
	public static class INTERNET_PER_CONN_OPTION_LIST extends Structure {
		public static class ByReference extends INTERNET_PER_CONN_OPTION_LIST
				implements Structure.ByReference {
			public ByReference() {
			}

public ByReference(Pointer memory) {
				super(memory);
			}
		}

public INTERNET_PER_CONN_OPTION_LIST() {
		}

public INTERNET_PER_CONN_OPTION_LIST(Pointer memory) {
			super(memory);
			read();
		}

public int dwSize;
		public String pszConnection;
		public int dwOptionCount;
		public int dwOptionError;
		public INTERNET_PER_CONN_OPTION.ByReference pOptions;

@SuppressWarnings("rawtypes")
		@Override
		protected List getFieldOrder() {
			return Arrays.asList(new String[] { "dwSize", "pszConnection",
					"dwOptionCount", "dwOptionError", "pOptions" });
		}
	}
}

이 소스는 java에서 registery값에 접근하는 jRegistryKey라는 라이브러리를 사용한다.

그 라이브러리를 이용하기위한 Interface가 WinInet.java이고, 실제 동작하는건 Proxy.java라고 보면 된다.

링크 : https://sourceforge.net/projects/jregistrykey/

라이브러리 사용하는 방법은 일반적이니 따로 적지 않겠다.

설명하자면, Windows의 Proxy 설정값은 Registery의 CURRENT_USER의 Software\\Microsoft\\Windows\\CurrentVersion\\Internet Settings

밑에 위치하게 되고 여기에서 프록시 설정에 대한 값들이 있는데,

ProxyEnable : 현재 프록시가 켜져있는지 아닌지의 값. 0 또는 1

ProxyServer : 프록시 서버 주소

ProxyOverride : 프록시 주소를 사용하지 않는 예외 주소들

AutoConfigURL : 프록시 자동구성 스크립트 주소

이다.

이 값을 이용해 현재 프록시가 켜져 있다면 꺼주고, 꺼져 있다면 같은 폴더의 dat폴더 밑 config.dat파일을 참조하여 값을 읽어와 그 설정과 같이

Proxy 설정을 해준다.

config.dat파일의 첫번째 라인을 자동구성 스크립트에, 두번째라인은 프록시 주소에, 세번째 라인은 예외 주소, 네번째 라인은 다이어로그가 실행되는 시간(1000이 1초) 로 구성된다.

이렇게 만들어진 프로그램을 바로가기로 만들어 우측버튼의 속성을 통해 바로가기키를 지정해두면 단축키로 프록시 활성/비활성화가 가능해진다.

만들어진 Tool로 손쉽게 프록시 활성/비활성화를 하면서 홈페이지을 조지면 된다!

저작자표시 비영리 변경금지

'Develop > UTIL' 카테고리의 다른 글

python으로 Proxy 변경 Tool 만들기 :) (0)	2017.07.20
Fortify 구버전 리포트 형태 excel로 파싱하기 : ) (0)	2017.07.20
[개발] Python으로 APK 뜯는 스크립트 만들기 (0)	2016.05.31

YPrefer's Develop&Security

[개발] Java로 Proxy 변경 Tool 만들기

'Develop > UTIL' 카테고리의 다른 글

+ Recent posts

티스토리툴바