mercurial-scm/hg-stable: mercurial/pure/osutil.py annotate

annotate mercurial/pure/osutil.py @ 43076:2372284d9457

formatting: blacken the codebase This is using my patch to black (https://github.com/psf/black/pull/826) so we don't un-wrap collection literals. Done with: hg files 'set:**.py - mercurial/thirdparty/** - "contrib/python-zstandard/**"' | xargs black -S # skip-blame mass-reformatting only # no-check-commit reformats foo_bar functions Differential Revision: https://phab.mercurial-scm.org/D6971

author	Augie Fackler <augie@google.com>
date	Sun, 06 Oct 2019 09:45:02 -0400
parents	f3fa10a5877d
children	687b865b95ad

rev	line source
8232 823f25b25dea pure/osutil: add copyright and license header Martin Geisler <mg@lazybytes.net> parents: 7704 diff changeset	1 # osutil.py - pure Python version of osutil.c
823f25b25dea pure/osutil: add copyright and license header Martin Geisler <mg@lazybytes.net> parents: 7704 diff changeset	2 #
823f25b25dea pure/osutil: add copyright and license header Martin Geisler <mg@lazybytes.net> parents: 7704 diff changeset	3 # Copyright 2009 Matt Mackall <mpm@selenic.com> and others
823f25b25dea pure/osutil: add copyright and license header Martin Geisler <mg@lazybytes.net> parents: 7704 diff changeset	4 #
823f25b25dea pure/osutil: add copyright and license header Martin Geisler <mg@lazybytes.net> parents: 7704 diff changeset	5 # This software may be used and distributed according to the terms of the
10263 25e572394f5c Update license to GPLv2+ Matt Mackall <mpm@selenic.com> parents: 9031 diff changeset	6 # GNU General Public License version 2 or any later version.
8232 823f25b25dea pure/osutil: add copyright and license header Martin Geisler <mg@lazybytes.net> parents: 7704 diff changeset	7
42531 f3fa10a5877d py3: use integer division for the value passed to xrange Rodrigo Damazio Bovendorp <rdamazio@google.com> parents: 40469 diff changeset	8 from __future__ import absolute_import, division
27338 810337ae1b76 osutil: use absolute_import Gregory Szorc <gregory.szorc@gmail.com> parents: 25645 diff changeset	9
27474 e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	10 import ctypes
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	11 import ctypes.util
7704 30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	12 import os
27474 e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	13 import socket
10651 5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	14 import stat as statmod
7704 30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	15
32406 a9c71d578a1c osutil: switch to policy importer Yuya Nishihara <yuya@tcha.org> parents: 31649 diff changeset	16 from .. import (
39660 3b421154d2ca py3: fix str vs bytes in enough places to run `hg version` on Windows Matt Harbison <matt_harbison@yahoo.com> parents: 38823 diff changeset	17 encoding,
30317 ba2c04059317 py3: use pycompat.ossep at certain places Pulkit Goyal <7895pulkit@gmail.com> parents: 29832 diff changeset	18 pycompat,
ba2c04059317 py3: use pycompat.ossep at certain places Pulkit Goyal <7895pulkit@gmail.com> parents: 29832 diff changeset	19 )
ba2c04059317 py3: use pycompat.ossep at certain places Pulkit Goyal <7895pulkit@gmail.com> parents: 29832 diff changeset	20
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	21
7704 30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	22 def _mode_to_kind(mode):
10651 5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	23 if statmod.S_ISREG(mode):
5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	24 return statmod.S_IFREG
5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	25 if statmod.S_ISDIR(mode):
5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	26 return statmod.S_IFDIR
5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	27 if statmod.S_ISLNK(mode):
5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	28 return statmod.S_IFLNK
5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	29 if statmod.S_ISBLK(mode):
5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	30 return statmod.S_IFBLK
5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	31 if statmod.S_ISCHR(mode):
5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	32 return statmod.S_IFCHR
5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	33 if statmod.S_ISFIFO(mode):
5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	34 return statmod.S_IFIFO
5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	35 if statmod.S_ISSOCK(mode):
5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	36 return statmod.S_IFSOCK
7704 30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	37 return mode
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	38
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	39
32545 0e8b0b9a7acc cffi: split modules from pure Yuya Nishihara <yuya@tcha.org> parents: 32539 diff changeset	40 def listdir(path, stat=False, skip=None):
7704 30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	41 '''listdir(path, stat=False) -> list_of_tuples
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	42
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	43 Return a sorted list containing information about the entries
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	44 in the directory.
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	45
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	46 If stat is True, each element is a 3-tuple:
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	47
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	48 (name, type, stat object)
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	49
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	50 Otherwise, each element is a 2-tuple:
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	51
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	52 (name, type)
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	53 '''
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	54 result = []
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	55 prefix = path
30317 ba2c04059317 py3: use pycompat.ossep at certain places Pulkit Goyal <7895pulkit@gmail.com> parents: 29832 diff changeset	56 if not prefix.endswith(pycompat.ossep):
ba2c04059317 py3: use pycompat.ossep at certain places Pulkit Goyal <7895pulkit@gmail.com> parents: 29832 diff changeset	57 prefix += pycompat.ossep
7704 30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	58 names = os.listdir(path)
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	59 names.sort()
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	60 for fn in names:
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	61 st = os.lstat(prefix + fn)
10651 5f091fc1bab7 style: use consistent variable names (mod) with imports which would shadow Benoit Boissinot <benoit.boissinot@ens-lyon.org>* parents: 10282 diff changeset	62 if fn == skip and statmod.S_ISDIR(st.st_mode):
7704 30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	63 return []
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	64 if stat:
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	65 result.append((fn, _mode_to_kind(st.st_mode), st))
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	66 else:
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	67 result.append((fn, _mode_to_kind(st.st_mode)))
30d1d313370b move mercurial.osutil to mercurial.pure.osutil Martin Geisler <mg@daimi.au.dk> parents: diff changeset	68 return result
8421 b6d0fa8c7685 posixfile: remove posixfile_nt and fix import bug in windows.py Sune Foldager <cryo@cyanite.org> parents: 8232 diff changeset	69
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	70
34645 75979c8d4572 codemod: use pycompat.iswindows Jun Wu <quark@fb.com> parents: 32545 diff changeset	71 if not pycompat.iswindows:
14413 5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	72 posixfile = open
27474 e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	73
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	74 _SCM_RIGHTS = 0x01
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	75 _socklen_t = ctypes.c_uint
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	76
30647 e995f00a9e9a py3: replace sys.platform with pycompat.sysplatform (part 2 of 2) Pulkit Goyal <7895pulkit@gmail.com> parents: 30644 diff changeset	77 if pycompat.sysplatform.startswith('linux'):
27474 e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	78 # socket.h says "the type should be socklen_t but the definition of
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	79 # the kernel is incompatible with this."
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	80 _cmsg_len_t = ctypes.c_size_t
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	81 _msg_controllen_t = ctypes.c_size_t
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	82 _msg_iovlen_t = ctypes.c_size_t
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	83 else:
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	84 _cmsg_len_t = _socklen_t
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	85 _msg_controllen_t = _socklen_t
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	86 _msg_iovlen_t = ctypes.c_int
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	87
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	88 class _iovec(ctypes.Structure):
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	89 _fields_ = [
29712 f15f31505f12 py3: use unicode literals in pure/osutil.py Pulkit Goyal <7895pulkit@gmail.com> parents: 29600 diff changeset	90 (u'iov_base', ctypes.c_void_p),
f15f31505f12 py3: use unicode literals in pure/osutil.py Pulkit Goyal <7895pulkit@gmail.com> parents: 29600 diff changeset	91 (u'iov_len', ctypes.c_size_t),
27474 e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	92 ]
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	93
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	94 class _msghdr(ctypes.Structure):
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	95 _fields_ = [
29712 f15f31505f12 py3: use unicode literals in pure/osutil.py Pulkit Goyal <7895pulkit@gmail.com> parents: 29600 diff changeset	96 (u'msg_name', ctypes.c_void_p),
f15f31505f12 py3: use unicode literals in pure/osutil.py Pulkit Goyal <7895pulkit@gmail.com> parents: 29600 diff changeset	97 (u'msg_namelen', _socklen_t),
f15f31505f12 py3: use unicode literals in pure/osutil.py Pulkit Goyal <7895pulkit@gmail.com> parents: 29600 diff changeset	98 (u'msg_iov', ctypes.POINTER(_iovec)),
f15f31505f12 py3: use unicode literals in pure/osutil.py Pulkit Goyal <7895pulkit@gmail.com> parents: 29600 diff changeset	99 (u'msg_iovlen', _msg_iovlen_t),
f15f31505f12 py3: use unicode literals in pure/osutil.py Pulkit Goyal <7895pulkit@gmail.com> parents: 29600 diff changeset	100 (u'msg_control', ctypes.c_void_p),
f15f31505f12 py3: use unicode literals in pure/osutil.py Pulkit Goyal <7895pulkit@gmail.com> parents: 29600 diff changeset	101 (u'msg_controllen', _msg_controllen_t),
f15f31505f12 py3: use unicode literals in pure/osutil.py Pulkit Goyal <7895pulkit@gmail.com> parents: 29600 diff changeset	102 (u'msg_flags', ctypes.c_int),
27474 e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	103 ]
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	104
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	105 class _cmsghdr(ctypes.Structure):
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	106 _fields_ = [
29712 f15f31505f12 py3: use unicode literals in pure/osutil.py Pulkit Goyal <7895pulkit@gmail.com> parents: 29600 diff changeset	107 (u'cmsg_len', _cmsg_len_t),
f15f31505f12 py3: use unicode literals in pure/osutil.py Pulkit Goyal <7895pulkit@gmail.com> parents: 29600 diff changeset	108 (u'cmsg_level', ctypes.c_int),
f15f31505f12 py3: use unicode literals in pure/osutil.py Pulkit Goyal <7895pulkit@gmail.com> parents: 29600 diff changeset	109 (u'cmsg_type', ctypes.c_int),
f15f31505f12 py3: use unicode literals in pure/osutil.py Pulkit Goyal <7895pulkit@gmail.com> parents: 29600 diff changeset	110 (u'cmsg_data', ctypes.c_ubyte * 0),
27474 e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	111 ]
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	112
29712 f15f31505f12 py3: use unicode literals in pure/osutil.py Pulkit Goyal <7895pulkit@gmail.com> parents: 29600 diff changeset	113 _libc = ctypes.CDLL(ctypes.util.find_library(u'c'), use_errno=True)
27971 f7d0c28d34b3 osutil: do not abort loading pure module just because libc has no recvmsg() Yuya Nishihara <yuya@tcha.org> parents: 27704 diff changeset	114 _recvmsg = getattr(_libc, 'recvmsg', None)
f7d0c28d34b3 osutil: do not abort loading pure module just because libc has no recvmsg() Yuya Nishihara <yuya@tcha.org> parents: 27704 diff changeset	115 if _recvmsg:
f7d0c28d34b3 osutil: do not abort loading pure module just because libc has no recvmsg() Yuya Nishihara <yuya@tcha.org> parents: 27704 diff changeset	116 _recvmsg.restype = getattr(ctypes, 'c_ssize_t', ctypes.c_long)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	117 _recvmsg.argtypes = (
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	118 ctypes.c_int,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	119 ctypes.POINTER(_msghdr),
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	120 ctypes.c_int,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	121 )
27971 f7d0c28d34b3 osutil: do not abort loading pure module just because libc has no recvmsg() Yuya Nishihara <yuya@tcha.org> parents: 27704 diff changeset	122 else:
f7d0c28d34b3 osutil: do not abort loading pure module just because libc has no recvmsg() Yuya Nishihara <yuya@tcha.org> parents: 27704 diff changeset	123 # recvmsg isn't always provided by libc; such systems are unsupported
f7d0c28d34b3 osutil: do not abort loading pure module just because libc has no recvmsg() Yuya Nishihara <yuya@tcha.org> parents: 27704 diff changeset	124 def _recvmsg(sockfd, msg, flags):
f7d0c28d34b3 osutil: do not abort loading pure module just because libc has no recvmsg() Yuya Nishihara <yuya@tcha.org> parents: 27704 diff changeset	125 raise NotImplementedError('unsupported platform')
27474 e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	126
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	127 def _CMSG_FIRSTHDR(msgh):
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	128 if msgh.msg_controllen < ctypes.sizeof(_cmsghdr):
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	129 return
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	130 cmsgptr = ctypes.cast(msgh.msg_control, ctypes.POINTER(_cmsghdr))
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	131 return cmsgptr.contents
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	132
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	133 # The pure version is less portable than the native version because the
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	134 # handling of socket ancillary data heavily depends on C preprocessor.
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	135 # Also, some length fields are wrongly typed in Linux kernel.
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	136 def recvfds(sockfd):
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	137 """receive list of file descriptors via socket"""
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	138 dummy = (ctypes.c_ubyte * 1)()
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	139 iov = _iovec(ctypes.cast(dummy, ctypes.c_void_p), ctypes.sizeof(dummy))
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	140 cbuf = ctypes.create_string_buffer(256)
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	141 msgh = _msghdr(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	142 None,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	143 0,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	144 ctypes.pointer(iov),
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	145 1,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	146 ctypes.cast(cbuf, ctypes.c_void_p),
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	147 ctypes.sizeof(cbuf),
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	148 0,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	149 )
27474 e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	150 r = _recvmsg(sockfd, ctypes.byref(msgh), 0)
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	151 if r < 0:
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	152 e = ctypes.get_errno()
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	153 raise OSError(e, os.strerror(e))
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	154 # assumes that the first cmsg has fds because it isn't easy to write
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	155 # portable CMSG_NXTHDR() with ctypes.
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	156 cmsg = _CMSG_FIRSTHDR(msgh)
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	157 if not cmsg:
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	158 return []
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	159 if (
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	160 cmsg.cmsg_level != socket.SOL_SOCKET
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	161 or cmsg.cmsg_type != _SCM_RIGHTS
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	162 ):
27474 e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	163 return []
e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	164 rfds = ctypes.cast(cmsg.cmsg_data, ctypes.POINTER(ctypes.c_int))
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	165 rfdscount = (
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	166 cmsg.cmsg_len - _cmsghdr.cmsg_data.offset
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	167 ) // ctypes.sizeof(ctypes.c_int)
38823 e7aa113b14f7 global: use pycompat.xrange() Gregory Szorc <gregory.szorc@gmail.com> parents: 34645 diff changeset	168 return [rfds[i] for i in pycompat.xrange(rfdscount)]
27474 e517a89c24e1 osutil: implement pure version of recvfds() for PyPy Yuya Nishihara <yuya@tcha.org> parents: 27338 diff changeset	169
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	170
14413 5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	171 else:
27338 810337ae1b76 osutil: use absolute_import Gregory Szorc <gregory.szorc@gmail.com> parents: 25645 diff changeset	172 import msvcrt
14413 5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	173
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	174 _kernel32 = ctypes.windll.kernel32
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	175
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	176 _DWORD = ctypes.c_ulong
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	177 _LPCSTR = _LPSTR = ctypes.c_char_p
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	178 _HANDLE = ctypes.c_void_p
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	179
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	180 _INVALID_HANDLE_VALUE = _HANDLE(-1).value
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	181
18959 2f6418d8a4c9 check-code: catch trailing space in comments Mads Kiilerich <madski@unity3d.com> parents: 17429 diff changeset	182 # CreateFile
14413 5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	183 _FILE_SHARE_READ = 0x00000001
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	184 _FILE_SHARE_WRITE = 0x00000002
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	185 _FILE_SHARE_DELETE = 0x00000004
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	186
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	187 _CREATE_ALWAYS = 2
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	188 _OPEN_EXISTING = 3
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	189 _OPEN_ALWAYS = 4
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	190
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	191 _GENERIC_READ = 0x80000000
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	192 _GENERIC_WRITE = 0x40000000
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	193
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	194 _FILE_ATTRIBUTE_NORMAL = 0x80
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	195
17429 72fa4ef2245f declare local constants instead of using magic values and comments Mads Kiilerich <mads@kiilerich.com> parents: 16686 diff changeset	196 # open_osfhandle flags
14413 5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	197 _O_RDONLY = 0x0000
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	198 _O_RDWR = 0x0002
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	199 _O_APPEND = 0x0008
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	200
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	201 _O_TEXT = 0x4000
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	202 _O_BINARY = 0x8000
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	203
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	204 # types of parameters of C functions used (required by pypy)
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	205
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	206 _kernel32.CreateFileA.argtypes = [
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	207 _LPCSTR,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	208 _DWORD,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	209 _DWORD,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	210 ctypes.c_void_p,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	211 _DWORD,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	212 _DWORD,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	213 _HANDLE,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	214 ]
14413 5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	215 _kernel32.CreateFileA.restype = _HANDLE
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	216
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	217 def _raiseioerror(name):
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	218 err = ctypes.WinError()
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	219 raise IOError(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	220 err.errno, r'%s: %s' % (encoding.strfromlocal(name), err.strerror)
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	221 )
14413 5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	222
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	223 class posixfile(object):
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	224 '''a file object aiming for POSIX-like semantics
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	225
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	226 CPython's open() returns a file that was opened without setting the
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	227 _FILE_SHARE_DELETE flag, which causes rename and unlink to abort.
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	228 This even happens if any hardlinked copy of the file is in open state.
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	229 We set _FILE_SHARE_DELETE here, so files opened with posixfile can be
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	230 renamed and deleted while they are held open.
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	231 Note that if a file opened with posixfile is unlinked, the file
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	232 remains but cannot be opened again or be recreated under the same name,
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	233 until all reading processes have closed the file.'''
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	234
39660 3b421154d2ca py3: fix str vs bytes in enough places to run `hg version` on Windows Matt Harbison <matt_harbison@yahoo.com> parents: 38823 diff changeset	235 def __init__(self, name, mode=b'r', bufsize=-1):
3b421154d2ca py3: fix str vs bytes in enough places to run `hg version` on Windows Matt Harbison <matt_harbison@yahoo.com> parents: 38823 diff changeset	236 if b'b' in mode:
14413 5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	237 flags = _O_BINARY
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	238 else:
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	239 flags = _O_TEXT
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	240
39660 3b421154d2ca py3: fix str vs bytes in enough places to run `hg version` on Windows Matt Harbison <matt_harbison@yahoo.com> parents: 38823 diff changeset	241 m0 = mode[0:1]
3b421154d2ca py3: fix str vs bytes in enough places to run `hg version` on Windows Matt Harbison <matt_harbison@yahoo.com> parents: 38823 diff changeset	242 if m0 == b'r' and b'+' not in mode:
14413 5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	243 flags \|= _O_RDONLY
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	244 access = _GENERIC_READ
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	245 else:
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	246 # work around http://support.microsoft.com/kb/899149 and
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	247 # set _O_RDWR for 'w' and 'a', even if mode has no '+'
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	248 flags \|= _O_RDWR
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	249 access = _GENERIC_READ \| _GENERIC_WRITE
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	250
39660 3b421154d2ca py3: fix str vs bytes in enough places to run `hg version` on Windows Matt Harbison <matt_harbison@yahoo.com> parents: 38823 diff changeset	251 if m0 == b'r':
14413 5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	252 creation = _OPEN_EXISTING
39660 3b421154d2ca py3: fix str vs bytes in enough places to run `hg version` on Windows Matt Harbison <matt_harbison@yahoo.com> parents: 38823 diff changeset	253 elif m0 == b'w':
14413 5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	254 creation = _CREATE_ALWAYS
39660 3b421154d2ca py3: fix str vs bytes in enough places to run `hg version` on Windows Matt Harbison <matt_harbison@yahoo.com> parents: 38823 diff changeset	255 elif m0 == b'a':
14413 5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	256 creation = _OPEN_ALWAYS
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	257 flags \|= _O_APPEND
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	258 else:
39660 3b421154d2ca py3: fix str vs bytes in enough places to run `hg version` on Windows Matt Harbison <matt_harbison@yahoo.com> parents: 38823 diff changeset	259 raise ValueError(r"invalid mode: %s" % pycompat.sysstr(mode))
14413 5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	260
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	261 fh = _kernel32.CreateFileA(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	262 name,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	263 access,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	264 _FILE_SHARE_READ \| _FILE_SHARE_WRITE \| _FILE_SHARE_DELETE,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	265 None,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	266 creation,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	267 _FILE_ATTRIBUTE_NORMAL,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	268 None,
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 42531 diff changeset	269 )
14413 5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	270 if fh == _INVALID_HANDLE_VALUE:
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	271 _raiseioerror(name)
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	272
16474 ee553e6cd8c4 pure/osutil: use Python's msvcrt module (issue3380) Adrian Buehlmann <adrian@cadifra.com> parents: 15040 diff changeset	273 fd = msvcrt.open_osfhandle(fh, flags)
14413 5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	274 if fd == -1:
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	275 _kernel32.CloseHandle(fh)
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	276 _raiseioerror(name)
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	277
30945 82f1ef8b4477 py3: convert the mode argument of os.fdopen to unicodes (2 of 2) Pulkit Goyal <7895pulkit@gmail.com> parents: 30647 diff changeset	278 f = os.fdopen(fd, pycompat.sysstr(mode), bufsize)
14413 5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	279 # unfortunately, f.name is '<fdopen>' at this point -- so we store
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	280 # the name on this wrapper. We cannot just assign to f.name,
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	281 # because that attribute is read-only.
31649 f80d9ddc40f3 py3: abuse r'' to preserve str-ness of literals passed to __setattr__() Yuya Nishihara <yuya@tcha.org> parents: 30945 diff changeset	282 object.__setattr__(self, r'name', name)
f80d9ddc40f3 py3: abuse r'' to preserve str-ness of literals passed to __setattr__() Yuya Nishihara <yuya@tcha.org> parents: 30945 diff changeset	283 object.__setattr__(self, r'_file', f)
14413 5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	284
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	285 def __iter__(self):
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	286 return self._file
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	287
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	288 def __getattr__(self, name):
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	289 return getattr(self._file, name)
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	290
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	291 def __setattr__(self, name, value):
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	292 '''mimics the read-only attributes of Python file objects
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	293 by raising 'TypeError: readonly attribute' if someone tries:
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	294 f = posixfile('foo.txt')
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	295 f.name = 'bla' '''
5ef18e28df19 pure: provide more correct implementation of posixfile for Windows Adrian Buehlmann <adrian@cadifra.com> parents: 10651 diff changeset	296 return self._file.__setattr__(name, value)
27704 051b0dcec98b osutil: implement __enter__ and __exit__ on posixfile Gregory Szorc <gregory.szorc@gmail.com> parents: 27512 diff changeset	297
051b0dcec98b osutil: implement __enter__ and __exit__ on posixfile Gregory Szorc <gregory.szorc@gmail.com> parents: 27512 diff changeset	298 def __enter__(self):
40469 120ecb17242b windows: ensure pure posixfile fd doesn't escape by entering context manager Matt Harbison <matt_harbison@yahoo.com> parents: 39660 diff changeset	299 self._file.__enter__()
120ecb17242b windows: ensure pure posixfile fd doesn't escape by entering context manager Matt Harbison <matt_harbison@yahoo.com> parents: 39660 diff changeset	300 return self
27704 051b0dcec98b osutil: implement __enter__ and __exit__ on posixfile Gregory Szorc <gregory.szorc@gmail.com> parents: 27512 diff changeset	301
051b0dcec98b osutil: implement __enter__ and __exit__ on posixfile Gregory Szorc <gregory.szorc@gmail.com> parents: 27512 diff changeset	302 def __exit__(self, exc_type, exc_value, exc_tb):
051b0dcec98b osutil: implement __enter__ and __exit__ on posixfile Gregory Szorc <gregory.szorc@gmail.com> parents: 27512 diff changeset	303 return self._file.__exit__(exc_type, exc_value, exc_tb)

Mercurial > public > mercurial-scm > hg-stable

annotate mercurial/pure/osutil.py @ 43076:2372284d9457