]> matita.cs.unibo.it Git - helm.git/blob - helm/software/matita/dist/static_link/static_link.ml
0.5.3
[helm.git] / helm / software / matita / dist / static_link / static_link.ml
1
2 open Printf
3
4 exception Found of string list
5
6 let ocamlobjinfo = "ocamlobjinfo"
7 let noautolink = "-noautolink"
8 let dummy_opt_cmd = "dummy_ocamlopt"
9 let opt_cmd = "ocamlopt"
10 let libdirs = [ "/lib"; "/usr/lib"; "/usr/lib/gcc/i486-linux-gnu/4.0.2" ]
11 let exceptions = [ "threads.cma", [ "-lthreads", "-lthreadsnat" ] ]
12
13 let blanks_RE = Str.regexp "[ \t\r\n]+"
14 let cmxa_RE = Str.regexp "\\.cmxa$"
15 let extra_cfiles_RE = Str.regexp "^.*Extra +C +object +files:\\(.*\\)$"
16 let extra_copts_RE = Str.regexp "^.*Extra +C +options:\\(.*\\)$"
17 let lib_RE = Str.regexp "^lib"
18 let l_RE = Str.regexp "^-l"
19 let opt_line_RE = Str.regexp (sprintf "^\\+ +%s +\\(.*\\)$" dummy_opt_cmd)
20 let trailing_cmxa_RE = Str.regexp ".*\\.cmxa$"
21
22 let message s = prerr_endline ("STATIC_LINK: " ^ s)
23 let warning s = message ("WARNING: " ^ s)
24
25 let handle_exceptions ~cma cflag =
26   try
27     let cma_exns = List.assoc (Filename.basename cma) exceptions in
28     let cflag' = List.assoc cflag cma_exns in
29     message (sprintf "using %s exception %s -> %s" cma cflag cflag');
30     cflag'
31   with Not_found -> cflag
32
33 let parse_cmdline () =
34   let mine, rest = ref [], ref [] in
35   let is_mine = ref true in
36   Array.iter
37     (function
38       | "--" -> is_mine := false
39       | s when !is_mine ->
40           if Str.string_match lib_RE s 0 then
41             warning (sprintf
42               ("libraries to be statically linked must be specified "
43               ^^ "without heading \"lib\", \"%s\" argument may be wrong") s);
44           mine := s :: !mine
45       | s -> rest := s :: !rest)
46     Sys.argv;
47   if !rest = [] then begin
48     prerr_endline "Usage:   static_link [ CLIB .. ] -- COMMAND [ ARG .. ]";
49     prerr_endline ("Example: static_link pcre expat --"
50       ^ " ocamlfind opt -package pcre,expat -linkpkg -o foo foo.ml");
51     exit 0
52   end;
53   List.tl (List.rev !mine), List.rev !rest
54
55 let extract_opt_flags cmd =
56   let ic = Unix.open_process_in cmd in
57   (try
58     while true do
59       let l = input_line ic in
60       if Str.string_match opt_line_RE l 0 then begin
61         message ("got ocamlopt line: " ^ l);
62         raise (Found (Str.split blanks_RE (Str.matched_group 1 l)));
63       end
64     done;
65     []  (* dummy value *)
66   with
67   | End_of_file -> failwith "compiler command not found"
68   | Found flags ->
69       close_in ic;
70       flags)
71
72 let cma_of_cmxa = Str.replace_first cmxa_RE ".cma"
73
74 let find_clib libname =
75   let rec aux =
76     function
77     | [] -> raise Not_found
78     | libdir :: tl ->
79         let fname = sprintf "%s/lib%s.a" libdir libname in
80         if Sys.file_exists fname then fname else aux tl
81   in
82   aux libdirs
83
84 let a_of_cflag cflag =  (* "-lfoo" -> "/usr/lib/libfoo.a" *)
85   let libname = Str.replace_first l_RE "" cflag in
86   find_clib libname
87
88 let cflags_of_cma fname =
89   let ic = Unix.open_process_in (sprintf "%s %s" ocamlobjinfo fname) in
90   let extra_copts = ref "" in
91   let extra_cfiles = ref "" in
92   (try
93     while true do
94       match input_line ic with
95       | s when Str.string_match extra_copts_RE s 0 ->
96           extra_copts := Str.matched_group 1 s
97       | s when Str.string_match extra_cfiles_RE s 0 ->
98           extra_cfiles := Str.matched_group 1 s
99       | _ -> ()
100     done
101   with End_of_file -> ());
102   close_in ic;
103   let extra_cfiles = List.rev (Str.split blanks_RE !extra_cfiles) in
104   let extra_copts = Str.split blanks_RE !extra_copts in
105   extra_copts @ extra_cfiles
106
107 let staticize static_libs flags =
108   let static_flags = List.map ((^) "-l") static_libs in
109   let aux ~add_cclib ~cma cflag =
110     let cflag =
111       if List.mem cflag static_flags
112       then
113         (try
114           let a = a_of_cflag cflag in
115           message (sprintf "using static %s instead of shared %s" a cflag);
116           a
117         with Not_found -> warning ("can't find lib for " ^ cflag); cflag)
118       else (handle_exceptions ~cma cflag)
119     in
120     if add_cclib then [ "-cclib"; cflag ] else [ cflag ]
121   in
122   List.fold_right
123     (fun flag acc ->
124       let cma = cma_of_cmxa flag in
125       if Str.string_match trailing_cmxa_RE flag 0 then begin
126         message ("processing native archive: " ^ flag);
127         let cflags = cflags_of_cma cma in
128         let cflags' =
129           List.fold_right
130             (fun cflag acc -> (aux ~add_cclib:true ~cma cflag) @ acc)
131             cflags []
132         in
133         flag :: (cflags' @ acc)
134       end else
135         (aux ~add_cclib:false ~cma flag) @ acc)
136     flags []
137
138 let quote_if_needed s =
139   try
140     ignore (Str.search_forward blanks_RE s 0);
141     "\"" ^ s ^ "\""
142   with Not_found -> s
143
144 let main () =
145   let static_libs, args = parse_cmdline () in
146   printf "C libraries to be linked-in: %s\n" (String.concat " " static_libs);
147   flush stdout;
148   let verbose_cmd =
149     sprintf "OCAMLFIND_COMMANDS='ocamlopt=%s' %s -verbose 2>&1" dummy_opt_cmd
150       (String.concat " " (List.map quote_if_needed args))
151   in
152   let orig_opt_flags = extract_opt_flags verbose_cmd in
153   message ("original ocamlopt flags: " ^ String.concat " " orig_opt_flags);
154   let opt_flags = staticize static_libs orig_opt_flags in
155   message ("new ocamlopt flags: " ^ String.concat " " opt_flags);
156   let flags = noautolink :: opt_flags in
157   let cmd = String.concat " " (opt_cmd :: flags) in
158   message ("executing command: " ^ cmd);
159   exit (Sys.command cmd)
160
161 let _ = main ()
162