1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
(* virt-builder
* Copyright (C) 2015 Red Hat Inc.
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License along
* with this program; if not, write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*)
open Std_utils
open Tools_utils
open Common_gettext.Gettext
open Printf
type csum_t =
| SHA1 of string
| SHA256 of string
| SHA512 of string
type csum_result =
| Good_checksum
| Mismatched_checksum of csum_t * string
| Missing_file
let string_of_csum_t = function
| SHA1 _ -> "sha1"
| SHA256 _ -> "sha256"
| SHA512 _ -> "sha512"
let string_of_csum = function
| SHA1 c -> c
| SHA256 c -> c
| SHA512 c -> c
let of_string csum_type csum_value =
match String.lowercase_ascii csum_type with
| "sha1" -> SHA1 csum_value
| "sha256" -> SHA256 csum_value
| "sha512" -> SHA512 csum_value
| _ -> invalid_arg csum_type
let compute_checksum csum_type ?tar filename =
let prog =
match csum_type with
| "sha1" -> "sha1sum"
| "sha256" -> "sha256sum"
| "sha512" -> "sha512sum"
| _ -> error (f_"unhandled checksum type ‘%s’") csum_type
in
let cmd =
match tar with
| None ->
sprintf "%s %s" prog (quote filename)
| Some tar ->
sprintf "tar xOf %s %s | %s"
(quote tar) (quote filename) prog
in
let lines = external_command cmd in
match lines with
| [] ->
error (f_"%s did not return any output") prog
| line :: _ ->
let csum_str = fst (String.split " " line) in
of_string csum_type csum_str
(* Check if the direct file exists or if it exists in the tarball. *)
let file_exists ?tar filename =
match tar with
| None -> Sys.file_exists filename
| Some tar ->
let cmd =
sprintf "tar tf %s %s >/dev/null 2>&1" (quote tar) (quote filename) in
Sys.command cmd = 0
let verify_checksum csum ?tar filename =
if not (file_exists ?tar filename) then
Missing_file
else (
let csum_type = string_of_csum_t csum in
let csum_actual = compute_checksum csum_type ?tar filename in
if csum = csum_actual then
Good_checksum
else
Mismatched_checksum (csum, string_of_csum csum_actual)
)
let verify_checksums checksums filename =
List.fold_left (
fun acc c ->
match acc with
| Good_checksum -> verify_checksum c filename
| (Mismatched_checksum _|Missing_file) as acc -> acc
) Good_checksum checksums
|
