Support src.cpp and src.c in the same rules dir

Author: lucic71

CMakeLists.txt

@@ -146,19 +146,20 @@ add_custom_target("check"
                   DEPENDS check-libcc2rs check-unit
                  )
 
-file(GLOB rule_src_files
-     ${PROJECT_SOURCE_DIR}/rules/*/src.cpp
-     ${PROJECT_SOURCE_DIR}/rules/*/src.c)
+file(GLOB rule_subdirs ${PROJECT_SOURCE_DIR}/rules/*)
 set(cpp_rules_ir_outputs)
 set(rust_rules_ir_outputs)
 set(rust_rules_inputs)
-foreach(_src IN LISTS rule_src_files)
-  get_filename_component(_rule_dir ${_src} DIRECTORY)
+foreach(_rule_dir IN LISTS rule_subdirs)
+  file(GLOB _srcs ${_rule_dir}/src.c ${_rule_dir}/src.cpp)
+  if(NOT _srcs)
+    continue()
+  endif()
   set(_out ${_rule_dir}/ir_src.json)
   add_custom_command(
     OUTPUT ${_out}
-    COMMAND $<TARGET_FILE:cpp-rule-preprocessor> --file ${_src}
-    DEPENDS ${_src} ${PROJECT_SOURCE_DIR}/cpp2rust/cpp_rule_preprocessor.cpp
+    COMMAND $<TARGET_FILE:cpp-rule-preprocessor> --dir ${_rule_dir}
+    DEPENDS ${_srcs} ${PROJECT_SOURCE_DIR}/cpp2rust/cpp_rule_preprocessor.cpp
     VERBATIM
   )
   list(APPEND cpp_rules_ir_outputs ${_out})

cpp2rust/cpp_rule_preprocessor.cpp

@@ -751,24 +751,40 @@ namespace {
 llvm::cl::OptionCategory cat("cpp-rule-preprocessor options");
 
 llvm::cl::opt<std::string>
-    SrcFile("file",
-            llvm::cl::desc("Path to a rule's src.c or src.cpp. ir_src.json is "
-                           "written next to it"),
-            llvm::cl::value_desc("src.c|src.cpp"), llvm::cl::Required,
-            llvm::cl::cat(cat));
+    SrcDir("dir",
+           llvm::cl::desc("Path to a rule directory containing src.c and/or "
+                          "src.cpp. ir_src.json is written into this dir."),
+           llvm::cl::value_desc("rule-dir"), llvm::cl::Required,
+           llvm::cl::cat(cat));
 
 } // namespace
 
 int main(int argc, char *argv[]) {
   llvm::cl::HideUnrelatedOptions(cat);
   llvm::cl::ParseCommandLineOptions(argc, argv);
 
-  fs::path src = SrcFile.getValue();
-  llvm::errs() << "Preprocessing " << src.string() << '\n';
+  fs::path dir = SrcDir.getValue();
   llvm::json::Object root;
-  cpp2rust::Extract(src, root);
+  for (const char *name : {"src.c", "src.cpp"}) {
+    auto path = dir / name;
+    if (!fs::exists(path)) {
+      continue;
+    }
+    llvm::errs() << "Preprocessing " << path.string() << '\n';
+    llvm::json::Object file_root;
+    cpp2rust::Extract(path, file_root);
+    for (auto &[k, v] : file_root) {
+      if (root.find(k) != root.end()) {
+        llvm::errs() << "ERROR: rule name " << k.str()
+                     << " defined in multiple files in " << dir.string()
+                     << '\n';
+        return EXIT_FAILURE;
+      }
+      root[k] = std::move(v);
+    }
+  }
 
-  auto out_path = src.parent_path() / "ir_src.json";
+  auto out_path = dir / "ir_src.json";
   std::error_code ec;
   llvm::raw_fd_ostream out(out_path.string(), ec);
   if (ec) {

rules/cstring/ir_unsafe.json

@@ -312,5 +312,81 @@
       "type": "*mut u8",
       "is_unsafe_pointer": true
     }
+  },
+  "f5": {
+    "body": [
+      {
+        "text": "libc::strchr("
+      },
+      {
+        "placeholder": {
+          "arg": 0,
+          "access": "read"
+        }
+      },
+      {
+        "text": " as *const i8, "
+      },
+      {
+        "placeholder": {
+          "arg": 1,
+          "access": "read"
+        }
+      },
+      {
+        "text": ") as *mut u8"
+      }
+    ],
+    "params": {
+      "a0": {
+        "type": "*const u8",
+        "is_unsafe_pointer": true
+      },
+      "a1": {
+        "type": "i32"
+      }
+    },
+    "return_type": {
+      "type": "*mut u8",
+      "is_unsafe_pointer": true
+    }
+  },
+  "f6": {
+    "body": [
+      {
+        "text": "libc::strchr("
+      },
+      {
+        "placeholder": {
+          "arg": 0,
+          "access": "read"
+        }
+      },
+      {
+        "text": " as *const i8, "
+      },
+      {
+        "placeholder": {
+          "arg": 1,
+          "access": "read"
+        }
+      },
+      {
+        "text": ") as *const u8"
+      }
+    ],
+    "params": {
+      "a0": {
+        "type": "*const u8",
+        "is_unsafe_pointer": true
+      },
+      "a1": {
+        "type": "i32"
+      }
+    },
+    "return_type": {
+      "type": "*const u8",
+      "is_unsafe_pointer": true
+    }
   }
 }

rules/cstring/src.c

@@ -10,3 +10,5 @@ void *f2(void *dst, int c, size_t n) { return memset(dst, c, n); }
 int f3(const void *s1, const void *s2, size_t n) { return memcmp(s1, s2, n); }
 
 void *f4(void *dst, const void *src, size_t n) { return memmove(dst, src, n); }
+
+char *f5(const char *a0, int a1) { return strchr(a0, a1); }

rules/cstring/src.cpp

@@ -0,0 +1,6 @@
+// Copyright (c) 2022-present INESC-ID.
+// Distributed under the MIT license that can be found in the LICENSE file.
+
+#include <cstring>
+
+const char *f6(const char *a0, int a1) { return strchr(a0, a1); }

rules/cstring/tgt_unsafe.rs

@@ -35,3 +35,11 @@ unsafe fn f4(a0: *mut u8, a1: *const u8, a2: usize) -> *mut u8 {
     }
     a0
 }
+
+unsafe fn f5(a0: *const u8, a1: i32) -> *mut u8 {
+    libc::strchr(a0 as *const i8, a1) as *mut u8
+}
+
+unsafe fn f6(a0: *const u8, a1: i32) -> *const u8 {
+    libc::strchr(a0 as *const i8, a1) as *const u8
+}