diff options
author | Matthias Schiffer <matthias@gamezock.de> | 2008-07-01 03:24:58 +0200 |
---|---|---|
committer | Matthias Schiffer <matthias@gamezock.de> | 2008-07-01 03:24:58 +0200 |
commit | 7601149f476c1aa3dae6cd55027a36b62efabf12 (patch) | |
tree | 947a8ebeaebaf6a366a0490975e5fa2fa689db9c /src/Common/RequestManager.cpp | |
parent | 7be6ee9c1cd2c2c5172ad4a094b67c25d6c5b72c (diff) | |
download | mad-7601149f476c1aa3dae6cd55027a36b62efabf12.tar mad-7601149f476c1aa3dae6cd55027a36b62efabf12.zip |
RequestManager hinzugef?gt
Diffstat (limited to 'src/Common/RequestManager.cpp')
-rw-r--r-- | src/Common/RequestManager.cpp | 111 |
1 files changed, 111 insertions, 0 deletions
diff --git a/src/Common/RequestManager.cpp b/src/Common/RequestManager.cpp new file mode 100644 index 0000000..9cc4c71 --- /dev/null +++ b/src/Common/RequestManager.cpp @@ -0,0 +1,111 @@ +/* + * RequestManager.cpp + * + * Copyright (C) 2008 Matthias Schiffer <matthias@gamezock.de> + * + * 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 3 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/>. + */ + +#include "RequestManager.h" + +#include <iostream> + +namespace Mad { +namespace Common { + +void RequestManager::receiveHandler(Net::Connection *connection, const Net::Packet &packet) { + std::map<Net::Connection*,RequestInfo>::iterator it = requestMap.find(connection); + + if(it == requestMap.end()) { + // TODO: Error + std::cerr << "Received a packet from a unregistered connection." << std::endl; + + return; + } + + RequestInfo &requestInfo = it->second; + RequestInfo::iterator it2; + + switch(packet.getType()) { + case Net::Packet::TYPE_PING: + std::cout << "Received ping request." << std::endl; + std::cout << " Request ID: " << packet.getRequestId() << std::endl; + + connection->send(Net::Packet(Net::Packet::TYPE_PONG, packet.getRequestId(), packet.getData(), packet.getLength())); + break; + case Net::Packet::TYPE_PONG: + it2 = requestInfo.find(packet.getRequestId()); + if(it2 == requestInfo.end() || it2->second.getType() != Request::TYPE_PING) { + // TODO: Error + std::cerr << "Received an unexpected ping reply." << std::endl; + + return; + } + + requestInfo.erase(it2); + + std::cout << "Received ping reply." << std::endl; + std::cout << " Request ID: " << packet.getRequestId() << std::endl; + break; + case Net::Packet::TYPE_DISCONNECT_REQ: + connection->send(Net::Packet(Net::Packet::TYPE_DISCONNECT_REP, packet.getRequestId())); + connection->disconnect(); + break; + case Net::Packet::TYPE_DISCONNECT_REP: + it2 = requestInfo.find(packet.getRequestId()); + if(it2 == requestInfo.end() || it2->second.getType() != Request::TYPE_DISCONNECT) { + // TODO: Error + std::cerr << "Received an unexpected disconnect reply." << std::endl; + + return; + } + + requestInfo.erase(it2); + + connection->disconnect(); + } +} + +bool RequestManager::sendRequest(Net::Connection *connection, Request::Type type) { + std::map<Net::Connection*,RequestInfo>::iterator it = requestMap.find(connection); + + if(it == requestMap.end()) + return false; + + RequestInfo &requestInfo = it->second; + + unsigned short request; + do { + request = getRequestId(); + } while(requestInfo.find(request) != requestInfo.end()); + + switch(type) { + case Request::TYPE_PING: + if(!connection->send(Net::Packet(Net::Packet::TYPE_PING, request))) + return false; + + requestInfo.insert(std::make_pair(request, Request(type))); + break; + case Request::TYPE_DISCONNECT: + if(!connection->send(Net::Packet(Net::Packet::TYPE_DISCONNECT_REQ, request))) + return false; + + requestInfo.insert(std::make_pair(request, Request(type))); + } + + return true; +} + +} +} |