mirror of
https://github.com/facebook/sapling.git
synced 2024-10-10 08:47:12 +03:00
7c6f9608c2
Summary: this is the bare minimum to support creating unix domain sockets. We only support using mknod to create a unix socket; other uses will yield an error. I've added an rdev field as a sibling of the existing mode field that we track, as that is the additional parameter that we need to track as part of the special file node. Special file nodes are tracked in the overlay as empty files. Reviewed By: bolinfest Differential Revision: D4774099 fbshipit-source-id: 0824b7e509063faa8bede7aff82a7c51930c4f83
63 lines
2.0 KiB
Python
63 lines
2.0 KiB
Python
#!/usr/bin/env python3
|
|
#
|
|
# Copyright (c) 2017-present, Facebook, Inc.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the BSD-style license found in the
|
|
# LICENSE file in the root directory of this source tree. An additional grant
|
|
# of patent rights can be found in the PATENTS file in the same directory.
|
|
|
|
from .lib import testcase
|
|
import os
|
|
import select
|
|
import socket
|
|
import stat
|
|
import threading
|
|
|
|
PAYLOAD = b'W00t\n'
|
|
|
|
|
|
@testcase.eden_repo_test
|
|
class UnixSocketTest:
|
|
def populate_repo(self):
|
|
self.repo.write_file('hello', 'hola\n')
|
|
self.repo.commit('Initial commit.')
|
|
|
|
def test_unixsock(self):
|
|
'''Create a UNIX domain socket in EdenFS and verify that a client
|
|
can connect and write, and that the server side can accept
|
|
and read data from it.'''
|
|
|
|
filename = os.path.join(self.mount, 'example.sock')
|
|
sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
|
|
sock.setblocking(0) # ensure that we don't block unexpectedly
|
|
try:
|
|
sock.bind(filename)
|
|
st = os.lstat(filename)
|
|
self.assertTrue(stat.S_ISSOCK(st.st_mode))
|
|
sock.listen(1)
|
|
|
|
class Client(threading.Thread):
|
|
def run(self):
|
|
s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
|
|
try:
|
|
s.setblocking(0) # don't block here either
|
|
s.connect(filename)
|
|
s.send(PAYLOAD)
|
|
finally:
|
|
s.close()
|
|
|
|
# spin up a thread to connect back to us and prove that
|
|
# the socket actually functions
|
|
client_thread = Client()
|
|
client_thread.start()
|
|
|
|
readable, _, _ = select.select([sock], [], [], 2)
|
|
self.assertTrue(sock in readable, msg='sock should be ready for accept')
|
|
conn, addr = sock.accept()
|
|
data = conn.recv(len(PAYLOAD))
|
|
self.assertEqual(PAYLOAD, data)
|
|
conn.close()
|
|
finally:
|
|
sock.close()
|