shithub: util

Info  •  Files  •  Log  •  Branches

read_stdin

--- a/llama2.c

+++ b/llama2.c

@@ -1151,13 +1151,13 @@

 }

 void read_stdin(const char* guide, char* buffer, long bufsize) {

+	int r;

     // read a line from stdin, up to but not including \n

     print("%s", guide);

-    if (read(0, buffer, bufsize) > 0) {

-        long len = strlen(buffer);

-        if (len > 0 && buffer[len - 1] == '\n') {

-            buffer[len - 1] = '\0'; // strip newline

-        }

+	buffer[0] = '\0';

+    if ((r = read(0, buffer, bufsize)) > 0) {

+		buffer[r] = '\0';

+       	buffer[strcspn(buffer, "\r\n")] = '\0'; // strip newline

     }

 }

-- 

⑨