Support for Microsoft Update

Author: ExplodingBottle

src/io/github/explodingbottle/jmagicproxy/implementation/WUProxy.java

@@ -43,6 +43,8 @@ public WUProxy() {
 
 	private static final String DLREP1_REPLACEMENT = "/v9/windowsupdate/a/selfupdate/";
 	private static final String DLREP2_REPLACEMENT = "/v11/3/windowsupdate/selfupdate/";
+	private static final String MU_DLREP1_REPLACEMENT = "/v9/microsoftupdate/a/selfupdate/";
+	private static final String MU_DLREP2_REPLACEMENT = "/v9/1/microsoftupdate/b/selfupdate/";
 
 	@Override
 	public ConnectionDirective onReceiveProxyRequest(HttpRequestHeader request) {
@@ -70,6 +72,21 @@ public ConnectionDirective onReceiveProxyRequest(HttpRequestHeader request) {
 			}
 		}
 
+		if (ProxyMain.getPropertiesProvider().getAsBoolean(PropertyKey.WUPROXY_REDIRECT_WUCLIENT)) {
+			if (request.getHost().toLowerCase().contains(MU_DLREP2_REPLACEMENT)) {
+				computed.setHost("download.windowsupdate.com");
+				computed.setPort(80);
+				String preHost = computed.getOutcomingRequest().getHost();
+				computed.getOutcomingRequest()
+						.setHost(preHost.toLowerCase().replace(MU_DLREP2_REPLACEMENT, MU_DLREP1_REPLACEMENT));
+				if (computed.getOutcomingRequest() != null) {
+					computed.getOutcomingRequest().getHeaders().put("Host", "download.windowsupdate.com");
+				}
+				logger.log(LoggingLevel.INFO,
+						"Replaced " + preHost + " by " + computed.getOutcomingRequest().getHost() + ".");
+			}
+		}
+
 		if (request.getHost().toLowerCase().contains("/windowsupdate/v6/shared/js/redirect.js")
 				|| request.getHost().toLowerCase().contains("/microsoftupdate/v6/shared/js/redirect.js")) {
 			logger.log(LoggingLevel.INFO, "Found a redirect.js request.");

src/io/github/explodingbottle/jmagicproxy/proxy/ConnectionDirectiveHandler.java

@@ -22,9 +22,13 @@
 import java.io.InputStream;
 import java.io.OutputStream;
 import java.net.Socket;
+import java.text.SimpleDateFormat;
 import java.util.ArrayList;
 import java.util.Collections;
+import java.util.Date;
 import java.util.List;
+import java.util.Locale;
+import java.util.TimeZone;
 
 import io.github.explodingbottle.jmagicproxy.ProxyMain;
 import io.github.explodingbottle.jmagicproxy.api.ConnectionDirective;
@@ -228,13 +232,6 @@ public void openSocket() {
 				try {
 
 					openingTool.run();
-					/*
-					 * referenceSocket = new Socket(InetAddress.getByName(directive.getHost()),
-					 * directive.getPort()); inputStream = referenceSocket.getInputStream();
-					 * outputStream = referenceSocket.getOutputStream(); rewriteDirectiveLine();
-					 * pipeThread = new SimpleInputOutputPipeThread(inputStream,
-					 * handlerThread.getOutputStream(), this); pipeThread.start();
-					 */
 				} catch (Exception e) {
 					logger.log(LoggingLevel.WARN, "Failed to open the outgoing socket.", e);
 					closeSocket();
@@ -246,6 +243,12 @@ public void openSocket() {
 				inputStream = new FileInputStream(directive.getFileInput());
 				handlerThread.getOutputStream().write(new String("HTTP/1.1 200 OK\r\n").getBytes());
 				handlerThread.getOutputStream().write(new String("Connection: Keep-Alive\r\n").getBytes());
+				SimpleDateFormat f = new SimpleDateFormat("EEE, dd MMM yyyy HH:mm:ss z", Locale.US);
+				f.setTimeZone(TimeZone.getTimeZone("GMT"));
+				String lM = f.format(new Date(directive.getFileInput().lastModified()));
+				handlerThread.getOutputStream().write(new String("Last-Modified: " + lM + "\r\n").getBytes());
+				handlerThread.getOutputStream()
+						.write(new String("Content-Type: application/octet-stream\r\n").getBytes());
 				handlerThread.getOutputStream().write(
 						new String("Content-Length: " + directive.getFileInput().length() + "\r\n\r\n").getBytes());
 				pipeThread = new SimpleInputOutputPipeThread(inputStream, handlerThread.getOutputStream(), this);

src/io/github/explodingbottle/jmagicproxy/proxy/ssl/SSLCommunicationServer.java

@@ -240,7 +240,6 @@ public void run() {
 				int read = heartInput.read(buffer, 0, buffer.length);
 				while (!interrupted() && read != -1) {
 					Integer offset = handleLineRead(read);
-					;
 					if (offset != null) {
 						if (outgoingHandler != null)
 							outgoingHandler.feedOutput(buffer, offset, read - offset);

src/io/github/explodingbottle/jmagicproxy/proxy/ssl/SSLDirectiveHandler.java

@@ -22,9 +22,13 @@
 import java.io.InputStream;
 import java.io.OutputStream;
 import java.net.Socket;
+import java.text.SimpleDateFormat;
 import java.util.ArrayList;
 import java.util.Collections;
+import java.util.Date;
 import java.util.List;
+import java.util.Locale;
+import java.util.TimeZone;
 
 import javax.net.ssl.SSLSocket;
 
@@ -117,6 +121,11 @@ public void openSocket() {
 				inputStream = new FileInputStream(directive.getFileInput());
 				parent.getHeartOutput().write(new String("HTTP/1.1 200 OK\r\n").getBytes());
 				parent.getHeartOutput().write(new String("Connection: Keep-Alive\r\n").getBytes());
+				SimpleDateFormat f = new SimpleDateFormat("EEE, dd MMM yyyy HH:mm:ss z", Locale.US);
+				f.setTimeZone(TimeZone.getTimeZone("GMT"));
+				String lM = f.format(new Date(directive.getFileInput().lastModified()));
+				parent.getHeartOutput().write(new String("Last-Modified: " + lM + "\r\n").getBytes());
+				parent.getHeartOutput().write(new String("Content-Type: application/octet-stream\r\n").getBytes());
 				parent.getHeartOutput().write(
 						new String("Content-Length: " + directive.getFileInput().length() + "\r\n\r\n").getBytes());
 				ioPipe = new SSLInputOutputPipeThread(inputStream, parent.getHeartOutput(), this);