python wrapper for virtiofsd

vfsd.py (text/x-python), 1.28 KB, created by Huy Logan on 2024-07-08 02:43:49 UTC

#!/usr/bin/python

import socket
import os
import grp
import sys
import multiprocessing
import signal

# /run/vfsd/${1} /mnt/${1}
if __name__ == '__main__':
    try:
        share = sys.argv[1]
        vfsd_daemon = '/usr/libexec/virtiofsd'
        socket_path = '/run/vfsd/' + share
        shared_path = '/mnt/' + share
        kvm_gid = grp.getgrnam('kvm').gr_gid
        print(socket_path)
        if os.path.exists(socket_path):
            os.unlink(socket_path)
        signal.signal(signal.SIGCHLD, signal.SIG_IGN)
        os.umask(0o1117)
        os.setregid(kvm_gid, kvm_gid)
        sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
        sock.bind(socket_path)
        sock.listen(1)
        sem = multiprocessing.Semaphore(1)
        while True:
            sem.acquire()
            newpid = os.fork()
            if newpid == 0:
                (connection, client_address) = sock.accept()
                sock.close()
                connection.settimeout(0)
                os.set_inheritable(connection.fileno(), True)
                args = [vfsd_daemon, '--log-level', 'info', '--fd', str(connection.fileno()), '--shared-dir', shared_path, '--xattr']
                print(args)
                sem.release()
                os.execv(vfsd_daemon, args)
    except:
        raise