Merge pull request #16 from jestabro/strip-version

T7246: strip version strings on config load

1 files changed, 9 insertions, 1 deletions

diff --git a/src/startup.ml b/src/startup.ml
index db2784d..76599b0 100644
--- a/src/startup.ml
+++ b/src/startup.ml
@@ -78,12 +78,20 @@ let create_server accept_connection sock =
         Lwt_unix.accept sock >>= accept_connection >>= serve
     in serve
 
+(* strip commponent version string *)
+let strip_version s =
+    let rex = Pcre.regexp ~flags:[`MULTILINE;`DOTALL] "(^//.*)" in
+    let res = Pcre.split ~max:0 ~rex s in
+    match res with
+    | h :: _ -> h
+    | [] -> panic "Failure applying regex to config string"
+
 (** Load the appliance configuration file *)
 let load_config file =
     try
         let chan = open_in file in
         let s = really_input_string chan (in_channel_length chan) in
-        let config = Vyos1x.Parser.from_string s in
+        let config = strip_version s |> Vyos1x.Parser.from_string in
         Ok config
     with
         | Sys_error msg -> Error msg