Source file `format_string.ml`

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
(**************************************************************************)
(*                                                                        *)
(*  This file is part of Frama-C.                                         *)
(*                                                                        *)
(*  Copyright (C) 2007-2023                                               *)
(*    CEA (Commissariat à l'énergie atomique et aux énergies              *)
(*         alternatives)                                                  *)
(*                                                                        *)
(*  you can redistribute it and/or modify it under the terms of the GNU   *)
(*  Lesser General Public License as published by the Free Software       *)
(*  Foundation, version 2.1.                                              *)
(*                                                                        *)
(*  It 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 Lesser General Public License for more details.                   *)
(*                                                                        *)
(*  See the GNU Lesser General Public License version 2.1                 *)
(*  for more details (enclosed in the file licenses/LGPLv2.1).            *)
(*                                                                        *)
(**************************************************************************)

type t =
  | String of string
  | WString of int64 list

exception OutOfBounds
exception NotAscii of int64

let get_char (s : t) (i : int) : char =
  match s with
  | String s ->
    begin try
        String.get s i
      with
        Invalid_argument _ -> raise OutOfBounds
    end
  | WString s ->
    begin try
        let c = List.nth s i in
        if (c >= Int64.zero && c<= (Int64.of_int 255)) then
          Char.chr (Int64.to_int c)
        else
          raise (NotAscii c)
      with
        Failure _ -> raise OutOfBounds
    end

let get_wchar (s : t) (i : int) : int64 =
  match s with
  | String s ->
    begin try
        Int64.of_int (Char.code (String.get s i))
      with
        Invalid_argument _ -> raise OutOfBounds
    end
  | WString s ->
    begin try
        List.nth s i
      with
        Failure _ -> raise OutOfBounds
    end

let sub_string (s : t) (start : int) (len : int) : string =
  let init_char i =
    get_char s (start + i)
  in
  String.init len init_char