att

Author: mateusmoutinho

src/consts.h

@@ -1,5 +1,3 @@
-
-
 #ifndef CONSTS_H
 #define CONSTS_H
 
@@ -11,6 +9,7 @@ const char *HELP_MESSAGE = "Usage: ./program [options]\n"
                            "  --port, -p <port>                Set the server port\n"
                            "  --dynamic_lib, -d <path>         Path to the dynamic library\n"
                            "  --callback, -c <name>            Callback function name in the dynamic library\n"
+                           "  --starter_callback, -s <name>    Starter callback function name to initialize server\n"
                            "  --password, --pass <password>    Password for firmware routes\n"
                            "  --allow_read_dynamic_lib         Allow reading the dynamic library via firmware route\n"
                            "  --allow_exit                     Allow server exit via firmware route\n"
@@ -34,6 +33,11 @@ const char *CALLBACK_FLAGS[]={
     "c"
 };
 
+const char *STARTER_CALLBACK_FLAGS[]={
+    "starter_callback",
+    "s"
+};
+
 const char *HELP_FLAGS[]={
     "help",
     "h"
@@ -71,4 +75,4 @@ const char *WRITE_DYNAMIC_LIB = "/cweb_firmware/write_dynamic_lib";
 const char *EXIT_FIRMWARE = "/cweb_firmware/exit";
 
 const char *PASSWORD_ENTRIE ="password";
-#endif 
+#endif

src/globals.h

@@ -3,6 +3,7 @@
 
 char dynamic_lib_path[2000] = {0};
 const char *callback_name;
+const char *starter_callback_name;
 int global_argc;
 char **global_argv;
 char password_sha[100] = {0};
@@ -12,4 +13,4 @@ bool allow_update_firmware = false;
 const char *VERSION = "0.1.0";
 
 
-#endif 
+#endif

src/main.c

@@ -1,4 +1,3 @@
-
 #include "consts.h"
 
 #include "../dependencies/CWebStudioOne.c"
@@ -67,7 +66,7 @@ int main(int argc, char *argv[]){
         return 1;
     }
 
-
+    starter_callback_name = CArgvParse_get_flag(&args,STARTER_CALLBACK_FLAGS,FLAGS_SIZE,0);
 
 
     allow_read_dynamic_lib = CArgvParse_is_flags_present(&args, ALLOW_READ_DYNAMIC_LIB_FLAGS, FLAGS_SIZE);
@@ -87,7 +86,40 @@ int main(int argc, char *argv[]){
 
     }
 
-    CwebServer server = newCwebSever(port_num, main_sever);
+    CwebServer server;
+    if(starter_callback_name){
+        #ifdef _WIN32
+        HMODULE handler = LoadLibrary(dynamic_lib_path);
+        if(!handler){
+            printf("Error loading dynamic library for starter callback: %lu\n", GetLastError());
+            return 1;
+        }
+        CwebServer (*starter_callback)() = (CwebServer (*)())GetProcAddress(handler,starter_callback_name);
+        if(!starter_callback){
+            printf("Starter callback function not found\n");
+            FreeLibrary(handler);
+            return 1;
+        }
+        server = starter_callback();
+        FreeLibrary(handler);
+        #else
+        void *handler = dlopen(dynamic_lib_path, RTLD_LAZY);
+        if(!handler){
+            printf("Error loading dynamic library for starter callback: %s\n", dlerror());
+            return 1;
+        }
+        CwebServer (*starter_callback)() = (CwebServer (*)())dlsym(handler,starter_callback_name);
+        if(!starter_callback){
+            printf("Starter callback function not found\n");
+            dlclose(handler);
+            return 1;
+        }
+        server = starter_callback();
+        dlclose(handler);
+        #endif
+    } else {
+        server = newCwebSever(port_num, main_sever);
+    }
     CwebServer_start(&server);
     return 0;
 }