You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
101 lines
3.5 KiB
101 lines
3.5 KiB
(*
|
|
* Copyright (c) 2016 - present Facebook, Inc.
|
|
* All rights reserved.
|
|
*
|
|
* This source code is licensed under the BSD style license found in the
|
|
* LICENSE file in the root directory of this source tree. An additional grant
|
|
* of patent rights can be found in the PATENTS file in the same directory.
|
|
*)
|
|
|
|
open! IStd
|
|
open PolyVariantEqual
|
|
module L = Logging
|
|
|
|
type zip_library = {zip_filename: string; zip_channel: Zip.in_file Lazy.t; models: bool}
|
|
|
|
let get_cache_dir infer_cache zip_filename =
|
|
let basename = Filename.basename zip_filename in
|
|
let key = basename ^ Utils.string_crc_hex32 zip_filename in
|
|
Filename.concat infer_cache key
|
|
|
|
|
|
let load_from_cache serializer zip_path cache_dir zip_library =
|
|
let absolute_path = Filename.concat cache_dir zip_path in
|
|
let deserialize = Serialization.read_from_file serializer in
|
|
let extract to_path =
|
|
if Sys.file_exists to_path <> `Yes then (
|
|
Unix.mkdir_p (Filename.dirname to_path) ;
|
|
let lazy zip_channel = zip_library.zip_channel in
|
|
let entry = Zip.find_entry zip_channel zip_path in
|
|
Zip.copy_entry_to_file zip_channel entry to_path ) ;
|
|
DB.filename_from_string to_path
|
|
in
|
|
match deserialize (extract absolute_path) with
|
|
| Some data ->
|
|
Some data
|
|
| None ->
|
|
None
|
|
| exception Not_found ->
|
|
None
|
|
|
|
|
|
let load_from_zip serializer zip_path zip_library =
|
|
let lazy zip_channel = zip_library.zip_channel in
|
|
let deserialize = Serialization.read_from_string serializer in
|
|
match deserialize (Zip.read_entry zip_channel (Zip.find_entry zip_channel zip_path)) with
|
|
| Some data ->
|
|
Some data
|
|
| None ->
|
|
None
|
|
| exception Not_found ->
|
|
None
|
|
|
|
|
|
let load_data serializer path zip_library =
|
|
let zip_path = Filename.concat Config.default_in_zip_results_dir path in
|
|
match Config.infer_cache with
|
|
| None ->
|
|
load_from_zip serializer zip_path zip_library
|
|
| Some infer_cache ->
|
|
let cache_dir = get_cache_dir infer_cache zip_library.zip_filename in
|
|
load_from_cache serializer zip_path cache_dir zip_library
|
|
|
|
|
|
(** list of the zip files to search for specs files *)
|
|
let zip_libraries =
|
|
(* delay until load is called, to avoid stating/opening files at init time *)
|
|
lazy
|
|
(let mk_zip_lib models zip_filename =
|
|
{models; zip_filename; zip_channel= lazy (Zip.open_in zip_filename)}
|
|
in
|
|
let zip_libs =
|
|
if Config.use_jar_cache && Config.infer_cache <> None then []
|
|
else
|
|
(* Order matters, jar files should appear in the order in which they should be searched for
|
|
specs files. Config.specs_library is in reverse order of appearance on command line. *)
|
|
let add_zip zip_libs fname =
|
|
if Filename.check_suffix fname ".jar" then
|
|
(* fname is a zip of specs *)
|
|
mk_zip_lib false fname :: zip_libs
|
|
else (* fname is a dir of specs *)
|
|
zip_libs
|
|
in
|
|
List.fold ~f:add_zip ~init:[] Config.specs_library
|
|
in
|
|
if Config.biabduction && not Config.models_mode && Sys.file_exists Config.models_jar = `Yes
|
|
then mk_zip_lib true Config.models_jar :: zip_libs
|
|
else zip_libs)
|
|
|
|
|
|
(* Search path in the list of zip libraries and use a cache directory to save already
|
|
deserialized data *)
|
|
let load serializer path =
|
|
let rec loop = function
|
|
| [] ->
|
|
None
|
|
| zip_library :: other_libraries ->
|
|
let opt = load_data serializer path zip_library in
|
|
if Option.is_some opt then opt else loop other_libraries
|
|
in
|
|
loop (Lazy.force zip_libraries)
|