RustPython · windelbouwman · Aug 13, 2019 · Aug 12, 2019 · Aug 12, 2019 · Aug 12, 2019
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/tests/snippets/stdlib_zlib.py b/tests/snippets/stdlib_zlib.py
@@ -0,0 +1,62 @@
+import zlib
+from testutils import assert_raises
+
+# checksum functions
+assert zlib.crc32(b"123") == 2286445522
+assert zlib.crc32(b"123", 1) == 2307525093
+assert zlib.crc32(b"123", 2) == 2345449404
+assert zlib.crc32(b"123", 3) == 2316230027
+assert zlib.crc32(b"123", 4) == 2403453710
+assert zlib.crc32(b"123", 5) == 2390991161
+assert zlib.crc32(b"123", 6) == 2361728864
+assert zlib.crc32(b"123", -123) == 3515918521
+assert zlib.crc32(b"123", -122) == 3554023136
+assert zlib.crc32(b"123", -121) == 3524558039
+assert zlib.crc32(b"123", -120) == 3645389802
+assert zlib.crc32(b"123", -119) == 3632943581
+assert zlib.crc32(b"123", -118) == 3670863748
+assert zlib.crc32(b"123") == zlib.crc32(b"123", 0)
+
+assert zlib.adler32(b"456") == 20906144
+assert zlib.adler32(b"456", 1) == 20906144
+assert zlib.adler32(b"456", 2) == 21102753
+assert zlib.adler32(b"456", 3) == 21299362
+assert zlib.adler32(b"456", 4) == 21495971
+assert zlib.adler32(b"456", 5) == 21692580
+assert zlib.adler32(b"456", 6) == 21889189
+assert zlib.adler32(b"456", -123) == 393267
+assert zlib.adler32(b"456", -122) == 589876
+assert zlib.adler32(b"456", -121) == 786485
+assert zlib.adler32(b"456", -120) == 983094
+assert zlib.adler32(b"456", -119) == 1179703
+assert zlib.adler32(b"456", -118) == 1376312
+assert zlib.adler32(b"456") == zlib.adler32(b"456", 1)
+
+# compression
+lorem = bytes("Lorem ipsum dolor sit amet", "utf-8")
+
+compressed_lorem_list = [
+    b"x\x01\x01\x1a\x00\xe5\xffLorem ipsum dolor sit amet\x83\xd5\t\xc5",
+    b"x\x01\xf3\xc9/J\xcdU\xc8,(.\xcdUH\xc9\xcf\xc9/R(\xce,QH\xccM-\x01\x00\x83\xd5\t\xc5",
+    b"x^\xf3\xc9/J\xcdU\xc8,(.\xcdUH\xc9\xcf\xc9/R(\xce,QH\xccM-\x01\x00\x83\xd5\t\xc5",
+    b"x^\xf3\xc9/J\xcdU\xc8,(.\xcdUH\xc9\xcf\xc9/R(\xce,QH\xccM-\x01\x00\x83\xd5\t\xc5",
+    b"x^\xf3\xc9/J\xcdU\xc8,(.\xcdUH\xc9\xcf\xc9/R(\xce,QH\xccM-\x01\x00\x83\xd5\t\xc5",
+    b"x^\xf3\xc9/J\xcdU\xc8,(.\xcdUH\xc9\xcf\xc9/R(\xce,QH\xccM-\x01\x00\x83\xd5\t\xc5",
+    b"x\x9c\xf3\xc9/J\xcdU\xc8,(.\xcdUH\xc9\xcf\xc9/R(\xce,QH\xccM-\x01\x00\x83\xd5\t\xc5",
+    b"x\xda\xf3\xc9/J\xcdU\xc8,(.\xcdUH\xc9\xcf\xc9/R(\xce,QH\xccM-\x01\x00\x83\xd5\t\xc5",
+    b"x\xda\xf3\xc9/J\xcdU\xc8,(.\xcdUH\xc9\xcf\xc9/R(\xce,QH\xccM-\x01\x00\x83\xd5\t\xc5",
+    b"x\xda\xf3\xc9/J\xcdU\xc8,(.\xcdUH\xc9\xcf\xc9/R(\xce,QH\xccM-\x01\x00\x83\xd5\t\xc5",
+]
+
+for level, text in enumerate(compressed_lorem_list):
+    assert zlib.compress(lorem, level) == text
+
+# default level
+assert zlib.compress(lorem) == zlib.compress(lorem, -1) == zlib.compress(lorem, 6)
+
+# decompression
+for text in compressed_lorem_list:
+    assert zlib.decompress(text) == lorem
+
+assert_raises(zlib.error, lambda: zlib.compress(b"123", -40))
+assert_raises(zlib.error, lambda: zlib.compress(b"123", 10))
diff --git a/vm/Cargo.toml b/vm/Cargo.toml
@@ -72,5 +72,9 @@ flamer = { version = "0.3", optional = true }
 pwd = "1"
 
 [target.'cfg(not(target_arch = "wasm32"))'.dependencies]
+crc32fast = "1.2.0"
+adler32 = "1.0.3"
+flate2 = { version = "1.0", features = ["zlib"], default-features = false }
+libz-sys = "1.0.25"
 gethostname = "0.2.0"
 subprocess = "0.1.18"
diff --git a/vm/src/stdlib/mod.rs b/vm/src/stdlib/mod.rs
@@ -43,6 +43,8 @@ mod pwd;
 pub mod signal;
 #[cfg(not(target_arch = "wasm32"))]
 mod subprocess;
+#[cfg(not(target_arch = "wasm32"))]
+mod zlib;
 
 use crate::pyobject::PyObjectRef;
 
@@ -99,6 +101,7 @@ pub fn get_module_inits() -> HashMap<String, StdlibInitFunc> {
         modules.insert("socket".to_string(), Box::new(socket::make_module));
         modules.insert("signal".to_string(), Box::new(signal::make_module));
         modules.insert("subprocess".to_string(), Box::new(subprocess::make_module));
+        modules.insert("zlib".to_string(), Box::new(zlib::make_module));
     }
 
     // Unix-only