···

       
303
       
 
       
    to_other_sink_as_well ~other:env#stdout

     

       
304
       
 
       
      (log :> Eio.Flow.sink_ty Eio.Flow.sink)

     

       
305
       
 
       
  in

     

       
306
       
-
       
  Eio.Process.spawn ~stdout ~sw env#proc ~cwd:eio_tmp cmd

     

       
307
       
 
       


     

       
308
       
 
       
(* 

     

       
309
       
 
       
                                 Apache License

     

···

       
303
       
 
       
    to_other_sink_as_well ~other:env#stdout

     

       
304
       
 
       
      (log :> Eio.Flow.sink_ty Eio.Flow.sink)

     

       
305
       
 
       
  in

     

       
306
       
+
       
  Eio.Process.spawn ~sw ~stdout env#proc ~cwd:eio_tmp cmd

     

       
307
       
 
       


     

       
308
       
 
       
(* 

     

       
309
       
 
       
                                 Apache License

     

···

       
252
       
 
       
  in

     

       
253
       
 
       
  let with_rootfs fn =

     

       
254
       
 
       
    if entry.pre.mode = R then (Store.Run.with_build ctx.store build fn, [])

     

       
255
       
-
       
    else Store.Run.with_clone ctx.store ~src:build new_cid fn

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
256
       
 
       
  in

     

       
257
       
 
       
  with_rootfs @@ function

     

       
258
       
 
       
  | `Exists path ->

     
···

       
300
       
 
       
                    "-c";

     

       
301
       
 
       
                    String.concat " " command ^ " && env > /tmp/shelter-env";

     

       
302
       
 
       
                  ];

     

       
303
       
-
       
                hostname = "";

     

       
304
       
 
       
                network = [ "host" ];

     

       
305
       
 
       
                user = (uid, gid);

     

       
306
       
 
       
                env = entry.pre.env;

     
···

       
317
       
 
       
          in

     

       
318
       
 
       
          `Runc (Runc.spawn ~sw log env config rootfs)

     

       
319
       
 
       
      in

     

       
320
       
-
       
      Switch.run @@ fun sw ->

     

       
321
       
-
       
      let log =

     

       
322
       
-
       
        Eio.Path.open_out ~sw ~create:(`If_missing 0o644)

     

       
323
       
-
       
          Eio.Path.(fs / rootfs / "log")

     

       
324
       
-
       
      in

     

       
325
       
-
       
      let res = spawn sw log in

     

       
326
       
-
       
      let start = Mtime_clock.now () in

     

       
327
       
-
       
      let res =

     

       
328
       
 
       
        match res with

     

       
329
       
-
       
        | `Runc r -> Eio.Process.await r

     

       
330
       
-
       
        | `Void v -> Void.to_eio_status (Eio.Promise.await v)

     

       
331
       
 
       
      in

     

       
332
       
 
       
      let stop = Mtime_clock.now () in

     

       
333
       
 
       
      let span = Mtime.span start stop in

     

···

       
252
       
 
       
  in

     

       
253
       
 
       
  let with_rootfs fn =

     

       
254
       
 
       
    if entry.pre.mode = R then (Store.Run.with_build ctx.store build fn, [])

     

       
255
       
+
       
    else

     

       
256
       
+
       
      let diff_path =

     

       
257
       
+
       
        Eio.Path.(fs / Filename.temp_dir "shelter-diff-" "" / "diff")

     

       
258
       
+
       
      in

     

       
259
       
+
       
      Store.Run.with_clone ctx.store ~src:build new_cid diff_path fn

     

       
260
       
 
       
  in

     

       
261
       
 
       
  with_rootfs @@ function

     

       
262
       
 
       
  | `Exists path ->

     
···

       
304
       
 
       
                    "-c";

     

       
305
       
 
       
                    String.concat " " command ^ " && env > /tmp/shelter-env";

     

       
306
       
 
       
                  ];

     

       
307
       
+
       
                hostname = "builder";

     

       
308
       
 
       
                network = [ "host" ];

     

       
309
       
 
       
                user = (uid, gid);

     

       
310
       
 
       
                env = entry.pre.env;

     
···

       
321
       
 
       
          in

     

       
322
       
 
       
          `Runc (Runc.spawn ~sw log env config rootfs)

     

       
323
       
 
       
      in

     

       
324
       
+
       
      let start, res =

     

       
325
       
+
       
        Switch.run @@ fun sw ->

     

       
326
       
+
       
        let log =

     

       
327
       
+
       
          Eio.Path.open_out ~sw ~create:(`If_missing 0o644)

     

       
328
       
+
       
            Eio.Path.(fs / rootfs / "log")

     

       
329
       
+
       
        in

     

       
330
       
+
       
        let res = spawn sw log in

     

       
331
       
+
       
        let start = Mtime_clock.now () in

     

       
332
       
 
       
        match res with

     

       
333
       
+
       
        | `Runc r -> (start, Eio.Process.await r)

     

       
334
       
+
       
        | `Void v -> (start, Void.to_eio_status (Eio.Promise.await v))

     

       
335
       
 
       
      in

     

       
336
       
 
       
      let stop = Mtime_clock.now () in

     

       
337
       
 
       
      let span = Mtime.span start stop in

     

···

       
94
       
 
       
  Zfs.clone src (tgt :> string)

     

       
95
       
 
       


     

       
96
       
 
       
let read_all fd =

     

       
97
       
-
       
  let buf = Buffer.create 128 in

     

       
98
       
-
       
  let bytes = Bytes.create 4096 in

     

       
99
       
 
       
  let rec loop () =

     

       
100
       
 
       
    match Unix.read fd bytes 0 4096 with

     

       
101
       
 
       
    | 0 | (exception End_of_file) -> Buffer.contents buf

     
···

       
105
       
 
       
  in

     

       
106
       
 
       
  loop ()

     

       
107
       
 
       


     

       
108
       
-
       
let diff t (data : Datasets.snapshot) (snap : Datasets.snapshot) =

     

       
109
       
 
       
  let data_fs =

     

       
110
       
 
       
    String.sub (data :> string) 0 (String.index (data :> string) '@')

     

       
111
       
 
       
  in

     

       
112
       
 
       
  let zh = Zfs.open_ t.zfs data_fs Zfs.Types.filesystem in

     

       
113
       
-
       
  let diff =

     

       
114
       
-
       
    let r, w = Unix.pipe ~cloexec:false () in

     

       
115
       
 
       
    try

     

       
116
       
-
       
      Zfs.show_diff zh ~from_:(data :> string) ~to_:(snap :> string) w;

     

       
117
       
-
       
      let f = read_all r in

     

       
118
       
-
       
      Unix.close r;

     

       
119
       
-
       
      f

     

       
120
       
-
       
    with e ->

     

       
121
       
-
       
      Unix.close r;

     

       
122
       
-
       
      raise e

     

       
0
       
 
       

     

       
123
       
 
       
  in

     

       
124
       
 
       
  Zfs.close zh;

     

       
0
       
 
       

     

       
125
       
 
       
  Diff.of_zfs diff

     

       
126
       
 
       


     

       
127
       
 
       
let cid s =

     
···

       
173
       
 
       
    mount_dataset t ds;

     

       
174
       
 
       
    fn ("/" ^ (ds :> string))

     

       
175
       
 
       


     

       
176
       
-
       
  let with_clone t ~src new_cid fn =

     

       
177
       
 
       
    let ds = Datasets.build t.pool (Cid.to_string src) in

     

       
178
       
 
       
    let tgt = Datasets.build t.pool (Cid.to_string new_cid) in

     

       
179
       
 
       
    let src_snap = Datasets.snapshot ds in

     

       
180
       
 
       
    let tgt_snap = Datasets.snapshot tgt in

     

       
181
       
 
       
    if Zfs.exists t.zfs (tgt :> string) Zfs.Types.dataset then

     

       
182
       
-
       
      (fn (`Exists ("/" ^ (tgt :> string))), diff t src_snap tgt_snap)

     

       
183
       
 
       
    else (

     

       
184
       
 
       
      clone t src_snap tgt;

     

       
185
       
 
       
      let v = with_build t new_cid fn in

     

       
186
       
 
       
      snapshot t tgt_snap;

     

       
187
       
-
       
      let d = diff t src_snap tgt_snap in

     

       
188
       
 
       
      (v, d))

     

       
189
       
 
       
end

     

···

       
94
       
 
       
  Zfs.clone src (tgt :> string)

     

       
95
       
 
       


     

       
96
       
 
       
let read_all fd =

     

       
97
       
+
       
  let buf = Buffer.create 10_000 in

     

       
98
       
+
       
  let bytes = Bytes.create 10_000 in

     

       
99
       
 
       
  let rec loop () =

     

       
100
       
 
       
    match Unix.read fd bytes 0 4096 with

     

       
101
       
 
       
    | 0 | (exception End_of_file) -> Buffer.contents buf

     
···

       
105
       
 
       
  in

     

       
106
       
 
       
  loop ()

     

       
107
       
 
       


     

       
108
       
+
       
let diff t (data : Datasets.snapshot) (snap : Datasets.snapshot) output =

     

       
109
       
 
       
  let data_fs =

     

       
110
       
 
       
    String.sub (data :> string) 0 (String.index (data :> string) '@')

     

       
111
       
 
       
  in

     

       
112
       
 
       
  let zh = Zfs.open_ t.zfs data_fs Zfs.Types.filesystem in

     

       
113
       
+
       
  let () =

     

       
0
       
 
       

     

       
114
       
 
       
    try

     

       
115
       
+
       
      Eio.Path.with_open_out ~create:(`If_missing 0o644) output

     

       
116
       
+
       
      @@ fun flow_fd ->

     

       
117
       
+
       
      let eio_fd = Eio_unix.Resource.fd_opt flow_fd in

     

       
118
       
+
       
      Eio_unix.Fd.use_exn_opt "zfs-diff" eio_fd @@ function

     

       
119
       
+
       
      | None -> Fmt.failwith "Output needs to have an FD"

     

       
120
       
+
       
      | Some fd ->

     

       
121
       
+
       
          Zfs.show_diff zh ~from_:(data :> string) ~to_:(snap :> string) fd

     

       
122
       
+
       
    with Unix.Unix_error (Unix.EBADF, _, _) -> ()

     

       
123
       
 
       
  in

     

       
124
       
 
       
  Zfs.close zh;

     

       
125
       
+
       
  let diff = Eio.Path.load output in

     

       
126
       
 
       
  Diff.of_zfs diff

     

       
127
       
 
       


     

       
128
       
 
       
let cid s =

     
···

       
174
       
 
       
    mount_dataset t ds;

     

       
175
       
 
       
    fn ("/" ^ (ds :> string))

     

       
176
       
 
       


     

       
177
       
+
       
  let with_clone t ~src new_cid output fn =

     

       
178
       
 
       
    let ds = Datasets.build t.pool (Cid.to_string src) in

     

       
179
       
 
       
    let tgt = Datasets.build t.pool (Cid.to_string new_cid) in

     

       
180
       
 
       
    let src_snap = Datasets.snapshot ds in

     

       
181
       
 
       
    let tgt_snap = Datasets.snapshot tgt in

     

       
182
       
 
       
    if Zfs.exists t.zfs (tgt :> string) Zfs.Types.dataset then

     

       
183
       
+
       
      (fn (`Exists ("/" ^ (tgt :> string))), diff t src_snap tgt_snap output)

     

       
184
       
 
       
    else (

     

       
185
       
 
       
      clone t src_snap tgt;

     

       
186
       
 
       
      let v = with_build t new_cid fn in

     

       
187
       
 
       
      snapshot t tgt_snap;

     

       
188
       
+
       
      let d = diff t src_snap tgt_snap output in

     

       
189
       
 
       
      (v, d))

     

       
190
       
 
       
end

     

···

       
137
       
 
       
  (* TODO: Other Diff Flags https://github.com/openzfs/zfs/blob/5b0c27cd14bbc07d50304c97735cc105d0258673/include/libzfs.h#L917? *)

     

       
138
       
 
       
  let res = C.Functions.diff handle (Obj.magic fd : int) from_ to_ 1 in

     

       
139
       
 
       
  if res = 0 then () else begin

     

       
140
       
-
       
    Format.printf "Diff got %i\n%!" res;

     

       
141
       
 
       
    invalid_arg "show_diff"

     

       
142
       
 
       
  end

     

···

       
137
       
 
       
  (* TODO: Other Diff Flags https://github.com/openzfs/zfs/blob/5b0c27cd14bbc07d50304c97735cc105d0258673/include/libzfs.h#L917? *)

     

       
138
       
 
       
  let res = C.Functions.diff handle (Obj.magic fd : int) from_ to_ 1 in

     

       
139
       
 
       
  if res = 0 then () else begin

     

       
0
       
 
       

     

       
140
       
 
       
    invalid_arg "show_diff"

     

       
141
       
 
       
  end