mirror of
https://github.com/xcat2/confluent.git
synced 2026-08-28 09:36:41 +00:00
Markus Hilger
acd6bb228c
Each is the only thing keeping its rule group from being selectable whole. userutil.py imported ctypes with a star; the names it uses are POINTER, byref, c_char_p, c_int, c_int32, c_uint and cdll. The oem lookup loop had an else with no break, so the else always ran. The alert parameter table wrapped int in a lambda that only forwards to it. And the watchdog interval passed 0 where os.environ.get documents a string, which worked because int(0) is 0.
44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
from ctypes import POINTER, byref, c_char_p, c_int, c_int32, c_uint, cdll
|
|
from ctypes.util import find_library
|
|
import confluent.util as util
|
|
import grp
|
|
import pwd
|
|
libc = cdll.LoadLibrary(find_library('c'))
|
|
_getgrouplist = libc.getgrouplist
|
|
_getgrouplist.restype = c_int32
|
|
|
|
|
|
class TooSmallException(Exception):
|
|
def __init__(self, count):
|
|
self.count = count
|
|
super(TooSmallException, self).__init__()
|
|
|
|
|
|
def getgrouplist(name, gid, ng=32):
|
|
_getgrouplist.argtypes = [c_char_p, c_uint, POINTER(c_uint * ng), POINTER(c_int)]
|
|
glist = (c_uint * ng)()
|
|
nglist = c_int(ng)
|
|
if not isinstance(name, bytes):
|
|
name = name.encode('utf-8')
|
|
count = _getgrouplist(name, gid, byref(glist), byref(nglist))
|
|
if count < 0:
|
|
raise TooSmallException(nglist.value)
|
|
for gidx in range(count):
|
|
gent = glist[gidx]
|
|
yield grp.getgrgid(gent).gr_name
|
|
|
|
|
|
def grouplist(username):
|
|
username = util.stringify(username)
|
|
pent = pwd.getpwnam(username)
|
|
try:
|
|
groups = getgrouplist(pent.pw_name, pent.pw_gid)
|
|
except TooSmallException as e:
|
|
groups = getgrouplist(pent.pw_name, pent.pw_gid, e.count)
|
|
return list(groups)
|
|
|
|
if __name__ == '__main__':
|
|
import sys
|
|
print(repr(grouplist(sys.argv[1])))
|
|
|