summaryrefslogtreecommitdiff
path: root/io
diff options
context:
space:
mode:
Diffstat (limited to 'io')
-rw-r--r--io/util.ts1
-rw-r--r--io/util_test.ts36
2 files changed, 37 insertions, 0 deletions
diff --git a/io/util.ts b/io/util.ts
index 3266e5018..185732b36 100644
--- a/io/util.ts
+++ b/io/util.ts
@@ -4,6 +4,7 @@ import { Buffer, Reader } from "deno";
// from `src`.
// Returns the number of bytes copied.
export function copyBytes(dst: Uint8Array, src: Uint8Array, off = 0): number {
+ off = Math.max(0, Math.min(off, dst.byteLength));
const r = dst.byteLength - off;
if (src.byteLength > r) {
src = src.subarray(0, r);
diff --git a/io/util_test.ts b/io/util_test.ts
new file mode 100644
index 000000000..7318d887c
--- /dev/null
+++ b/io/util_test.ts
@@ -0,0 +1,36 @@
+import { test, assert } from "../testing/mod.ts";
+import { copyBytes } from "./util.ts";
+
+test(function testCopyBytes() {
+ let dst = new Uint8Array(4);
+
+ dst.fill(0);
+ let src = Uint8Array.of(1, 2);
+ let len = copyBytes(dst, src, 0);
+ assert(len === 2);
+ assert.equal(dst, Uint8Array.of(1, 2, 0, 0));
+
+ dst.fill(0);
+ src = Uint8Array.of(1, 2);
+ len = copyBytes(dst, src, 1);
+ assert(len === 2);
+ assert.equal(dst, Uint8Array.of(0, 1, 2, 0));
+
+ dst.fill(0);
+ src = Uint8Array.of(1, 2, 3, 4, 5);
+ len = copyBytes(dst, src);
+ assert(len === 4);
+ assert.equal(dst, Uint8Array.of(1, 2, 3, 4));
+
+ dst.fill(0);
+ src = Uint8Array.of(1, 2);
+ len = copyBytes(dst, src, 100);
+ assert(len === 0);
+ assert.equal(dst, Uint8Array.of(0, 0, 0, 0));
+
+ dst.fill(0);
+ src = Uint8Array.of(3, 4);
+ len = copyBytes(dst, src, -2);
+ assert(len === 2);
+ assert.equal(dst, Uint8Array.of(3, 4, 0, 0));
+});