Slimber/Server.h


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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126

/*
 * Copyright (c) 2010-2018 Isode Limited.
 * All rights reserved.
 * See the COPYING file for more information.
 */

#pragma once

#include <memory>
#include <vector>

#include <boost/optional.hpp>

#include <Swiften/Base/IDGenerator.h>
#include <Swiften/JID/JID.h>
#include <Swiften/LinkLocal/LinkLocalServiceInfo.h>
#include <Swiften/Network/BoostConnectionServer.h>
#include <Swiften/Network/BoostIOServiceThread.h>
#include <Swiften/Parser/PayloadParsers/FullPayloadParserFactoryCollection.h>
#include <Swiften/Parser/PlatformXMLParserFactory.h>
#include <Swiften/Serializer/PayloadSerializers/FullPayloadSerializerCollection.h>

#include <Limber/Server/ServerFromClientSession.h>
#include <Limber/Server/UserRegistry.h>

#include <Slimber/ServerError.h>

namespace Swift {
    class DNSSDServiceID;

    class VCardCollection;
    class LinkLocalConnector;
    class LinkLocalServiceBrowser;
    class LinkLocalPresenceManager;
    class BoostConnectionServer;
    class SessionTracer;
    class RosterPayload;
    class Presence;
    class EventLoop;

    class Server {
        public:
            Server(
                    unsigned short clientConnectionPort,
                    unsigned short linkLocalConnectionPort,
                    LinkLocalServiceBrowser* browser,
                    VCardCollection* vCardCollection,
                    EventLoop* eventLoop);
            ~Server();

            void start();
            void stop();

            unsigned short getLinkLocalPort() const {
                return linkLocalConnectionPort;
            }

            unsigned short getClientToServerPort() const {
                return clientConnectionPort;
            }

            boost::signals2::signal<void (bool)> onSelfConnected;
            boost::signals2::signal<void (boost::optional<ServerError>)> onStopped;

        private:
            void stop(boost::optional<ServerError>);

            void handleNewClientConnection(std::shared_ptr<Connection> c);
            void handleSessionStarted();
            void handleSessionFinished(std::shared_ptr<ServerFromClientSession>);
            void handleElementReceived(std::shared_ptr<ToplevelElement> element, std::shared_ptr<ServerFromClientSession> session);
            void handleRosterChanged(std::shared_ptr<RosterPayload> roster);
            void handlePresenceChanged(std::shared_ptr<Presence> presence);
            void handleServiceRegistered(const DNSSDServiceID& service);
            void handleNewLinkLocalConnection(std::shared_ptr<Connection> connection);
            void handleLinkLocalSessionFinished(std::shared_ptr<Session> session);
            void handleLinkLocalElementReceived(std::shared_ptr<ToplevelElement> element, std::shared_ptr<Session> session);
            void handleConnectFinished(std::shared_ptr<LinkLocalConnector> connector, bool error);
            void handleClientConnectionServerStopped(
                    boost::optional<BoostConnectionServer::Error>);
            void handleLinkLocalConnectionServerStopped(
                    boost::optional<BoostConnectionServer::Error>);
            std::shared_ptr<Session> getLinkLocalSessionForJID(const JID& jid);
            std::shared_ptr<LinkLocalConnector> getLinkLocalConnectorForJID(const JID& jid);
            void registerLinkLocalSession(std::shared_ptr<Session> session);
            void unregisterService();
            LinkLocalServiceInfo getLinkLocalServiceInfo(std::shared_ptr<Presence> presence);

        private:
            class DummyUserRegistry : public UserRegistry {
                public:
                    DummyUserRegistry() {}

                    virtual bool isValidUserPassword(const JID&, const SafeByteArray&) const {
                        return true;
                    }
            };

        private:
            IDGenerator idGenerator;
            FullPayloadParserFactoryCollection payloadParserFactories;
            FullPayloadSerializerCollection payloadSerializers;
            BoostIOServiceThread boostIOServiceThread;
            DummyUserRegistry userRegistry;
            PlatformXMLParserFactory xmlParserFactory;
            bool linkLocalServiceRegistered;
            bool rosterRequested;
            unsigned short clientConnectionPort;
            unsigned short linkLocalConnectionPort;
            LinkLocalServiceBrowser* linkLocalServiceBrowser;
            VCardCollection* vCardCollection;
            EventLoop* eventLoop;
            LinkLocalPresenceManager* presenceManager;
            bool stopping;
            std::shared_ptr<BoostConnectionServer> serverFromClientConnectionServer;
            std::vector<boost::signals2::connection> serverFromClientConnectionServerSignalConnections;
            std::shared_ptr<ServerFromClientSession> serverFromClientSession;
            std::shared_ptr<Presence> lastPresence;
            JID selfJID;
            std::shared_ptr<BoostConnectionServer> serverFromNetworkConnectionServer;
            std::vector<boost::signals2::connection> serverFromNetworkConnectionServerSignalConnections;
            std::vector< std::shared_ptr<Session> > linkLocalSessions;
            std::vector< std::shared_ptr<LinkLocalConnector> > connectors;
            std::vector< std::shared_ptr<SessionTracer> > tracers;
    };
}