summaryrefslogtreecommitdiff
path: root/spec/ruby/library/socket/unixsocket/recvfrom_spec.rb
blob: c0e1cf670b66f96f9e593d9dd83c0309a55ec93c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
require_relative '../spec_helper'
require_relative '../fixtures/classes'

describe "UNIXSocket#recvfrom" do
  platform_is_not :windows do
    before :each do
      @path = SocketSpecs.socket_path
      @server = UNIXServer.open(@path)
      @client = UNIXSocket.open(@path)
    end

    after :each do
      @client.close
      @server.close
      SocketSpecs.rm_socket @path
    end

    it "receives len bytes from sock" do
      @client.send("foobar", 0)
      sock = @server.accept
      sock.recvfrom(6).first.should == "foobar"
      sock.close
    end

    it "returns an array with data and information on the sender" do
      @client.send("foobar", 0)
      sock = @server.accept
      data = sock.recvfrom(6)
      data.first.should == "foobar"
      data.last.should == ["AF_UNIX", ""]
      sock.close
    end

    it "uses different message options" do
      @client.send("foobar", Socket::MSG_PEEK)
      sock = @server.accept
      peek_data = sock.recvfrom(6, Socket::MSG_PEEK) # Does not retrieve the message
      real_data = sock.recvfrom(6)

      real_data.should == peek_data
      peek_data.should == ["foobar", ["AF_UNIX", ""]]
      sock.close
    end
  end
end


with_feature :unix_socket do
  describe 'UNIXSocket#recvfrom' do
    describe 'using a socket pair' do
      before do
        @client, @server = UNIXSocket.socketpair
        @client.write('hello')
      end

      after do
        @client.close
        @server.close
      end

      it 'returns an Array containing the data and address information' do
        @server.recvfrom(5).should == ['hello', ['AF_UNIX', '']]
      end
    end

    # These specs are taken from the rdoc examples on UNIXSocket#recvfrom.
    describe 'using a UNIX socket constructed using UNIXSocket.for_fd' do
      before do
        @path1 = SocketSpecs.socket_path
        @path2 = SocketSpecs.socket_path.chop + '2'
        rm_r(@path2)

        @client_raw = Socket.new(:UNIX, :DGRAM)
        @client_raw.bind(Socket.sockaddr_un(@path1))

        @server_raw = Socket.new(:UNIX, :DGRAM)
        @server_raw.bind(Socket.sockaddr_un(@path2))

        @socket = UNIXSocket.for_fd(@server_raw.fileno)
        @socket.autoclose = false
      end

      after do
        @client_raw.close
        @server_raw.close # also closes @socket

        rm_r @path1
        rm_r @path2
      end

      it 'returns an Array containing the data and address information' do
        @client_raw.send('hello', 0, Socket.sockaddr_un(@path2))

        @socket.recvfrom(5).should == ['hello', ['AF_UNIX', @path1]]
      end
    end
  end
end