Try not to crash with too big a files

Author: champo

src/main/java/ar/edu/itba/pdc/duta/net/Server.java

@@ -38,15 +38,18 @@ public class Server {
 	private InetSocketAddress connectTo;
 
 	private int port;
+	
+	private int sizeLimit;
 
-	private Server(int port, int adminPort, InetSocketAddress connectTo) {
+	private Server(int port, int adminPort, InetSocketAddress connectTo, int sizeLimit) {
 		super();
 		resolver = new ConnectionPool();
 		filters = new Filters();
 
 		Server.adminPort = adminPort;
 		this.port = port;
 		this.connectTo = connectTo;
+		this.sizeLimit = sizeLimit;
 	}
 
 	public void start() {
@@ -135,6 +138,7 @@ public static void main(String[] args) throws IOException {
 		InetSocketAddress connectTo = null;
 		int port = 9999;
 		int adminPort = 1337;
+		int sizeLimit = 100 * 1024 * 1024;
 
 		for (int i = 0; i < args.length; i++) {
 			String arg = args[i];
@@ -181,6 +185,23 @@ public static void main(String[] args) throws IOException {
 					return;
 				}
 
+			} else if (arg.startsWith("--size-limit=")) {
+
+				String portString = arg.substring(arg.indexOf('=') + 1);
+				
+				try {
+					int size = Integer.parseInt(portString);
+					if (size >= 0) {
+						sizeLimit = size;
+					}
+					continue;
+				} catch (NumberFormatException e) {
+				}
+				
+				System.out.println("'" + sizeLimit + "' is not a size number (must be in MBs)");
+				usage();
+				return;
+
 			}
 
 		}
@@ -190,8 +211,9 @@ public static void main(String[] args) throws IOException {
 		if (connectTo != null) {
 			System.out.println("Chaining to: " + connectTo);
 		}
+		System.out.println("Size limit: " + sizeLimit);
 
-		Server.run(port, adminPort, connectTo);
+		Server.run(port, adminPort, connectTo, sizeLimit);
 		System.exit(0);
 	}
 
@@ -200,10 +222,11 @@ private static void usage() {
 		System.out.println("\t--chain=ip:port    Chain to another proxy");
 		System.out.println("\t--port=port        Listen for requests on `port`");
 		System.out.println("\t--admin-port=port  Listen for admin requests on `port`");
+		System.out.println("\t--size-limit=size  Limit the size of messages for transformations");
 	}
 
-	private static void run(int port, int adminPort, InetSocketAddress connectTo) {
-		instance = new Server(port, adminPort, connectTo);
+	private static void run(int port, int adminPort, InetSocketAddress connectTo, int sizeLimit) {
+		instance = new Server(port, adminPort, connectTo, sizeLimit);
 		instance.start();
 	}
 
@@ -232,5 +255,9 @@ public static Filters getFilters() {
 	public static InetSocketAddress getChainAddress() {
 		return instance.connectTo;
 	}
+	
+	public static int getSizeLimit() {
+		return instance.sizeLimit;
+	}
 
 }

src/main/java/ar/edu/itba/pdc/duta/proxy/filter/http/ImageRotationFilter.java

@@ -16,9 +16,11 @@
 import org.apache.log4j.Logger;
 
 import ar.edu.itba.pdc.duta.admin.Stats;
+import ar.edu.itba.pdc.duta.http.MessageFactory;
 import ar.edu.itba.pdc.duta.http.model.MediaType;
 import ar.edu.itba.pdc.duta.http.model.Message;
 import ar.edu.itba.pdc.duta.http.model.MessageHeader;
+import ar.edu.itba.pdc.duta.net.Server;
 import ar.edu.itba.pdc.duta.net.buffer.DataBuffer;
 import ar.edu.itba.pdc.duta.proxy.filter.Filter;
 import ar.edu.itba.pdc.duta.proxy.filter.FilterPart;
@@ -66,6 +68,10 @@ public Interest checkInterest(MessageHeader header) {
 
 		@Override
 		public Message filter(Operation op, Message msg) {
+			
+			if (Server.getSizeLimit() <= msg.getBody().getWriteIndex()) {
+				return MessageFactory.build(501, "Not Implemented", null);
+			}
 
 			try {
 

src/main/java/ar/edu/itba/pdc/duta/proxy/filter/http/L33tFilter.java

@@ -10,6 +10,7 @@
 import ar.edu.itba.pdc.duta.http.model.MediaType;
 import ar.edu.itba.pdc.duta.http.model.Message;
 import ar.edu.itba.pdc.duta.http.model.MessageHeader;
+import ar.edu.itba.pdc.duta.net.Server;
 import ar.edu.itba.pdc.duta.net.buffer.DataBuffer;
 import ar.edu.itba.pdc.duta.proxy.filter.Filter;
 import ar.edu.itba.pdc.duta.proxy.filter.FilterPart;
@@ -65,6 +66,10 @@ public Interest checkInterest(MessageHeader header) {
 
 		@Override
 		public Message filter(Operation op, Message msg) {
+			
+			if (Server.getSizeLimit() <= msg.getBody().getWriteIndex()) {
+				return MessageFactory.build(501, "Not Implemented", null);
+			}
 
 			byte[] bytes;