#!/usr/bin/env python

# FreeBSD support modules

import struct, pwd, grp

def getpeerid(sock):
    """ Get peer credentials on a UNIX domain socket.

        Returns a nested tuple: (uid, (gids)) """

    LOCAL_PEERCRED = 0x001
    NGROUPS = 16

#struct xucred {
#        u_int   cr_version;             /* structure layout version */
#        uid_t   cr_uid;                 /* effective user id */
#        short   cr_ngroups;             /* number of groups */
#        gid_t   cr_groups[NGROUPS];     /* groups */
#        void    *_cr_unused1;           /* compatibility with old ucred */
#};

    xucred_fmt = '2ih16iP'
    res = tuple(struct.unpack(xucred_fmt, sock.getsockopt(0, LOCAL_PEERCRED, struct.calcsize(xucred_fmt))))
    
    # Check this is the above version of the structure
    if res[0] != 0:
        raise OSError

    return (res[1], res[3:3+res[2]])

def getuidbyname(username):
    if str(username).isdigit():
        return int(username)
    return pwd.getpwnam(username)[2]

def getgidbyname(grname):
    if str(grname).isdigit():
        return int(grname)
    return grp.getgrnam(grname)[2]

if __name__ == "__main__":

    MYSOCK='/tmp/mysock'
    import os, socket

    if os.path.exists(MYSOCK):
        os.unlink(MYSOCK)
    s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
    s.bind(MYSOCK)
    os.chmod(MYSOCK, 0666)
    s.listen(10)
    (conn, addr) = s.accept()

    print getpeerid(conn)

    for i in conn.makefile():
        print i.rstrip()
    conn.close()