def test_wrap_oserror():
class FakeSpace:
w_OSError = [OSError]
w_EnvironmentError = [EnvironmentError]
def wrap(self, obj):
return [obj]
def call_function(self, exc, w_errno, w_msg, w_filename=None):
return (exc, w_errno, w_msg, w_filename)
space = FakeSpace()
#
e = wrap_oserror(space, OSError(errno.EBADF, "foobar"))
assert isinstance(e, OperationError)
assert e.w_type == [OSError]
assert e.get_w_value(space) == ([OSError], [errno.EBADF],
[os.strerror(errno.EBADF)], None)
#
e = wrap_oserror(space, OSError(errno.EBADF, "foobar"),
filename = "test.py",
exception_name = "w_EnvironmentError")
assert isinstance(e, OperationError)
assert e.w_type == [EnvironmentError]
assert e.get_w_value(space) == ([EnvironmentError], [errno.EBADF],
[os.strerror(errno.EBADF)],
["test.py"])
#
e = wrap_oserror(space, OSError(errno.EBADF, "foobar"),
filename = "test.py",
w_exception_class = [SystemError])
assert isinstance(e, OperationError)
assert e.w_type == [SystemError]
assert e.get_w_value(space) == ([SystemError], [errno.EBADF],
[os.strerror(errno.EBADF)],
["test.py"])
def execve(space, w_command, w_args, w_env):
""" execve(path, args, env)
Execute a path with arguments and environment, replacing current process.
path: path of executable file
args: iterable of arguments
env: dictionary of strings mapping to strings
"""
command = fsencode_w(space, w_command)
try:
args_w = space.unpackiterable(w_args)
if len(args_w) < 1:
raise oefmt(space.w_ValueError,
"execv() must have at least one argument")
args = [fsencode_w(space, w_arg) for w_arg in args_w]
except OperationError as e:
if not e.match(space, space.w_TypeError):
raise
raise oefmt(space.w_TypeError,
"execv() arg 2 must be an iterable of strings")
#
if w_env is None: # when called via execv() above
try:
os.execv(command, args)
except OSError as e:
raise wrap_oserror(space, e)
else:
env = _env2interp(space, w_env)
try:
os.execve(command, args, env)
except OSError as e:
raise wrap_oserror(space, e)
def descr_init(self, space, w_name, mode='r', closefd=True):
if space.isinstance_w(w_name, space.w_float):
raise oefmt(space.w_TypeError,
"integer argument expected, got float")
fd = -1
try:
fd = space.c_int_w(w_name)
except OperationError as e:
pass
else:
if fd < 0:
raise oefmt(space.w_ValueError, "negative file descriptor")
self.readable, self.writable, self.appending, flags = decode_mode(space, mode)
fd_is_own = False
try:
if fd >= 0:
try:
os.fstat(fd)
except OSError as e:
if e.errno == errno.EBADF:
raise wrap_oserror(space, e)
# else: pass
self.fd = fd
self.closefd = bool(closefd)
else:
self.closefd = True
if not closefd:
raise oefmt(space.w_ValueError,
"Cannot use closefd=False with file name")
from pypy.module.posix.interp_posix import (
dispatch_filename, rposix)
try:
self.fd = dispatch_filename(rposix.open)(
space, w_name, flags, 0666)
except OSError as e:
raise wrap_oserror2(space, e, w_name,
exception_name='w_IOError')
finally:
fd_is_own = True
self._dircheck(space, w_name)
space.setattr(self, space.wrap("name"), w_name)
if self.appending:
# For consistent behaviour, we explicitly seek to the end of file
# (otherwise, it might be done only on the first write()).
try:
os.lseek(self.fd, 0, os.SEEK_END)
except OSError as e:
raise wrap_oserror(space, e, exception_name='w_IOError')
except:
if not fd_is_own:
self.fd = -1
raise
def pipe(space):
"Create a pipe. Returns (read_end, write_end)."
try:
fd1, fd2 = os.pipe()
except OSError as e:
raise wrap_oserror(space, e)
return space.newtuple([space.wrap(fd1), space.wrap(fd2)])
def confstr(space, w_name):
num = confname_w(space, w_name, os.confstr_names)
try:
res = os.confstr(num)
except OSError as e:
raise wrap_oserror(space, e)
return space.wrap(res)
def isatty(space, fd):
"""Return True if 'fd' is an open file descriptor connected to the
slave end of a terminal."""
try:
res = os.isatty(fd)
except OSError, e:
raise wrap_oserror(space, e)
def readall_w(self, space):
self._check_closed(space)
self._check_readable(space)
total = 0
builder = StringBuilder()
while True:
newsize = int(new_buffersize(self.fd, total))
try:
chunk = os.read(self.fd, newsize - total)
except OSError, e:
if e.errno == errno.EINTR:
space.getexecutioncontext().checksignals()
continue
if total > 0:
# return what we've got so far
break
if e.errno == errno.EAGAIN:
return space.w_None
raise wrap_oserror(space, e,
exception_name='w_IOError')
if not chunk:
break
builder.append(chunk)
total += len(chunk)
def _dump_rpy_heap(space, fd):
try:
ok = rgc.dump_rpy_heap(fd)
except OSError as e:
raise wrap_oserror(space, e)
if not ok:
raise missing_operation(space)
def lchown(space, path, uid, gid):
"""Change the owner and group id of path to the numeric uid and gid.
This function will not follow symbolic links."""
try:
os.lchown(path, uid, gid)
except OSError as e:
raise wrap_oserror(space, e, path)
def readlink(space, path):
"Return a string representing the path to which the symbolic link points."
try:
result = os.readlink(path)
except OSError as e:
raise wrap_oserror(space, e, path)
return space.wrap(result)
def lchown(space, path, uid, gid):
check_uid_range(space, uid)
check_uid_range(space, gid)
try:
os.lchown(path, uid, gid)
except OSError, e:
raise wrap_oserror(space, e, path)
def pathconf(space, path, w_name):
num = confname_w(space, w_name, os.pathconf_names)
try:
res = os.pathconf(path, num)
except OSError as e:
raise wrap_oserror(space, e)
return space.wrap(res)
def enable(self, space, fileno, period_usec):
if self.is_enabled:
raise oefmt(space.w_ValueError, "_vmprof already enabled")
self.fileno = fileno
self.is_enabled = True
self.write_header(fileno, period_usec)
if not self.ever_enabled:
if we_are_translated():
res = pypy_vmprof_init()
if res:
raise OperationError(
space.w_IOError,
space.wrap(rffi.charp2str(vmprof_get_error())))
self.ever_enabled = True
self.gather_all_code_objs(space)
space.register_code_callback(vmprof_register_code)
if we_are_translated():
# does not work untranslated
res = vmprof_enable(fileno, period_usec, 0,
lltype.nullptr(rffi.CCHARP.TO), 0)
else:
res = 0
if res == -1:
raise wrap_oserror(space, OSError(rposix.get_saved_errno(),
"_vmprof.enable"))
def write(space, fd, data):
"""Write a string to a file descriptor. Return the number of bytes
actually written, which may be smaller than len(data)."""
try:
res = os.write(fd, data)
except OSError, e:
raise wrap_oserror(space, e)
def fstatvfs(space, fd):
try:
st = rposix_stat.fstatvfs(fd)
except OSError as e:
raise wrap_oserror(space, e)
else:
return build_statvfs_result(space, st)
def fsync(space, w_fd):
"""Force write of file with filedescriptor to disk."""
fd = space.c_filedescriptor_w(w_fd)
try:
os.fsync(fd)
except OSError, e:
raise wrap_oserror(space, e)
def spawnv(space, mode, path, w_args):
args = [space.str0_w(w_arg) for w_arg in space.unpackiterable(w_args)]
try:
ret = os.spawnv(mode, path, args)
except OSError as e:
raise wrap_oserror(space, e)
return space.wrap(ret)
def spawnve(space, mode, path, w_args, w_env):
args = [space.str0_w(w_arg) for w_arg in space.unpackiterable(w_args)]
env = _env2interp(space, w_env)
try:
ret = os.spawnve(mode, path, args, env)
except OSError, e:
raise wrap_oserror(space, e)
def fstat(space, fd):
"""Perform a stat system call on the file referenced to by an open
file descriptor."""
try:
st = os.fstat(fd)
except OSError, e:
raise wrap_oserror(space, e)
def open(space, fname, flag, mode=0777):
"""Open a file (for low level IO).
Return a file descriptor (a small integer)."""
try:
fd = os.open(fname, flag, mode)
except OSError, e:
raise wrap_oserror(space, e)
def openpty(space):
"Open a pseudo-terminal, returning open fd's for both master and slave end."
try:
master_fd, slave_fd = os.openpty()
except OSError as e:
raise wrap_oserror(space, e)
return space.newtuple([space.wrap(master_fd), space.wrap(slave_fd)])
def isatty_w(self, space):
self._check_closed(space)
try:
res = os.isatty(self.fd)
except OSError as e:
raise wrap_oserror(space, e, exception_name='w_IOError')
return space.wrap(res)
def getpid(space):
"Return the current process id."
try:
pid = os.getpid()
except OSError as e:
raise wrap_oserror(space, e)
return space.wrap(pid)
def nice(space, inc):
"Decrease the priority of process by inc and return the new priority."
try:
res = os.nice(inc)
except OSError as e:
raise wrap_oserror(space, e)
return space.wrap(res)
def _getfullpathname(space, path):
"""helper for ntpath.abspath """
posix = __import__(os.name) # nt specific
try:
fullpath = posix._getfullpathname(path)
except OSError, e:
raise wrap_oserror(space, e)
def dup(space, fd):
"""Create a copy of the file descriptor. Return the new file
descriptor."""
try:
newfd = os.dup(fd)
except OSError, e:
raise wrap_oserror(space, e)
def seek_w(self, space, pos, whence=0):
self._check_closed(space)
try:
pos = os.lseek(self.fd, pos, whence)
except OSError, e:
raise wrap_oserror(space, e,
exception_name='w_IOError')
def open_cdll(space, name):
try:
return CDLL(name)
except DLOpenError as e:
raise wrap_dlopenerror(space, e, name or "<None>")
except OSError as e:
raise wrap_oserror(space, e)
def tell_w(self, space):
self._check_closed(space)
try:
pos = os.lseek(self.fd, 0, 1)
except OSError, e:
raise wrap_oserror(space, e,
exception_name='w_IOError')
def unsetenv(space, name):
"""Delete an environment variable."""
try:
del os.environ[name]
except KeyError:
pass
except OSError, e:
raise wrap_oserror(space, e)
def putenv(space, name, value):
"""Change or add an environment variable."""
if _WIN32 and len(name) > _MAX_ENV:
raise OperationError(
space.w_ValueError,
space.wrap("the environment variable is longer than %d bytes" %
_MAX_ENV))
try:
os.environ[name] = value
except OSError, e:
raise wrap_oserror(space, e)
def listdir(space, dirname):
"""Return a list containing the names of the entries in the directory.
\tpath: path of directory to list
The list is in arbitrary order. It does not include the special
entries '.' and '..' even if they are present in the directory."""
try:
result = os.listdir(dirname)
except OSError, e:
raise wrap_oserror(space, e)
def prepare_file_argument(space, fileobj):
fileobj.direct_flush()
if fileobj.cffi_fileobj is None:
fd = fileobj.direct_fileno()
if fd < 0:
raise OperationError(space.w_ValueError,
space.wrap("file has no OS file descriptor"))
try:
fileobj.cffi_fileobj = CffiFileObj(fd, fileobj.mode)
except OSError, e:
raise wrap_oserror(space, e)
def readinto_w(self, space, w_buffer):
self._check_closed(space)
self._check_readable(space)
rwbuffer = space.rwbuffer_w(w_buffer)
length = rwbuffer.getlength()
try:
buf = os.read(self.fd, length)
except OSError, e:
if e.errno == errno.EAGAIN:
return space.w_None
raise wrap_oserror(space, e, exception_name='w_IOError')
def write_w(self, space, w_data):
self._check_closed(space)
self._check_writable(space)
data = space.getarg_w('y*', w_data).as_str()
try:
n = os.write(self.fd, data)
except OSError, e:
if e.errno == errno.EAGAIN:
return space.w_None
raise wrap_oserror(space, e, exception_name='w_IOError')
def setregid(space, rgid, egid):
""" setregid(rgid, egid)
Set the current process's real and effective group ids.
"""
check_uid_range(space, rgid)
check_uid_range(space, egid)
try:
os.setregid(rgid, egid)
except OSError, e:
raise wrap_oserror(space, e)
def uname(space):
""" uname() -> (sysname, nodename, release, version, machine)
Return a tuple identifying the current operating system.
"""
try:
r = os.uname()
except OSError as e:
raise wrap_oserror(space, e)
l_w = [space.newfilename(i) for i in [r[0], r[1], r[2], r[3], r[4]]]
return space.newtuple(l_w)
def mmap_error(space, e):
if isinstance(e, RValueError):
return OperationError(space.w_ValueError, space.wrap(e.message))
elif isinstance(e, RTypeError):
return OperationError(space.w_TypeError, space.wrap(e.message))
elif isinstance(e, OSError):
w_error = space.fromcache(Cache).w_error
return wrap_oserror(space, e, w_exception_class=w_error)
else:
# bogus 'e'?
return OperationError(space.w_SystemError, space.wrap('%s' % e))
def initgroups(space, username, gid):
""" initgroups(username, gid) -> None
Call the system initgroups() to initialize the group access list with all of
the groups of which the specified username is a member, plus the specified
group id.
"""
try:
os.initgroups(username, gid)
except OSError as e:
raise wrap_oserror(space, e)
def _close(self, space):
if self.fd < 0:
return
fd = self.fd
self.fd = -1
try:
verify_fd(fd)
os.close(fd)
except OSError, e:
raise wrap_oserror(space, e, exception_name='w_IOError')
def prepare_file_argument(space, w_fileobj):
w_fileobj.direct_flush()
if w_fileobj.cffi_fileobj is None:
fd = w_fileobj.direct_fileno()
if fd < 0:
raise oefmt(space.w_ValueError, "file has no OS file descriptor")
try:
w_fileobj.cffi_fileobj = CffiFileObj(fd, w_fileobj.mode)
except OSError as e:
raise wrap_oserror(space, e)
return rffi.cast(rffi.CCHARP, w_fileobj.cffi_fileobj.llf)
def get_osfhandle(space, fd):
"""get_osfhandle()
Return the handle corresponding to the file descriptor (windows only)
"""
from rpython.rlib import rwin32 # Windows only
try:
ret = rwin32.get_osfhandle(fd)
return space.newint(rffi.cast(rffi.INT, ret))
except OSError as e:
raise wrap_oserror(space, e)
def read_w(self, space, w_size=None):
self._check_closed(space)
self._check_readable(space)
size = convert_size(space, w_size)
if size < 0:
return self.readall_w(space)
try:
s = os.read(self.fd, size)
except OSError, e:
raise wrap_oserror(space, e, exception_name='w_IOError')
def descr_new(space, w_subtype, kind, value, maxvalue):
if kind != RECURSIVE_MUTEX and kind != SEMAPHORE:
raise OperationError(space.w_ValueError,
space.wrap("unrecognized kind"))
counter = space.fromcache(CounterState).getCount()
name = "/mp%d-%d" % (os.getpid(), counter)
try:
handle = create_semaphore(space, name, value, maxvalue)
except OSError, e:
raise wrap_oserror(space, e)
def read_w(self, space, w_size=None):
self._check_closed(space)
self._check_readable(space)
size = convert_size(space, w_size)
if size < 0:
return self.readall_w(space)
while True:
try:
s = os.read(self.fd, size)
break
except OSError as e:
if e.errno == errno.EAGAIN:
return space.w_None
wrap_oserror(space,
e,
exception_name='w_IOError',
eintr_retry=True)
return space.newbytes(s)
def truncate_w(self, space, w_size=None):
self._check_closed(space)
self._check_writable(space)
if space.is_none(w_size):
w_size = self.tell_w(space)
try:
self._truncate(space.r_longlong_w(w_size))
except OSError as e:
raise wrap_oserror(space, e, exception_name='w_IOError')
return w_size
def setgroups(space, w_list):
""" setgroups(list)
Set the groups of the current process to list.
"""
list = []
for w_gid in space.unpackiterable(w_list):
list.append(space.c_uid_t_w(w_gid))
try:
os.setgroups(list[:])
except OSError as e:
raise wrap_oserror(space, e)
def fork(space):
run_fork_hooks('before', space)
try:
pid = os.fork()
except OSError, e:
try:
run_fork_hooks('parent', space)
except:
# Don't clobber the OSError if the fork failed
pass
raise wrap_oserror(space, e)
def getresgid(space):
""" getresgid() -> (rgid, egid, sgid)
Get tuple of the current process's real, effective, and saved group ids.
"""
try:
(rgid, egid, sgid) = os.getresgid()
except OSError as e:
raise wrap_oserror(space, e)
return space.newtuple([wrap_gid(space, rgid),
wrap_gid(space, egid),
wrap_gid(space, sgid)])
def mmap(space,
fileno,
length,
flags=rmmap.MAP_SHARED,
prot=rmmap.PROT_WRITE | rmmap.PROT_READ,
access=rmmap._ACCESS_DEFAULT):
try:
return space.wrap(
W_MMap(space, rmmap.mmap(fileno, length, flags, prot, access)))
except OSError, e:
raise wrap_oserror(space, e, 'w_EnvironmentError')
def _close(self, space):
if self.fd < 0:
return
fd = self.fd
self.fd = -1
try:
os.close(fd)
except OSError as e:
raise wrap_oserror(space, e,
w_exception_class=space.w_IOError,
eintr_retry=False)
def truncate_w(self, space, w_size=None):
self._check_closed(space)
self._check_writable(space)
if space.is_none(w_size):
w_size = self.tell_w(space)
try:
self._truncate(space.r_longlong_w(w_size))
except OSError as e:
raise wrap_oserror(space, e, w_exception_class=space.w_IOError,
eintr_retry=False)
return w_size
def utime(space, path, w_tuple):
""" utime(path, (atime, mtime))
utime(path, None)
Set the access and modified time of the file to the given values. If the
second form is used, set the access and modified times to the current time.
"""
if space.is_w(w_tuple, space.w_None):
try:
os.utime(path, None)
return
except OSError, e:
raise wrap_oserror(space, e)
def modify(self, space, w_fd, events):
"""
Modify an already registered file descriptor.
fd -- either an integer, or an object with a fileno() method returning an
int.
events -- an optional bitmask describing the type of events to check for
"""
fd = space.c_filedescriptor_w(w_fd)
if fd not in self.fddict:
raise wrap_oserror(space,
OSError(errno.ENOENT, "poll.modify"),
w_exception_class=space.w_IOError)
self.fddict[fd] = events
def test_wrap_oserror():
class FakeSpace:
w_OSError = [OSError]
w_EnvironmentError = [EnvironmentError]
w_None = None
def wrap(self, obj, lgt=-1):
return [obj]
newint = newtext = newfilename = wrap
def call_function(self, exc, w_errno, w_msg, w_filename=None, *args):
return (exc, w_errno, w_msg, w_filename)
space = FakeSpace()
#
e = wrap_oserror(space, OSError(errno.EBADF, "foobar"))
assert isinstance(e, OperationError)
assert e.w_type == [OSError]
assert e.get_w_value(space) == ([OSError], [errno.EBADF],
[os.strerror(errno.EBADF)], None)
#
e = wrap_oserror(space,
OSError(errno.EBADF, "foobar"),
filename="test.py",
w_exception_class=space.w_EnvironmentError)
assert isinstance(e, OperationError)
assert e.w_type == [EnvironmentError]
assert e.get_w_value(space) == ([EnvironmentError], [errno.EBADF],
[os.strerror(errno.EBADF)], ["test.py"])
#
e = wrap_oserror(space,
OSError(errno.EBADF, "foobar"),
filename="test.py",
w_exception_class=[SystemError])
assert isinstance(e, OperationError)
assert e.w_type == [SystemError]
assert e.get_w_value(space) == ([SystemError], [errno.EBADF],
[os.strerror(errno.EBADF)], ["test.py"])
def write_w(self, space, w_data):
self._check_closed(space)
self._check_writable(space)
data = space.getarg_w('s*', w_data).as_str()
try:
n = os.write(self.fd, data)
except OSError as e:
if e.errno == errno.EAGAIN:
return space.w_None
raise wrap_oserror(space, e, w_exception_class=space.w_IOError)
return space.newint(n)
def __init__(self, space, name, mode):
self.flags = libffi.FUNCFLAG_CDECL
self.space = space
if name is None:
self.name = "<None>"
else:
self.name = name
try:
self.cdll = libffi.CDLL(name, mode)
except DLOpenError as e:
raise wrap_dlopenerror(space, e, self.name)
except OSError as e:
raise wrap_oserror(space, e)
def _sendall(self, space, message, size):
while size > 0:
# XXX inefficient
data = rffi.charpsize2str(message, size)
try:
count = self.WRITE(data)
except OSError, e:
if e.errno == EINTR:
space.getexecutioncontext().checksignals()
continue
raise wrap_oserror(space, e)
size -= count
message = rffi.ptradd(message, count)
def readinto_w(self, space, w_buffer):
self._check_closed(space)
self._check_readable(space)
rwbuffer = space.getarg_w('w*', w_buffer)
length = rwbuffer.getlength()
target_address = lltype.nullptr(rffi.CCHARP.TO)
if length > 64:
try:
target_address = rwbuffer.get_raw_address()
except ValueError:
pass
if not target_address:
# unoptimized case
try:
buf = os.read(self.fd, length)
except OSError as e:
if e.errno == errno.EAGAIN:
return space.w_None
raise wrap_oserror(space, e,
exception_name='w_IOError')
rwbuffer.setslice(0, buf)
return space.newint(len(buf))
else:
# optimized case: reading more than 64 bytes into a rwbuffer
# with a valid raw address
got = c_read(self.fd, target_address, length)
keepalive_until_here(rwbuffer)
got = rffi.cast(lltype.Signed, got)
if got >= 0:
return space.newint(got)
else:
err = get_saved_errno()
if err == errno.EAGAIN:
return space.w_None
e = OSError(err, "read failed")
raise wrap_oserror(space, e, exception_name='w_IOError')
def rebuild(space, w_cls, w_handle, kind, maxvalue, name):
#
if sys_platform != 'win32' and name is not None:
# like CPython, in this case ignore 'w_handle'
try:
handle = reopen_semaphore(name)
except OSError as e:
raise wrap_oserror(space, e)
else:
handle = handle_w(space, w_handle)
#
self = space.allocate_instance(W_SemLock, w_cls)
self.__init__(space, handle, kind, maxvalue, name)
return self
