def test_lseek():
def char_should_be(c, should):
if c != should:
print "Wrong char: '%s' should be '%s'" % (c, should)
def entry_point(argv):
fd = os.open('/hi.txt', os.O_RDONLY, 0777)
char_should_be(os.read(fd, 1), "H")
new = os.lseek(fd, 3, os.SEEK_CUR)
if new != 4:
print "Wrong offset, %d should be 4" % new
char_should_be(os.read(fd, 1), "o")
new = os.lseek(fd, -3, os.SEEK_END)
if new != 11:
print "Wrong offset, %d should be 11" % new
char_should_be(os.read(fd, 1), "d")
new = os.lseek(fd, 7, os.SEEK_SET)
if new != 7:
print "Wrong offset, %d should be 7" % new
char_should_be(os.read(fd, 1), "w")
print "All ok!"
return 0
exe = compile(entry_point)
proc = SandboxedProcWithFiles([exe])
output, error = proc.communicate("")
assert output == "All ok!\n"
assert error == ""
def test_lib():
def entry_point(argv):
fd = os.open("/tmp/foobar", os.O_RDONLY, 0777)
assert fd == 77
res = os.read(fd, 123)
assert res == "he\x00llo"
count = os.write(fd, "world\x00!\x00")
assert count == 42
for arg in argv:
count = os.write(fd, arg)
assert count == 61
os.close(fd)
return 0
exe = compile(entry_point)
proc = MockSandboxedProc([exe, 'x1', 'y2'], expected = [
("open", ("/tmp/foobar", os.O_RDONLY, 0777), 77),
("read", (77, 123), "he\x00llo"),
("write", (77, "world\x00!\x00"), 42),
("write", (77, exe), 61),
("write", (77, "x1"), 61),
("write", (77, "y2"), 61),
("close", (77,), None),
])
proc.handle_forever()
assert proc.seen == len(proc.expected)
def test_fstat():
def compare(a, b, i):
if a != b:
print "stat and fstat differ @%d: %s != %s" % (i, a, b)
def entry_point(argv):
try:
# Open a file, and compare stat and fstat
fd = os.open('/hi.txt', os.O_RDONLY, 0777)
st = os.stat('/hi.txt')
fs = os.fstat(fd)
# RPython requires the index for stat to be a constant.. :(
compare(st[0], fs[0], 0)
compare(st[1], fs[1], 1)
compare(st[2], fs[2], 2)
compare(st[3], fs[3], 3)
compare(st[4], fs[4], 4)
compare(st[5], fs[5], 5)
compare(st[6], fs[6], 6)
compare(st[7], fs[7], 7)
compare(st[8], fs[8], 8)
compare(st[9], fs[9], 9)
except OSError as e:
print "OSError: %s" % (e.errno,)
print "All ok!"
return 0
exe = compile(entry_point)
proc = SandboxedProcWithFiles([exe])
output, error = proc.communicate("")
assert output == "All ok!\n"
assert error == ""
def test_fstat():
def compare(a, b, i):
if a != b:
print "stat and fstat differ @%d: %s != %s" % (i, a, b)
def entry_point(argv):
try:
# Open a file, and compare stat and fstat
fd = os.open('/hi.txt', os.O_RDONLY, 0777)
st = os.stat('/hi.txt')
fs = os.fstat(fd)
# RPython requires the index for stat to be a constant.. :(
compare(st[0], fs[0], 0)
compare(st[1], fs[1], 1)
compare(st[2], fs[2], 2)
compare(st[3], fs[3], 3)
compare(st[4], fs[4], 4)
compare(st[5], fs[5], 5)
compare(st[6], fs[6], 6)
compare(st[7], fs[7], 7)
compare(st[8], fs[8], 8)
compare(st[9], fs[9], 9)
except OSError as e:
print "OSError: %s" % (e.errno,)
print "All ok!"
return 0
exe = compile(entry_point)
proc = SandboxedProcWithFiles([exe])
output, error = proc.communicate("")
assert output == "All ok!\n"
assert error == ""
def test_lib():
def entry_point(argv):
fd = os.open("/tmp/foobar", os.O_RDONLY, 0777)
assert fd == 77
res = os.read(fd, 123)
assert res == "he\x00llo"
count = os.write(fd, "world\x00!\x00")
assert count == 42
for arg in argv:
count = os.write(fd, arg)
assert count == 61
os.close(fd)
return 0
exe = compile(entry_point)
proc = MockSandboxedProc([exe, 'x1', 'y2'],
expected=[
("open", ("/tmp/foobar", os.O_RDONLY, 0777),
77),
("read", (77, 123), "he\x00llo"),
("write", (77, "world\x00!\x00"), 42),
("write", (77, exe), 61),
("write", (77, "x1"), 61),
("write", (77, "y2"), 61),
("close", (77, ), None),
])
proc.handle_forever()
assert proc.seen == len(proc.expected)
def test_lseek():
def char_should_be(c, should):
if c != should:
print "Wrong char: '%s' should be '%s'" % (c, should)
def entry_point(argv):
fd = os.open('/hi.txt', os.O_RDONLY, 0777)
char_should_be(os.read(fd, 1), "H")
new = os.lseek(fd, 3, os.SEEK_CUR)
if new != 4:
print "Wrong offset, %d should be 4" % new
char_should_be(os.read(fd, 1), "o")
new = os.lseek(fd, -3, os.SEEK_END)
if new != 11:
print "Wrong offset, %d should be 11" % new
char_should_be(os.read(fd, 1), "d")
new = os.lseek(fd, 7, os.SEEK_SET)
if new != 7:
print "Wrong offset, %d should be 7" % new
char_should_be(os.read(fd, 1), "w")
print "All ok!"
return 0
exe = compile(entry_point)
proc = SandboxedProcWithFiles([exe])
output, error = proc.communicate("")
assert output == "All ok!\n"
assert error == ""
def test_getuid():
def entry_point(argv):
import os
print "uid is %s" % os.getuid()
print "euid is %s" % os.geteuid()
print "gid is %s" % os.getgid()
print "egid is %s" % os.getegid()
return 0
exe = compile(entry_point)
proc = SandboxedProcWithFiles([exe])
output, error = proc.communicate("")
assert output == "uid is 1000\neuid is 1000\ngid is 1000\negid is 1000\n"
assert error == ""
def test_foobar():
py.test.skip("to be updated")
foobar = rffi.llexternal("foobar", [rffi.CCHARP], rffi.LONG)
def entry_point(argv):
s = rffi.str2charp(argv[1]); n = foobar(s); rffi.free_charp(s)
s = rffi.str2charp(argv[n]); n = foobar(s); rffi.free_charp(s)
return n
exe = compile(entry_point)
proc = MockSandboxedProc([exe, 'spam', 'egg'], expected = [
("foobar", ("spam",), 2),
("foobar", ("egg",), 0),
])
proc.handle_forever()
assert proc.seen == len(proc.expected)
def test_oserror():
def entry_point(argv):
try:
os.open("/tmp/foobar", os.O_RDONLY, 0777)
except OSError as e:
os.close(e.errno) # nonsense, just to see outside
return 0
exe = compile(entry_point)
proc = MockSandboxedProc([exe], expected = [
("open", ("/tmp/foobar", os.O_RDONLY, 0777), OSError(-42, "baz")),
("close", (-42,), None),
])
proc.handle_forever()
assert proc.seen == len(proc.expected)
def test_socketio():
class SocketProc(VirtualizedSocketProc, SimpleIOSandboxedProc):
def build_virtual_root(self):
pass
def entry_point(argv):
fd = os.open("tcp://python.org:80", os.O_RDONLY, 0777)
os.write(fd, 'GET /\n')
print os.read(fd, 50)
return 0
exe = compile(entry_point)
proc = SocketProc([exe])
output, error = proc.communicate("")
assert output.startswith('HTTP/1.0 400 Bad request')
def test_oserror():
def entry_point(argv):
try:
os.open("/tmp/foobar", os.O_RDONLY, 0777)
except OSError as e:
os.close(e.errno) # nonsense, just to see outside
return 0
exe = compile(entry_point)
proc = MockSandboxedProc([exe], expected = [
("open", ("/tmp/foobar", os.O_RDONLY, 0777), OSError(-42, "baz")),
("close", (-42,), None),
])
proc.handle_forever()
assert proc.seen == len(proc.expected)
def test_foobar():
py.test.skip("to be updated")
foobar = rffi.llexternal("foobar", [rffi.CCHARP], rffi.LONG)
def entry_point(argv):
s = rffi.str2charp(argv[1]); n = foobar(s); rffi.free_charp(s)
s = rffi.str2charp(argv[n]); n = foobar(s); rffi.free_charp(s)
return n
exe = compile(entry_point)
proc = MockSandboxedProc([exe, 'spam', 'egg'], expected = [
("foobar", ("spam",), 2),
("foobar", ("egg",), 0),
])
proc.handle_forever()
assert proc.seen == len(proc.expected)
def test_socketio():
class SocketProc(VirtualizedSocketProc, SimpleIOSandboxedProc):
def build_virtual_root(self):
pass
def entry_point(argv):
fd = os.open("tcp://python.org:80", os.O_RDONLY, 0777)
os.write(fd, 'GET /\n')
print os.read(fd, 50)
return 0
exe = compile(entry_point)
proc = SocketProc([exe])
output, error = proc.communicate("")
assert output.startswith('HTTP/1.1 301 Moved Permanently')
def test_getuid():
def entry_point(argv):
import os
print "uid is %s" % os.getuid()
print "euid is %s" % os.geteuid()
print "gid is %s" % os.getgid()
print "egid is %s" % os.getegid()
return 0
exe = compile(entry_point)
proc = SandboxedProcWithFiles([exe])
output, error = proc.communicate("")
assert output == "uid is 1000\neuid is 1000\ngid is 1000\negid is 1000\n"
assert error == ""
def test_getuid():
if not hasattr(os, 'getuid'):
py.test.skip("posix only")
def entry_point(argv):
import os
print "uid is %s" % os.getuid()
print "euid is %s" % os.geteuid()
print "gid is %s" % os.getgid()
print "egid is %s" % os.getegid()
return 0
exe = compile(entry_point)
proc = SandboxedProcWithFiles([exe])
output, error = proc.communicate("")
assert output == "uid is 1000\neuid is 1000\ngid is 1000\negid is 1000\n"
assert error == ""
def test_getuid():
if not hasattr(os, 'getuid'):
py.test.skip("posix only")
def entry_point(argv):
import os
print "uid is %s" % os.getuid()
print "euid is %s" % os.geteuid()
print "gid is %s" % os.getgid()
print "egid is %s" % os.getegid()
return 0
exe = compile(entry_point)
proc = SandboxedProcWithFiles([exe])
output, error = proc.communicate("")
assert output == "uid is 1000\neuid is 1000\ngid is 1000\negid is 1000\n"
assert error == ""
def test_simpleio():
def entry_point(argv):
print "Please enter a number:"
buf = ""
while True:
t = os.read(0, 1) # 1 character from stdin
if not t:
raise EOFError
if t == '\n':
break
buf += t
num = int(buf)
print "The double is:", num * 2
return 0
exe = compile(entry_point)
proc = SimpleIOSandboxedProc([exe, 'x1', 'y2'])
output, error = proc.communicate("21\n")
assert output == "Please enter a number:\nThe double is: 42\n"
assert error == ""
def test_simpleio():
def entry_point(argv):
print "Please enter a number:"
buf = ""
while True:
t = os.read(0, 1) # 1 character from stdin
if not t:
raise EOFError
if t == '\n':
break
buf += t
num = int(buf)
print "The double is:", num * 2
return 0
exe = compile(entry_point)
proc = SimpleIOSandboxedProc([exe, 'x1', 'y2'])
output, error = proc.communicate("21\n")
assert output == "Please enter a number:\nThe double is: 42\n"
assert error == ""
def test_too_many_opens():
def entry_point(argv):
try:
open_files = []
for i in range(500):
fd = os.open('/hi.txt', os.O_RDONLY, 0777)
open_files.append(fd)
txt = os.read(fd, 100)
if txt != "Hello, world!\n":
print "Wrong content: %s" % txt
except OSError as e:
# We expect to get EMFILE, for opening too many files.
if e.errno != errno.EMFILE:
print "OSError: %s!" % (e.errno, )
else:
print "We opened 500 fake files! Shouldn't have been able to."
for fd in open_files:
os.close(fd)
try:
open_files = []
for i in range(500):
fd = os.open('/this.pyc', os.O_RDONLY, 0777)
open_files.append(fd)
except OSError as e:
# We expect to get EMFILE, for opening too many files.
if e.errno != errno.EMFILE:
print "OSError: %s!" % (e.errno, )
else:
print "We opened 500 real files! Shouldn't have been able to."
print "All ok!"
return 0
exe = compile(entry_point)
proc = SandboxedProcWithFiles([exe])
output, error = proc.communicate("")
assert output == "All ok!\n"
assert error == ""
def test_too_many_opens():
def entry_point(argv):
try:
open_files = []
for i in range(500):
fd = os.open('/hi.txt', os.O_RDONLY, 0777)
open_files.append(fd)
txt = os.read(fd, 100)
if txt != "Hello, world!\n":
print "Wrong content: %s" % txt
except OSError as e:
# We expect to get EMFILE, for opening too many files.
if e.errno != errno.EMFILE:
print "OSError: %s!" % (e.errno,)
else:
print "We opened 500 fake files! Shouldn't have been able to."
for fd in open_files:
os.close(fd)
try:
open_files = []
for i in range(500):
fd = os.open('/this.pyc', os.O_RDONLY, 0777)
open_files.append(fd)
except OSError as e:
# We expect to get EMFILE, for opening too many files.
if e.errno != errno.EMFILE:
print "OSError: %s!" % (e.errno,)
else:
print "We opened 500 real files! Shouldn't have been able to."
print "All ok!"
return 0
exe = compile(entry_point)
proc = SandboxedProcWithFiles([exe])
output, error = proc.communicate("")
assert output == "All ok!\n"
assert error == ""
return 0
exe = compile(entry_point)
proc = SocketProc([exe])
output, error = proc.communicate("")
assert output.startswith('HTTP/1.1 301 Moved Permanently')
@supported
def test_oserror():
def entry_point(argv):
try:
os.open("/tmp/foobar", os.O_RDONLY, 0777)
except OSError, e:
os.close(e.errno) # nonsense, just to see outside
return 0
exe = compile(entry_point)
proc = MockSandboxedProc([exe], expected = [
("open", ("/tmp/foobar", os.O_RDONLY, 0777), OSError(-42, "baz")),
("close", (-42,), None),
])
proc.handle_forever()
assert proc.seen == len(proc.expected)
@supported
class SandboxedProcWithFiles(VirtualizedSandboxedProc, SimpleIOSandboxedProc):
"""A sandboxed process with a simple virtualized filesystem.
For testing file operations.
proc = SocketProc([exe])
output, error = proc.communicate("")
assert output.startswith('HTTP/1.1 301 Moved Permanently')
@supported
def test_oserror():
def entry_point(argv):
try:
os.open("/tmp/foobar", os.O_RDONLY, 0777)
except OSError, e:
os.close(e.errno) # nonsense, just to see outside
return 0
exe = compile(entry_point)
proc = MockSandboxedProc([exe],
expected=[
("open", ("/tmp/foobar", os.O_RDONLY, 0777),
OSError(-42, "baz")),
("close", (-42, ), None),
])
proc.handle_forever()
assert proc.seen == len(proc.expected)
@supported
class SandboxedProcWithFiles(VirtualizedSandboxedProc, SimpleIOSandboxedProc):
"""A sandboxed process with a simple virtualized filesystem.
