1 files changed, 111 insertions, 0 deletions
diff --git a/java/org/brotli/wrapper/enc/EncoderJNI.java b/java/org/brotli/wrapper/enc/EncoderJNI.java
new file mode 100755
index 0000000..dd7cff3
--- /dev/null
+++ b/java/org/brotli/wrapper/enc/EncoderJNI.java
@@ -0,0 +1,111 @@
+/* Copyright 2017 Google Inc. All Rights Reserved.
+
+   Distributed under MIT license.
+   See file LICENSE for detail or copy at https://opensource.org/licenses/MIT
+*/
+
+package org.brotli.wrapper.enc;
+
+import java.io.IOException;
+import java.nio.ByteBuffer;
+
+/**
+ * JNI wrapper for brotli encoder.
+ */
+class EncoderJNI {
+  private static native ByteBuffer nativeCreate(long[] context, ByteBuffer customDictionary);
+  private static native void nativePush(long[] context, int length);
+  private static native ByteBuffer nativePull(long[] context);
+  private static native void nativeDestroy(long[] context);
+
+  enum Operation {
+    PROCESS,
+    FLUSH,
+    FINISH
+  }
+
+  static class Wrapper {
+    protected final long[] context = new long[4];
+    private final ByteBuffer inputBuffer;
+
+    Wrapper(int inputBufferSize, int quality, int lgwin, ByteBuffer customDictionary)
+        throws IOException {
+      if (customDictionary != null && !customDictionary.isDirect()) {
+        throw new IllegalArgumentException("LZ77 dictionary must be direct ByteBuffer");
+      }
+      this.context[1] = inputBufferSize;
+      this.context[2] = quality;
+      this.context[3] = lgwin;
+      this.inputBuffer = nativeCreate(this.context, customDictionary);
+      if (this.context[0] == 0) {
+        throw new IOException("failed to initialize native brotli encoder");
+      }
+      this.context[1] = 1;
+      this.context[2] = 0;
+      this.context[3] = 0;
+    }
+
+    void push(Operation op, int length) {
+      if (length < 0) {
+        throw new IllegalArgumentException("negative block length");
+      }
+      if (context[0] == 0) {
+        throw new IllegalStateException("brotli encoder is already destroyed");
+      }
+      if (!isSuccess() || hasMoreOutput()) {
+        throw new IllegalStateException("pushing input to encoder in unexpected state");
+      }
+      if (hasRemainingInput() && length != 0) {
+        throw new IllegalStateException("pushing input to encoder over previous input");
+      }
+      context[1] = op.ordinal();
+      nativePush(context, length);
+    }
+
+    boolean isSuccess() {
+      return context[1] != 0;
+    }
+
+    boolean hasMoreOutput() {
+      return context[2] != 0;
+    }
+
+    boolean hasRemainingInput() {
+      return context[3] != 0;
+    }
+
+    ByteBuffer getInputBuffer() {
+      return inputBuffer;
+    }
+
+    ByteBuffer pull() {
+      if (context[0] == 0) {
+        throw new IllegalStateException("brotli encoder is already destroyed");
+      }
+      if (!isSuccess() || !hasMoreOutput()) {
+        throw new IllegalStateException("pulling while data is not ready");
+      }
+      return nativePull(context);
+    }
+
+    /**
+     * Releases native resources.
+     */
+    void destroy() {
+      if (context[0] == 0) {
+        throw new IllegalStateException("brotli encoder is already destroyed");
+      }
+      nativeDestroy(context);
+      context[0] = 0;
+    }
+
+    @Override
+    protected void finalize() throws Throwable {
+      if (context[0] != 0) {
+        /* TODO: log resource leak? */
+        destroy();
+      }
+      super.finalize();
+    }
+  }
+}