gh-30 fix memory leak

unknow-server-servlet/src/main/java/unknow/server/servlet/http2/Http2Processor.java

@@ -98,6 +98,7 @@ public boolean isClosed() {
 	public void close() {
 		for (Http2Stream s : streams.values())
 			s.close(true);
+		streams.clear();
 	}
 
 	/**
@@ -146,10 +147,12 @@ private void readFrame(Buffers buf) throws InterruptedException {
 					return;
 				}
 
-				streams.set(id, s = new Http2Stream(co, id, this, initialWindow));
+				s = new Http2Stream(co, id, this, initialWindow);
 
 				if ((flags & 0x1) == 1) // END_STREAM 
 					s.close(false);
+				else
+					streams.set(id, s);
 				if ((flags & 0x8) == 1) {
 					pad = buf.read(false);
 					if (pad >= size) {
@@ -162,7 +165,7 @@ private void readFrame(Buffers buf) throws InterruptedException {
 					buf.skip(5);
 				}
 
-				r = new FrameHeader(size, flags, id, pad).process(buf);
+				r = new FrameHeader(size, flags, id, pad, s).process(buf);
 				return;
 			case 2: // priority
 				r = new FrameReader(size, flags, id).process(buf);
@@ -238,11 +241,7 @@ private void readFrame(Buffers buf) throws InterruptedException {
 					return;
 				}
 
-				r = new FrameHeader(size, flags, id, pad).process(buf);
-
-				wantContinuation = (flags & 0x4) == 0;
-				if (!wantContinuation)
-					s.start();
+				r = new FrameHeader(size, flags, id, pad, s).process(buf);
 				return;
 			default:
 				goaway(PROTOCOL_ERROR);
@@ -502,6 +501,7 @@ protected FrameGoAway(int size, int flags, int id) {
 			b = new byte[4];
 		}
 
+		@Override
 		public FrameReader process(Buffers buf) throws InterruptedException {
 			if (lastId >= 0)
 				return super.process(buf);
@@ -552,9 +552,9 @@ private class FrameHeader extends FrameReader {
 		private final Buffers remain;
 		private int pad;
 
-		protected FrameHeader(int size, int flags, int id, int pad) {
+		protected FrameHeader(int size, int flags, int id, int pad, Http2Stream s) {
 			super(size, flags, id);
-			this.s = streams.get(id);
+			this.s = s;
 			this.remain = new Buffers();
 			this.pad = pad;
 		}

unknow-server-util/src/main/java/unknow/server/util/data/IntArrayMap.java

@@ -69,15 +69,19 @@ public T get(int key) {
 	public T set(int key, T value) {
 		int i = Arrays.binarySearch(keys, 0, len, key);
 		if (i >= 0) {
+			T old = values[i];
 			values[i] = value;
-			return null;
+			return old;
 		}
 		ensure(++len);
 		i = -i - 1;
-		T old = values[i];
+		if (i < len - 1) {
+			System.arraycopy(keys, i, keys, i + 1, len - i - 1);
+			System.arraycopy(values, i, values, i + 1, len - i - 1);
+		}
 		keys[i] = key;
 		values[i] = value;
-		return old;
+		return null;
 	}
 
 	/**
@@ -91,12 +95,12 @@ public boolean setOnce(int key, T value) {
 		int i = Arrays.binarySearch(keys, 0, len, key);
 		if (i >= 0)
 			return false;
-		if (i < len) {
-			System.arraycopy(keys, i, keys, i + 1, len - i);
-			System.arraycopy(values, i, values, i + 1, len - i);
-		}
 		ensure(++len);
 		i = -i - 1;
+		if (i < len - 1) {
+			System.arraycopy(keys, i, keys, i + 1, len - i - 1);
+			System.arraycopy(values, i, values, i + 1, len - i - 1);
+		}
 		keys[i] = key;
 		values[i] = value;
 		return true;
@@ -107,7 +111,24 @@ public boolean setOnce(int key, T value) {
 	 * @return the removed value
 	 */
 	public T remove(int key) {
-		return set(key, null);
+		int i = Arrays.binarySearch(keys, 0, len, key);
+		if (i < 0)
+			return null;
+		T old = values[i];
+		len--;
+		System.arraycopy(keys, i + 1, keys, i, len - i);
+		System.arraycopy(values, i + 1, values, i, len - i);
+		values[len] = null;
+		return old;
+	}
+
+	/**
+	 * empty the map
+	 */
+	public void clear() {
+		for (int i = 0; i < len; i++)
+			values[i] = null;
+		len = 0;
 	}
 
 	/**