From e317b000a77ecd4b9d9086840c41802362e716ad Mon Sep 17 00:00:00 2001
From: Javier Sagredo <javier.sagredo@iohk.io>
Date: Tue, 16 Jun 2026 23:06:21 +0200
Subject: [PATCH] More

---
 serve.pl | 24 +++++++++++++++++-------
 sula.pl  |  3 ++-
 2 files changed, 19 insertions(+), 8 deletions(-)

diff --git a/serve.pl b/serve.pl
index 87f5329..bb6df47 100644
--- a/serve.pl
+++ b/serve.pl
@@ -1,4 +1,4 @@
-:- module(serve, [serve/3]).
+:- module(serve, [serve/4]).
 
 :- use_module(library(dcgs)).
 :- use_module(config).
@@ -6,21 +6,22 @@
 :- use_module(library(iso_ext)).
 :- use_module(library(lists)).
 :- use_module(library(pio)).
+:- use_module(library(charsio)).
 :- use_module(library(sockets)).
+:- use_module(library(debug)).
 :- use_module(log).
 :- use_module(mime).
 :- use_module(response).
 
-%% serve(+Stream, +Path, +Query)
+%% serve(+Stream, +Root, +Path, -Query)
 %
 % Serve the file at Path to Stream
-serve(S, /, Q) :-
-    serve(S, '/index.gmi', Q).
-serve(S, Path, _) :-
+serve(S, Root, /, Q) :-
+    serve(S, Root, '/index.gmi', Q).
+serve(S, Root, Path, _) :-
     atom_chars(Path, Chars),
     (  is_absolute(Chars)
     -> guess_mime(Chars, Mime),
-       content(Root),
        append(Root, Chars, File),
        (  file_exists(File)
        -> log_msg("response", "File does exist~n", []),
@@ -58,10 +59,19 @@ serve_binary(S, Mime, File) :-
             phrase(response(success, Mime), Response0),
             format(S, "~s", [Response0]),
             open(stream(S), write, _, [type(binary)]),
-            catch(copy_stream(FileStream, S),
+            catch(src_dest(FileStream, S),
                   error(existence_error(stream, _), _),
                   log_msg("response", "Client disconnected mid-stream~n", [])),
             log_msg("response", "Sent binary response~n", [])
         ),
         close(FileStream)
       ).
+
+src_dest(FileStream, DestStream) :-
+    get_n_chars(FileStream, 4096, Chars),
+    ( Chars = [] -> true
+    ; phrase_to_stream(seq(Chars), DestStream),
+      false).
+
+src_dest(FileStream, DestStream) :-
+    src_dest(FileStream, DestStream).
diff --git a/sula.pl b/sula.pl
index 7ce3d32..086416f 100755
--- a/sula.pl
+++ b/sula.pl
@@ -118,5 +118,6 @@ handle_conn_error(Error) :-
 req_serve(S, ClientCert) :-
     read_request(S, Path, Query),
     save_client(ClientCert),
-    serve(S, Path, Query),
+    content(Root),
+    serve(S, Root, Path, Query),
     !.
-- 
2.54.0