mmap加载配置文件，uring实现持久化

common/config.c

@@ -6,6 +6,12 @@
 #include <strings.h>      // strcasecmp
 #include <libxml/parser.h>
 #include <libxml/tree.h>
+#include <errno.h>
+#include <sys/mman.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <unistd.h>
 
 static xmlNodePtr find_child(xmlNodePtr parent, const char *name)
 {
@@ -87,6 +93,52 @@ static void parse_allocator(const char *s, AllocatorType *out)
     else *out = ALLOC_OTHER;
 }
 
+static int read_file_mmap(const char *filename, void **out_addr, size_t *out_len, int *out_fd) {
+    if (!filename || !out_addr || !out_len || !out_fd) return -1;
+
+    *out_addr = NULL;
+    *out_len = 0;
+    *out_fd = -1;
+
+    int fd = open(filename, O_RDONLY);
+    if (fd < 0) {
+        fprintf(stderr, "config_load: open(%s) failed: %s\n", filename, strerror(errno));
+        return -1;
+    }
+
+    struct stat st;
+    if (fstat(fd, &st) != 0) {
+        fprintf(stderr, "config_load: fstat(%s) failed: %s\n", filename, strerror(errno));
+        close(fd);
+        return -1;
+    }
+
+    if (!S_ISREG(st.st_mode)) {
+        fprintf(stderr, "config_load: %s is not a regular file\n", filename);
+        close(fd);
+        return -1;
+    }
+
+    if (st.st_size <= 0) {
+        fprintf(stderr, "config_load: %s is empty\n", filename);
+        close(fd);
+        return -1;
+    }
+
+    size_t len = (size_t)st.st_size;
+    void *addr = mmap(NULL, len, PROT_READ, MAP_PRIVATE, fd, 0);
+    if (addr == MAP_FAILED) {
+        fprintf(stderr, "config_load: mmap(%s) failed: %s\n", filename, strerror(errno));
+        close(fd);
+        return -1;
+    }
+
+    *out_addr = addr;
+    *out_len = len;
+    *out_fd = fd;
+    return 0;
+}
+
 /* ---- 对外的枚举转字符串工具 ---- */
 
 const char *log_level_to_string(LogLevel lvl)
@@ -129,27 +181,9 @@ const char *allocator_to_string(AllocatorType a)
 
 /* ---- 主函数：从 XML 加载配置 ---- */
 
-int config_load(const char *filename, AppConfig *out_cfg)
-{
-    if (!filename || !out_cfg) return -1;
-
-    set_default_config(out_cfg);
-
-    xmlDocPtr doc = xmlReadFile(filename, "UTF-8", XML_PARSE_NOBLANKS);
-    if (!doc) {
-        fprintf(stderr, "config_load: failed to read file %s\n", filename);
-        return -1;
-    }
-
-    xmlNodePtr root = xmlDocGetRootElement(doc);
-    if (!root || xmlStrcmp(root->name, BAD_CAST "config") != 0) {
-        fprintf(stderr, "config_load: root element is not <config>\n");
-        xmlFreeDoc(doc);
-        return -1;
-    }
-
-    /* server 部分 */
-    xmlNodePtr server = find_child(root, "server");
+/* server 部分 */
+void server_load(xmlNodePtr *root, AppConfig *out_cfg){
+    xmlNodePtr server = find_child(*root, "server");
     if (server) {
         /* ip */
         xmlNodePtr ip_node = find_child(server, "ip");
@@ -205,9 +239,11 @@ int config_load(const char *filename, AppConfig *out_cfg)
             }
         }
     }
+}
 
-    /* log 部分 */
-    xmlNodePtr log = find_child(root, "log");
+/* log 部分 */
+void log_load(xmlNodePtr *root, AppConfig *out_cfg){
+    xmlNodePtr log = find_child(*root, "log");
     if (log) {
         xmlNodePtr lvl_node = find_child(log, "level");
         if (lvl_node) {
@@ -218,9 +254,11 @@ int config_load(const char *filename, AppConfig *out_cfg)
             }
         }
     }
+}
 
-    /* persistence 部分 */
-    xmlNodePtr pers = find_child(root, "persistence");
+/* persistence 部分 */
+void persist_load(xmlNodePtr *root, AppConfig *out_cfg){
+    xmlNodePtr pers = find_child(*root, "persistence");
     if (pers) {
         xmlNodePtr type_node = find_child(pers, "type");
         if (type_node) {
@@ -281,9 +319,11 @@ int config_load(const char *filename, AppConfig *out_cfg)
             }
         }
     }
+}
 
-    /* memory 部分 */
-    xmlNodePtr mem = find_child(root, "memory");
+/* memory 部分 */
+void memory_load(xmlNodePtr *root, AppConfig *out_cfg){
+    xmlNodePtr mem = find_child(*root, "memory");
     if (mem) {
         xmlNodePtr alloc_node = find_child(mem, "allocator");
         if (alloc_node) {
@@ -294,9 +334,61 @@ int config_load(const char *filename, AppConfig *out_cfg)
             }
         }
     }
-
-    xmlFreeDoc(doc);
-    /* xmlCleanupParser() 建议在程序退出时 main 里统一调用 */
-
-    return 0;
+}
+
+int config_load(const char *filename, AppConfig *out_cfg)
+{
+    if (!filename || !out_cfg) return -1;
+
+    set_default_config(out_cfg);
+
+    int rc = -1;
+    xmlDocPtr doc = NULL;
+
+    int fd = -1;
+    void *addr = NULL;
+    size_t len = 0;
+
+    if (read_file_mmap(filename, &addr, &len, &fd) != 0) {
+        // read_file_mmap 已经打印错误
+        return -1;
+    }
+
+    /*
+     * 用 xmlReadMemory 从内存解析。
+     * - "UTF-8"：你原来指定了 UTF-8；如果希望自动探测，可以传 NULL。
+     * - XML_PARSE_NONET：禁用网络访问（防 XXE/外部实体拉取）
+     * - XML_PARSE_NOBLANKS：保持你原来的行为
+     * 你也可以加 XML_PARSE_NOERROR | XML_PARSE_NOWARNING 减少噪音，但调试阶段不建议。
+     */
+    int parse_opts = XML_PARSE_NOBLANKS | XML_PARSE_NONET;
+
+    
+    // xmlDocPtr doc = xmlReadFile(filename, "UTF-8", XML_PARSE_NOBLANKS);
+    doc = xmlReadMemory((const char *)addr, (int)len, filename, "UTF-8", parse_opts);
+    if (!doc) {
+        fprintf(stderr, "config_load: failed to read file %s\n", filename);
+        goto cleanup;
+    }
+
+    xmlNodePtr root = xmlDocGetRootElement(doc);
+    if (!root || xmlStrcmp(root->name, BAD_CAST "config") != 0) {
+        fprintf(stderr, "config_load: root element is not <config>\n");
+        goto cleanup;
+    }
+
+    server_load(&root, out_cfg);
+
+    log_load(&root, out_cfg);
+
+    persist_load(&root, out_cfg);
+
+    memory_load(&root, out_cfg);
+
+    rc = 0;
+cleanup:
+    if (doc) xmlFreeDoc(doc);
+    if (addr && len) munmap(addr, len);
+    if (fd >= 0) close(fd);
+    return rc;
 }