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
|
{-# LANGUAGE OverloadedStrings #-}
-- Copyright (C) 2012 John Millikin <john@john-millikin.com>
--
-- Licensed under the Apache License, Version 2.0 (the "License");
-- you may not use this file except in compliance with the License.
-- You may obtain a copy of the License at
--
-- http://www.apache.org/licenses/LICENSE-2.0
--
-- Unless required by applicable law or agreed to in writing, software
-- distributed under the License is distributed on an "AS IS" BASIS,
-- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-- See the License for the specific language governing permissions and
-- limitations under the License.
module DBusTests.Wire (test_Wire) where
import Data.Bifunctor (first)
import Data.Either
import System.Posix.Types (Fd(..))
import Test.Tasty
import Test.Tasty.HUnit
import qualified Data.ByteString.Char8 ()
import DBus
import DBus.Internal.Message
import DBus.Internal.Types
import DBus.Internal.Wire
import DBusTests.Util
test_Wire :: TestTree
test_Wire = testGroup "Wire" $
[ test_Unmarshal
, test_FileDescriptors
]
test_Unmarshal :: TestTree
test_Unmarshal = testGroup "unmarshal"
[ test_UnmarshalUnexpectedEof
]
test_UnmarshalUnexpectedEof :: TestTree
test_UnmarshalUnexpectedEof = testCase "unexpected-eof" $ do
let unmarshaled = unmarshalWithFds "0" []
assertBool "invalid unmarshalled parse" (isLeft unmarshaled)
let Left err = unmarshaled
unmarshalErrorMessage err
@=? "Unexpected end of input while parsing message header."
test_FileDescriptors :: TestTree
test_FileDescriptors = testGroup "Unix File Descriptor Passing" $
[ test_FileDescriptors_Marshal
, test_FileDescriptors_UnmarshalHeaderError
]
test_FileDescriptors_Marshal :: TestTree
test_FileDescriptors_Marshal = testCaseSteps "(un)marshal round trip" $ \step -> do
let baseMsg = methodCall "/" "org.example.iface" "Foo"
step "marshal"
let msg = baseMsg { methodCallBody = [toVariant [Fd 2, Fd 1, Fd 2, Fd 3, Fd 1]] }
Right (bytes, fds) = marshalWithFds LittleEndian firstSerial msg
fds @?= [Fd 2, Fd 1, Fd 3]
step "unmarshal"
let result = receivedMessageBody <$> unmarshalWithFds bytes [Fd 4, Fd 5, Fd 6]
result @?= Right [toVariant [Fd 4, Fd 5, Fd 4, Fd 6, Fd 5]]
test_FileDescriptors_UnmarshalHeaderError :: TestTree
test_FileDescriptors_UnmarshalHeaderError = testCase "UnixFdHeader mismatch" $ do
let msg = (methodCall "/" "org.example.iface" "Foo")
{ methodCallBody = [toVariant [Fd 1, Fd 2, Fd 3]] }
Right (bytes, _fds) = marshalWithFds LittleEndian firstSerial msg
let result = first unmarshalErrorMessage (unmarshalWithFds bytes [Fd 4, Fd 6])
result @?= Left ("File descriptor count in message header (3)"
<> " does not match the number of file descriptors received from the socket (2).")
|
