From 566e9363be3033be80badf58963da75e83c0d053 Mon Sep 17 00:00:00 2001
From: revsic <revsic99@gmail.com>
Date: Sun, 7 Apr 2019 01:14:07 +0900
Subject: [PATCH] obfs, rand: Add sample for random generator

---
 obfuscator.hpp             | 2 +-
 obfuscator/obfs/random.hpp | 2 +-
 sample/CMakeLists.txt      | 1 +
 sample/random.cpp          | 8 ++++++++
 4 files changed, 11 insertions(+), 2 deletions(-)
 create mode 100644 sample/random.cpp

diff --git a/obfuscator.hpp b/obfuscator.hpp
index 83d2500..23a3dbe 100644
--- a/obfuscator.hpp
+++ b/obfuscator.hpp
@@ -7,7 +7,7 @@
 #define COMPILE_TIME_SEQUENCE
 #define OBFS_FINITE_STATE_MACHINE
 #define COMPILE_TIME_RANDOM
-#define MAKE_RAND_VAL(MOD) RAND_VAL<__LINE__, MOD>
+#define MAKE_RAND_VAL(MOD) obfs::RAND_VAL<__LINE__, MOD>
 #define OBFS_STRING
 #define MAKE_STRING(Var, String, ...) constexpr auto Var = obfs::make_string<__VA_ARGS__>(String);
 
diff --git a/obfuscator/obfs/random.hpp b/obfuscator/obfs/random.hpp
index f5b9ace..d93eed3 100644
--- a/obfuscator/obfs/random.hpp
+++ b/obfuscator/obfs/random.hpp
@@ -42,6 +42,6 @@ namespace obfs {
     constexpr size_t RAND_VAL = Xorshiftplus<SEED, Idx>::value % Mod;
 }
 
-#define MAKE_RAND_VAL(MOD) RAND_VAL<__LINE__, MOD>
+#define MAKE_RAND_VAL(MOD) obfs::RAND_VAL<__LINE__, MOD>
 
 #endif
\ No newline at end of file
diff --git a/sample/CMakeLists.txt b/sample/CMakeLists.txt
index 30d79ff..a4a1cc6 100644
--- a/sample/CMakeLists.txt
+++ b/sample/CMakeLists.txt
@@ -4,3 +4,4 @@ project(sample)
 set(CMAKE_CXX_STANDARD 17)
 add_executable(string_obfs string_obfs.cpp)
 add_executable(state_machine state_machine.cpp)
+add_executable(random random.cpp)
diff --git a/sample/random.cpp b/sample/random.cpp
new file mode 100644
index 0000000..8237086
--- /dev/null
+++ b/sample/random.cpp
@@ -0,0 +1,8 @@
+#include "../obfuscator.hpp"
+
+#include <iostream>
+
+int main() {
+    std::cout << MAKE_RAND_VAL(100) << std::endl;
+    return 0;
+}
\ No newline at end of file