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
104
105
106
107
108
109
type t = C.Types.Thread.t Ctypes.ptr
let self () =
Ctypes.addr (C.Functions.Thread.self ())
let equal =
C.Functions.Thread.equal
let thread_trampoline =
C.Functions.Thread.get_trampoline ()
let make_thread_options stack_size =
let module O = C.Types.Thread.Options in
let options = Ctypes.make O.t in
begin match stack_size with
| None ->
Ctypes.setf options O.flags O.no_flags
| Some n ->
Ctypes.setf options O.flags O.has_stack_size;
Ctypes.setf options O.stack_size (Unsigned.Size_t.of_int n)
end;
Ctypes.addr options
let create ?stack_size f =
let thread = Ctypes.addr (Ctypes.make C.Types.Thread.t) in
let f = Error.catch_exceptions f in
let f_gc_root = Ctypes.Root.create f in
let result =
C.Functions.Thread.create
thread
(make_thread_options stack_size)
thread_trampoline
f_gc_root
in
if result < 0 then begin
Ctypes.Root.release f_gc_root;
Error.result_from_c result
end
else
Ok thread
let create_c ?stack_size ?(argument = Nativeint.zero) f =
let thread = Ctypes.addr (Ctypes.make C.Types.Thread.t) in
C.Functions.Thread.create_c thread (make_thread_options stack_size) f argument
|> Error.to_result thread
let join thread =
C.Blocking.Thread.join thread
|> Error.to_result ()
module Priority =
struct
type t = [
| `HIGHEST
| `ABOVE_NORMAL
| `NORMAL
| `BELOW_NORMAL
| `LOWEST
]
end
let setpriority thread priority =
let priority =
match priority with
| `HIGHEST -> C.Types.Thread.Priority.highest
| `ABOVE_NORMAL -> C.Types.Thread.Priority.above_normal
| `NORMAL -> C.Types.Thread.Priority.normal
| `BELOW_NORMAL -> C.Types.Thread.Priority.below_normal
| `LOWEST -> C.Types.Thread.Priority.lowest
in
C.Functions.Thread.setpriority (Ctypes.(!@) thread) priority
|> Error.to_result ()
let getpriority thread =
let priority = Ctypes.(allocate int) 0 in
C.Functions.Thread.getpriority (Ctypes.(!@) thread) priority
|> Error.to_result_f (fun () -> Ctypes.(!@) priority)
let c mask =
mask
|> Bytes.unsafe_to_string
|> Ctypes.CArray.of_string
|> Ctypes.CArray.start
let setaffinity thread cpu_mask =
let mask_size = Bytes.length cpu_mask in
let old_mask = Bytes.create mask_size in
let mask_size = Unsigned.Size_t.of_int mask_size in
C.Functions.Thread.setaffinity thread (c cpu_mask) (c old_mask) mask_size
|> Error.to_result old_mask
let getaffinity thread =
match System_info.cpumask_size () with
| Error _ as error ->
error
| Ok mask_size ->
let cpu_mask = Bytes.create mask_size in
let mask_size = Unsigned.Size_t.of_int mask_size in
C.Functions.Thread.getaffinity thread (c cpu_mask) mask_size
|> Error.to_result cpu_mask
let getcpu () =
let cpu = C.Functions.Thread.getcpu () in
Error.to_result cpu cpu