added uuid extension

Author: fabiosvm

extensions/CMakeLists.txt

@@ -86,6 +86,10 @@ add_library(crypto_mod SHARED
 add_library(regex_mod SHARED
   regex.c)
 
+add_library(uuid_mod SHARED
+  uuid.c
+  deps/uuid4.c)
+
 target_link_libraries(sqlite_mod    ${STATIC_LIB_TARGET})
 target_link_libraries(curl_mod      ${STATIC_LIB_TARGET})
 target_link_libraries(redis_mod     ${STATIC_LIB_TARGET})
@@ -97,6 +101,7 @@ target_link_libraries(zeromq_mod    ${STATIC_LIB_TARGET})
 target_link_libraries(leveldb_mod   ${STATIC_LIB_TARGET})
 target_link_libraries(crypto_mod    ${STATIC_LIB_TARGET})
 target_link_libraries(regex_mod     ${STATIC_LIB_TARGET})
+target_link_libraries(uuid_mod      ${STATIC_LIB_TARGET})
 
 set_target_properties(sqlite_mod    PROPERTIES PREFIX "")
 set_target_properties(curl_mod      PROPERTIES PREFIX "")
@@ -109,6 +114,7 @@ set_target_properties(zeromq_mod    PROPERTIES PREFIX "")
 set_target_properties(leveldb_mod   PROPERTIES PREFIX "")
 set_target_properties(crypto_mod    PROPERTIES PREFIX "")
 set_target_properties(regex_mod     PROPERTIES PREFIX "")
+set_target_properties(uuid_mod      PROPERTIES PREFIX "")
 
 if(MSVC)
   target_link_libraries(sqlite_mod  PRIVATE SQLite::SQLite3)

extensions/deps/uuid4.c

@@ -0,0 +1,90 @@
+/**
+ * Copyright (c) 2018 rxi
+ *
+ * This library is free software; you can redistribute it and/or modify it
+ * under the terms of the MIT license. See LICENSE for details.
+ */
+
+#include <stdio.h>
+#include <stdint.h>
+
+#if defined(_WIN32)
+#include <windows.h>
+#include <wincrypt.h>
+#endif
+
+#include "uuid4.h"
+
+
+static uint64_t seed[2];
+
+
+static uint64_t xorshift128plus(uint64_t *s) {
+  /* http://xorshift.di.unimi.it/xorshift128plus.c */
+  uint64_t s1 = s[0];
+  const uint64_t s0 = s[1];
+  s[0] = s0;
+  s1 ^= s1 << 23;
+  s[1] = s1 ^ s0 ^ (s1 >> 18) ^ (s0 >> 5);
+  return s[1] + s0;
+}
+
+
+int uuid4_init(void) {
+#if defined(__linux__) || defined(__APPLE__) || defined(__FreeBSD__)
+  int res;
+  FILE *fp = fopen("/dev/urandom", "rb");
+  if (!fp) {
+    return UUID4_EFAILURE;
+  }
+  res = fread(seed, 1, sizeof(seed), fp);
+  fclose(fp);
+  if ( res != sizeof(seed) ) {
+    return UUID4_EFAILURE;
+  }
+
+#elif defined(_WIN32)
+  int res;
+  HCRYPTPROV hCryptProv;
+  res = CryptAcquireContext(
+    &hCryptProv, NULL, NULL, PROV_RSA_FULL, CRYPT_VERIFYCONTEXT);
+  if (!res) {
+    return UUID4_EFAILURE;
+  }
+  res = CryptGenRandom(hCryptProv, (DWORD) sizeof(seed), (PBYTE) seed);
+  CryptReleaseContext(hCryptProv, 0);
+  if (!res) {
+    return UUID4_EFAILURE;
+  }
+
+#else
+  #error "unsupported platform"
+#endif
+  return UUID4_ESUCCESS;
+}
+
+
+void uuid4_generate(char *dst) {
+  static const char *template = "xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx";
+  static const char *chars = "0123456789abcdef";
+  union { unsigned char b[16]; uint64_t word[2]; } s;
+  const char *p;
+  int i, n;
+  /* get random */
+  s.word[0] = xorshift128plus(seed);
+  s.word[1] = xorshift128plus(seed);
+  /* build string */
+  p = template;
+  i = 0;
+  while (*p) {
+    n = s.b[i >> 1];
+    n = (i & 1) ? (n >> 4) : (n & 0xf);
+    switch (*p) {
+      case 'x'  : *dst = chars[n];              i++;  break;
+      case 'y'  : *dst = chars[(n & 0x3) + 8];  i++;  break;
+      default   : *dst = *p;
+    }
+    dst++, p++;
+  }
+  *dst = '\0';
+}

extensions/deps/uuid4.h

@@ -0,0 +1,22 @@
+/**
+ * Copyright (c) 2018 rxi
+ *
+ * This library is free software; you can redistribute it and/or modify it
+ * under the terms of the MIT license. See LICENSE for details.
+ */
+
+#ifndef UUID4_H
+#define UUID4_H
+
+#define UUID4_VERSION "1.0.0"
+#define UUID4_LEN 37
+
+enum {
+  UUID4_ESUCCESS =  0,
+  UUID4_EFAILURE = -1
+};
+
+int  uuid4_init(void);
+void uuid4_generate(char *dst);
+
+#endif

extensions/uuid.c

@@ -0,0 +1,31 @@
+//
+// The Hook Programming Language
+// uuid.c
+//
+
+#include "uuid.h"
+#include "deps/uuid4.h"
+
+static void random_call(HkState *state, HkValue *args);
+
+static void random_call(HkState *state, HkValue *args)
+{
+  (void) args;
+  HkString *str = hk_string_new_with_capacity(UUID4_LEN);
+  uuid4_init();
+  uuid4_generate(str->chars);
+  str->length = UUID4_LEN;
+  str->chars[str->length] = '\0';
+  hk_state_push_string(state, str);
+}
+
+HK_LOAD_MODULE_HANDLER(uuid)
+{
+  hk_state_push_string_from_chars(state, -1, "uuid");
+  hk_return_if_not_ok(state);
+  hk_state_push_string_from_chars(state, -1, "random");
+  hk_return_if_not_ok(state);
+  hk_state_push_new_native(state, "random", 0, random_call);
+  hk_return_if_not_ok(state);
+  hk_state_construct(state, 1);
+}

extensions/uuid.h

@@ -0,0 +1,13 @@
+//
+// The Hook Programming Language
+// uuid.h
+//
+
+#ifndef UUID_H
+#define UUID_H
+
+#include <hook.h>
+
+HK_LOAD_MODULE_HANDLER(uuid);
+
+#endif // UUID_H