[GR-54171] Flush wrapped stdout/stderr OutputStream after every write

PullRequest: graalpython/3337

Author: msimacek

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/modules/SignalModuleBuiltins.java

@@ -543,9 +543,10 @@ final class Signals {
     static final String[] SIGNAL_NAMES = new String[SIGMAX + 1];
 
     static {
-        for (String signal : new String[]{"ABRT", "ALRM", "BUS", "FPE", "HUP", "ILL", "INFO", "INT", "KILL", "LOST",
-                        "PIPE", "PWR", "QUIT", "SEGV", "SYS", "TERM", "TRAP", "TSTP", "TTIN", "TTOUT", "USR1", "USR2",
-                        "VTALRM", "WINCH", "CHLD"}) {
+        for (String signal : new String[]{"HUP", "INT", "BREAK", "QUIT", "ILL", "TRAP", "IOT", "ABRT", "EMT", "FPE",
+                        "KILL", "BUS", "SEGV", "SYS", "PIPE", "ALRM", "TERM", "USR1", "USR2", "CLD", "CHLD", "PWR",
+                        "IO", "URG", "WINCH", "POLL", "STOP", "TSTP", "CONT", "TTIN", "TTOU", "VTALRM", "PROF",
+                        "XCPU", "XFSZ", "RTMIN", "RTMAX", "INFO", "STKFLT"}) {
             try {
                 int number = new sun.misc.Signal(signal).getNumber();
                 if (number > SIGMAX) {

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/runtime/PosixResources.java

@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2018, 2023, Oracle and/or its affiliates. All rights reserved.
+ * Copyright (c) 2018, 2024, Oracle and/or its affiliates. All rights reserved.
  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
  *
  * The Universal Permissive License (UPL), Version 1.0
@@ -43,11 +43,13 @@
 import java.io.IOException;
 import java.io.InputStream;
 import java.io.OutputStream;
+import java.nio.ByteBuffer;
 import java.nio.channels.Channel;
 import java.nio.channels.Channels;
 import java.nio.channels.FileLock;
 import java.nio.channels.Pipe;
 import java.nio.channels.SeekableByteChannel;
+import java.nio.channels.WritableByteChannel;
 import java.util.ArrayList;
 import java.util.Collections;
 import java.util.HashMap;
@@ -154,6 +156,33 @@ public Process destroyForcibly() {
         }
     }
 
+    private static class FlushingWritableByteChannel implements WritableByteChannel {
+        private final OutputStream stream;
+        private final WritableByteChannel delegate;
+
+        private FlushingWritableByteChannel(OutputStream stream) {
+            this.stream = stream;
+            delegate = Channels.newChannel(stream);
+        }
+
+        @Override
+        public int write(ByteBuffer src) throws IOException {
+            int res = delegate.write(src);
+            stream.flush();
+            return res;
+        }
+
+        @Override
+        public boolean isOpen() {
+            return delegate.isOpen();
+        }
+
+        @Override
+        public void close() throws IOException {
+            delegate.close();
+        }
+    }
+
     private static class ChannelWrapper {
         Channel channel;
         int cnt;
@@ -184,7 +213,7 @@ void setNewChannel(InputStream inputStream) {
         }
 
         void setNewChannel(OutputStream outputStream) {
-            this.channel = Channels.newChannel(outputStream);
+            this.channel = new FlushingWritableByteChannel(outputStream);
             this.cnt = 1;
         }
     }