1/*
2 * Copyright (C) 2012 Apple Inc. All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
6 * are met:
7 * 1. Redistributions of source code must retain the above copyright
8 *    notice, this list of conditions and the following disclaimer.
9 * 2. Redistributions in binary form must reproduce the above copyright
10 *    notice, this list of conditions and the following disclaimer in the
11 *    documentation and/or other materials provided with the distribution.
12 *
13 * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23 * THE POSSIBILITY OF SUCH DAMAGE.
24 */
25
26#include "config.h"
27#include "MessageReceiverMap.h"
28
29#include "MessageDecoder.h"
30#include "MessageReceiver.h"
31
32namespace CoreIPC {
33
34MessageReceiverMap::MessageReceiverMap()
35{
36}
37
38MessageReceiverMap::~MessageReceiverMap()
39{
40}
41
42void MessageReceiverMap::addMessageReceiver(StringReference messageReceiverName, MessageReceiver* messageReceiver)
43{
44    ASSERT(!m_globalMessageReceivers.contains(messageReceiverName));
45    m_globalMessageReceivers.set(messageReceiverName, messageReceiver);
46}
47
48void MessageReceiverMap::addMessageReceiver(StringReference messageReceiverName, uint64_t destinationID, MessageReceiver* messageReceiver)
49{
50    ASSERT(!m_messageReceivers.contains(std::make_pair(messageReceiverName, destinationID)));
51    ASSERT(!m_globalMessageReceivers.contains(messageReceiverName));
52
53    m_messageReceivers.set(std::make_pair(messageReceiverName, destinationID), messageReceiver);
54}
55
56void MessageReceiverMap::removeMessageReceiver(StringReference messageReceiverName)
57{
58    ASSERT(m_globalMessageReceivers.contains(messageReceiverName));
59
60    m_globalMessageReceivers.remove(messageReceiverName);
61}
62
63void MessageReceiverMap::removeMessageReceiver(StringReference messageReceiverName, uint64_t destinationID)
64{
65    ASSERT(m_messageReceivers.contains(std::make_pair(messageReceiverName, destinationID)));
66
67    m_messageReceivers.remove(std::make_pair(messageReceiverName, destinationID));
68}
69
70void MessageReceiverMap::invalidate()
71{
72    m_globalMessageReceivers.clear();
73    m_messageReceivers.clear();
74}
75
76bool MessageReceiverMap::dispatchMessage(Connection* connection, MessageDecoder& decoder)
77{
78    if (MessageReceiver* messageReceiver = m_globalMessageReceivers.get(decoder.messageReceiverName())) {
79        ASSERT(!decoder.destinationID());
80
81        messageReceiver->didReceiveMessage(connection, decoder);
82        return true;
83    }
84
85    if (MessageReceiver* messageReceiver = m_messageReceivers.get(std::make_pair(decoder.messageReceiverName(), decoder.destinationID()))) {
86        messageReceiver->didReceiveMessage(connection, decoder);
87        return true;
88    }
89
90    return false;
91}
92
93bool MessageReceiverMap::dispatchSyncMessage(Connection* connection, MessageDecoder& decoder, OwnPtr<MessageEncoder>& replyEncoder)
94{
95    if (MessageReceiver* messageReceiver = m_globalMessageReceivers.get(decoder.messageReceiverName())) {
96        ASSERT(!decoder.destinationID());
97
98        messageReceiver->didReceiveSyncMessage(connection, decoder, replyEncoder);
99        return true;
100    }
101
102    if (MessageReceiver* messageReceiver = m_messageReceivers.get(std::make_pair(decoder.messageReceiverName(), decoder.destinationID()))) {
103        messageReceiver->didReceiveSyncMessage(connection, decoder, replyEncoder);
104        return true;
105    }
106
107    return false;
108}
109
110} // namespace CoreIPC
111