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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
|
package com.seaofnodes.simple;
import com.seaofnodes.simple.codegen.CodeGen;
import com.seaofnodes.simple.codegen.RegAllocTestSupport.CheckedCodeGen;
import com.seaofnodes.simple.type.TypeInteger;
import com.seaofnodes.simple.util.Ary;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.concurrent.TimeUnit;
import org.junit.Test;
import static org.junit.Assert.*;
public class StreamIOTest extends TestC {
@Test public void testReadUntilEOF() throws Exception {
String src="""
val main={ ->
val bytes=sys.io.stdin();
if(!bytes) return 1;
return sys.io.write(1,bytes)==bytes# ? 0 : 2;
};
""";
String exe=compile(src,"streamEcho",new Ary<>(new String[]{RELEASE_SYS_DIR}),
CALL_CONVENTION,null,null,-1);
for(int size:new int[]{0,8192,8193,65537}) {
String input="a\0bc\n".repeat(size/5)+"z".repeat(size%5);
Path in=Files.createTempFile(Path.of("build/objs"),"stream-in-",".txt");
Path out=Files.createTempFile(Path.of("build/objs"),"stream-out-",".txt");
Process child=null;
try {
Files.writeString(in,input);
ProcessBuilder builder=new ProcessBuilder(exe).redirectOutput(out.toFile());
if(size!=8193) builder.redirectInput(in.toFile());
child=builder.start();
if(size==8193) { // Also exercise a real pipe, ending only when its writer closes.
child.getOutputStream().write(input.getBytes(java.nio.charset.StandardCharsets.UTF_8));
child.getOutputStream().close();
}
assertTrue(child.waitFor(TEST_TIMEOUT_SECONDS,TimeUnit.SECONDS));
assertEquals(0,child.exitValue());
assertEquals(input,Files.readString(out));
} finally {
if(child!=null) child.destroyForcibly();
Files.deleteIfExists(in);
Files.deleteIfExists(out);
}
}
}
@Test public void testUnsignedArrayGrowth() throws Exception {
runSYS("""
val main={ ->
val b=new sys.aryu8();
val first=b.buf;
b.add('a');
if(b.buf!=first || b.len!=1) return 1;
b.addAry("bcdefghij");
if(b.len!=10 || b.buf[0]!='a' || b.buf[9]!='j') return 2;
b.undo(4294967295);
if(b.len!=0) return 3;
b.len=2147483648;
b.undo(1);
if(b.len!=2147483647) return 4;
b.clear();
val nums=new sys.aryi64(new i64[0]);
nums.add(10).add(20);
if(nums.len!=2 || nums.get(0)!=10 || nums.get(1)!=20) return 5;
nums.len=4294967295;
if(nums.len!=4294967295) return 6;
nums.len=2;
if(sys.ary.capacity(0,1)!=1) return 7;
if(sys.ary.capacity(8,8)!=8 || sys.ary.capacity(8,9)!=16) return 8;
if(sys.ary.capacity(2147483648,2147483649)!=4294967295) return 9;
if(sys.ary.capacity(4294967295,4294967296)!=0) return 10;
if(sys.ary.cmp("abc",4294967295,"",0,4294967295)!=0) return 11;
nums.str().add('\\n').write(1);
return 0;
};
""","unsignedArrayGrowth","[10, 20]\n",-1);
}
@Test public void testShortInterruptedAndFailedIO() throws Exception {
String src="""
{i32 i64 -> i32} test_io_mode="C";
{i64} test_io_count="C";
{i32} test_io_status="C";
val main={ ->
test_io_mode(1,0);
val bytes=sys.io.read(700);
if(!bytes) return 1;
if(bytes#!=24593) return 2;
for(u32 i=0; i<bytes#; i++) if(bytes[i]!=i-(i/251)*251) return 3;
test_io_mode(1,0);
if(sys.io.write(701,bytes)!=bytes# || test_io_count()!=bytes#) return 4;
if(test_io_status()) return 5;
test_io_mode(1,3);
if(sys.io.writeOff(701,bytes,3,5)!=5 || test_io_count()!=5) return 6;
if(test_io_status()) return 7;
test_io_mode(1,24590);
if(sys.io.writeOff(701,bytes,24590,4294967295)!=3) return 8;
if(test_io_status()) return 9;
test_io_mode(1,0);
if(sys.io.writeOff(701,bytes,24593,0)!=0 || test_io_count()!=0) return 10;
if(sys.io.writeOff(701,bytes,4294967295,1)!=-1) return 11;
if(sys.libc.errno()!=sys.libc.EINVAL || test_io_count()!=0) return 12;
test_io_mode(2,0);
if(sys.io.read(700)!=null || sys.libc.errno()!=sys.libc.EIO) return 13;
test_io_mode(3,0);
if(sys.io.write(701,bytes)!=-1 || sys.libc.errno()!=sys.libc.EIO) return 14;
if(test_io_count()!=7) return 15;
test_io_mode(4,0);
if(sys.io.write(701,bytes)!=-1 || sys.libc.errno()!=sys.libc.EIO) return 16;
test_io_mode(5,0);
if(sys.io.write(701,bytes)!=-1 || test_io_status()) return 17;
if(sys.io.read(-1)!=null) return 18;
val b=new sys.aryu8().addAry(bytes);
test_io_mode(1,0);
if(b.write(701)!=b.len || test_io_count()!=b.len || test_io_status()) return 19;
sys.io.p("ok\\n");
return 0;
};
""";
CodeGen code=new CheckedCodeGen(null,"build/objs",new Ary<>(new String[]{RELEASE_SYS_DIR}),
"streamFaults",src,126,true,TypeInteger.BOT);
code.driver(CPU_PORT,CALL_CONVENTION,false,true);
String exe="build/objs/streamFaults"+(OS.startsWith("Windows") ? ".exe" : "");
Ary<String> gcc=new Ary<>(String.class);
gcc.add("gcc");
gcc.add(OS.startsWith("Windows") ? "-Wl,--image-base,0x400000" : "-no-pie");
gcc.add("-Wl,--wrap=read,--wrap=write");
gcc.add(C_DRIVERS_DIR+"stream_io.c");
gcc.addAll(linkObjs(code));
gcc.add(runtimeObject("simple_crt"));
gcc.add(runtimeObject("simple_args"));
gcc.addAll(new String[]{"-lm","-o",exe});
exec(TEST_TIMEOUT_SECONDS,gcc.asAry());
assertEquals("ok\n",exec(TEST_TIMEOUT_SECONDS,exe));
}
}
|
