1 files changed, 108 insertions, 0 deletions
diff --git a/src/server/bnetserver/REST/LoginRESTService.h b/src/server/bnetserver/REST/LoginRESTService.h
new file mode 100644
index 00000000000..01c08577688
--- /dev/null
+++ b/src/server/bnetserver/REST/LoginRESTService.h
@@ -0,0 +1,108 @@
+/*
+ * Copyright (C) 2008-2016 TrinityCore <http://www.trinitycore.org/>
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License as published by the
+ * Free Software Foundation; either version 2 of the License, or (at your
+ * option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
+ * more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#ifndef LoginRESTService_h__
+#define LoginRESTService_h__
+
+#include "Session.h"
+#include "Define.h"
+#include "Login.pb.h"
+#include <boost/asio/io_service.hpp>
+#include <boost/asio/ip/tcp.hpp>
+#include <boost/asio/ip/address.hpp>
+#include <boost/asio/deadline_timer.hpp>
+#include <atomic>
+#include <mutex>
+#include <thread>
+
+struct soap;
+struct soap_plugin;
+
+class LoginRESTService
+{
+public:
+    LoginRESTService() : _stopped(false), _port(0) { }
+
+    static LoginRESTService& Instance();
+
+    bool Start(boost::asio::io_service& ioService);
+    void Stop();
+
+    boost::asio::ip::tcp::endpoint const& GetAddressForClient(boost::asio::ip::address const& address) const;
+
+    std::unique_ptr<Battlenet::Session::AccountInfo> VerifyLoginTicket(std::string const& id);
+
+private:
+    void Run();
+
+    friend int32 handle_get_plugin(soap* soapClient);
+    int32 HandleGet(soap* soapClient);
+
+    friend int32 handle_post_plugin(soap* soapClient);
+    int32 HandlePost(soap* soapClient);
+
+    int32 SendResponse(soap* soapClient, google::protobuf::Message const& response);
+
+    std::string CalculateShaPassHash(std::string const& name, std::string const& password);
+
+    void AddLoginTicket(std::string const& id, std::unique_ptr<Battlenet::Session::AccountInfo> accountInfo);
+    void CleanupLoginTickets(boost::system::error_code const& error);
+
+    struct LoginTicket
+    {
+        std::string Id;
+        std::unique_ptr<Battlenet::Session::AccountInfo> Account;
+        std::time_t ExpiryTime;
+    };
+
+    struct ResponseCodePlugin
+    {
+        static char const* const PluginId;
+        static int32 Init(soap* s, soap_plugin*, void*);
+        static void Destroy(soap* s, soap_plugin* p);
+        static int32 ChangeResponse(soap* s, int32 originalResponse, size_t contentLength);
+
+        int32(*fresponse)(soap* s, int32 status, size_t length);
+        int32 ErrorCode;
+    };
+
+    struct ContentTypePlugin
+    {
+        static char const* const PluginId;
+        static int32 Init(soap* s, soap_plugin* p, void*);
+        static void Destroy(soap* s, soap_plugin* p);
+        static int32 OnSetHeader(soap* s, char const* key, char const* value);
+
+        int32(*fposthdr)(soap* s, char const* key, char const* value);
+        char const* ContentType;
+    };
+
+    std::thread _thread;
+    std::atomic<bool> _stopped;
+    Battlenet::JSON::Login::FormInputs _formInputs;
+    std::string _bindIP;
+    int32 _port;
+    boost::asio::ip::tcp::endpoint _externalAddress;
+    boost::asio::ip::tcp::endpoint _localAddress;
+    std::mutex _loginTicketMutex;
+    std::unordered_map<std::string, LoginTicket> _validLoginTickets;
+    boost::asio::deadline_timer* _loginTicketCleanupTimer;
+};
+
+#define sLoginService LoginRESTService::Instance()
+
+#endif // LoginRESTService_h__