initial commit

third_party/libhv/unittest/sha1_test.c

@@ -0,0 +1,64 @@
+/*
+ * @build: gcc -o bin/sha1 unittest/sha1_test.c util/sha1.c -I. -Iutil
+ *
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <assert.h>
+
+#include "sha1.h"
+
+static void test() {
+    unsigned char ch = '1';
+    char sha1[41] = {0};
+    hv_sha1_hex(&ch, 1, sha1, sizeof(sha1));
+    assert(strcmp(sha1, "356a192b7913b04c54574d18c28d46e6395428ab") == 0);
+}
+
+int main(int argc, char* argv[]) {
+    test();
+
+    if (argc < 2) {
+        printf("Usage: sha1 file\n");
+        printf("       sha1 -s string\n");
+        return -10;
+    }
+
+    char sha1[41] = {0};
+
+    if (argc == 2) {
+        const char* filepath = argv[1];
+        FILE* fp = fopen(filepath, "rb");
+        if (fp == NULL) {
+            printf("Open file '%s' failed!\n", filepath);
+            return -20;
+        }
+        fseek(fp, 0, SEEK_END);
+        long filesize = ftell(fp);
+        // printf("filesize=%ld\n", filesize);
+        fseek(fp, 0, SEEK_SET);
+        unsigned char* filebuf = (unsigned char*)malloc(filesize);
+        size_t nread = fread(filebuf, 1, filesize, fp);
+        assert(nread == filesize);
+        hv_sha1_hex(filebuf, filesize, sha1, sizeof(sha1));
+
+        free(filebuf);
+        fclose(fp);
+    }
+    else if (argc == 3) {
+        const char* flags = argv[1];
+        if (flags[0] == '-' && flags[1] == 's') {
+            hv_sha1_hex((unsigned char*)argv[2], strlen(argv[2]), sha1, sizeof(sha1));
+        }
+        else {
+            printf("Unrecognized flags '%s'\n", flags);
+            return -40;
+        }
+    }
+
+    puts(sha1);
+
+    return 0;
+}