X-Git-Url: https://git.syn-net.org/debian/?p=anytun.git;a=blobdiff_plain;f=src%2FpacketSource.cpp;h=bd3590d2e9cbfe0ab3ba960285e8fade0ec07d97;hp=a5443ad67736fbc300603ee4cca29c740212f95d;hb=ef0cacf2508418915d3f64b04003be3c13fed3cc;hpb=ece844834d2cecc028ce81ca283f5d441088580e diff --git a/src/packetSource.cpp b/src/packetSource.cpp index a5443ad..bd3590d 100644 --- a/src/packetSource.cpp +++ b/src/packetSource.cpp @@ -11,7 +11,7 @@ * tunneling and relaying of packets of any protocol. * * - * Copyright (C) 2007-2009 Othmar Gsenger, Erwin Nindl, + * Copyright (C) 2007-2009 Othmar Gsenger, Erwin Nindl, * Christian Pointner * * This file is part of Anytun. @@ -55,11 +55,11 @@ UDPPacketSource::UDPPacketSource(std::string localaddr, std::string port) UDPPacketSource::~UDPPacketSource() { std::list::iterator it = sockets_.begin(); - for(;it != sockets_.end(); ++it) { -/// this might be a needed by the receiver thread, TODO cleanup -// delete[](it->buf_); -// delete(it->sem_); -// delete(it->sock_); + for(; it != sockets_.end(); ++it) { + /// this might be a needed by the receiver thread, TODO cleanup + // delete[](it->buf_); + // delete(it->sem_); + // delete(it->sock_); } } @@ -74,13 +74,15 @@ void UDPPacketSource::onResolve(PacketSourceResolverIt& it) sock.len_ = 0; sock.sem_ = NULL; sock.sock_ = new proto::socket(io_service_); - if(!sock.sock_) + if(!sock.sock_) { AnytunError::throwErr() << "memory error"; + } sock.sock_->open(e.protocol()); #ifndef _MSC_VER - if(e.protocol() == proto::v6()) + if(e.protocol() == proto::v6()) { sock.sock_->set_option(boost::asio::ip::v6_only(true)); + } #endif sock.sock_->bind(e); sockets_.push_back(sock); @@ -88,15 +90,16 @@ void UDPPacketSource::onResolve(PacketSourceResolverIt& it) it++; } - // prepare multi-socket recv + // prepare multi-socket recv if(sockets_.size() > 1) { std::list::iterator it = sockets_.begin(); - for(;it != sockets_.end(); ++it) { + for(; it != sockets_.end(); ++it) { it->len_ = MAX_PACKET_LENGTH; - it->buf_ = new u_int8_t[it->len_]; - if(!it->buf_) + it->buf_ = new uint8_t[it->len_]; + if(!it->buf_) { AnytunError::throwErr() << "memory error"; - + } + it->sem_ = new Semaphore(); if(!it->sem_) { delete[](it->buf_); @@ -125,7 +128,7 @@ void UDPPacketSource::recv_thread(std::list::iterator it) result.it_ = it; for(;;) { it->sem_->down(); - result.len_ = static_cast(it->sock_->receive_from(boost::asio::buffer(it->buf_, it->len_), result.remote_)); + result.len_ = static_cast(it->sock_->receive_from(boost::asio::buffer(it->buf_, it->len_), result.remote_)); { Lock lock(thread_result_mutex_); thread_result_queue_.push(result); @@ -134,10 +137,11 @@ void UDPPacketSource::recv_thread(std::list::iterator it) } } -u_int32_t UDPPacketSource::recv(u_int8_t* buf, u_int32_t len, PacketSourceEndpoint& remote) +uint32_t UDPPacketSource::recv(uint8_t* buf, uint32_t len, PacketSourceEndpoint& remote) { - if(sockets_.size() == 1) - return static_cast(sockets_.front().sock_->receive_from(boost::asio::buffer(buf, len), remote)); + if(sockets_.size() == 1) { + return static_cast(sockets_.front().sock_->receive_from(boost::asio::buffer(buf, len), remote)); + } thread_result_sem_.down(); ThreadResult result; @@ -154,10 +158,10 @@ u_int32_t UDPPacketSource::recv(u_int8_t* buf, u_int32_t len, PacketSourceEndpoi return len; } -void UDPPacketSource::send(u_int8_t* buf, u_int32_t len, PacketSourceEndpoint remote) +void UDPPacketSource::send(uint8_t* buf, uint32_t len, PacketSourceEndpoint remote) { std::list::iterator it = sockets_.begin(); - for(;it != sockets_.end(); ++it) { + for(; it != sockets_.end(); ++it) { if(it->sock_->local_endpoint().protocol() == remote.protocol()) { it->sock_->send_to(boost::asio::buffer(buf, len), remote); return;