1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162 |
- #include "udpsocket.hpp"
- udpsocket::udpsocket(int port) : m_port(port){
- if ((s = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)) < 0) {
- throw std::logic_error(std::string("Socket creation failed: ") + strerror(errno));
- }
- std::memset((char*)&addr, 0, sizeof(addr));
- addr.sin_family = AF_INET;
- addr.sin_addr.s_addr = htonl(INADDR_ANY);
- addr.sin_port = htons(this->port());
- if (bind(s, (struct sockaddr*)&addr, sizeof(addr)) < 0) {
- throw std::logic_error(std::string("Socket binding failed: ") + strerror(errno));
- }
- }
- udpsocket::udpsocket(udpsocket&& o) : addr(o.addr),s(o.s),m_port(o.m_port) {
- o.s = 0;
- o.m_port = 0;
- }
- udpsocket& udpsocket::operator=(udpsocket&& o){
- addr = o.addr;
- s = o.s;
- m_port = o.m_port;
- o.s = 0;
- o.m_port = 0;
- return *this;
- }
- void udpsocket::write(const std::string& msg, const std::string& dest, int port)const{
- std::vector<char> _msg(msg.begin(), msg.end());
- write(_msg, dest, port);
- }
- void udpsocket::write(const std::vector<char>& msg, const std::string& dest, int port)const{
- hostent *server_host;
- errno = 0;
- server_host = gethostbyname(dest.c_str());
- if(errno){
- throw std::logic_error("Could not resolve hostname " + dest + ": " + strerror(errno));
- }
-
- struct sockaddr_in server_addr;
- server_addr.sin_family = AF_INET;
- memcpy(&server_addr.sin_addr, server_host->h_addr,
- sizeof(struct in_addr)); server_addr.sin_port = htons(port);
-
- if(sendto(s, msg.data(), msg.size(), 0,(const sockaddr*)&server_addr, sizeof(server_addr)) < 0){
- throw std::logic_error(std::string("Could not send packet: ") + strerror(errno));
- }
- }
- std::vector<char> udpsocket::receive()const{
- std::vector<char> ret(1024);
- int l;
- if ((l = read(s, ret.data(), 1024)) <= 0) {
- throw std::logic_error(std::string("Could not receive packet: ") + strerror(errno));
- }
- ret.resize(l);
- return ret;
- }
- void udpsocket::close(){
- shutdown(s, 2);
- }
- int udpsocket::port()const {
- return this->m_port;
- }
|