owlps/owlps-positioner/inputudpsocket.cc

235 lines
6.5 KiB
C++

/*
* This file is part of the Owl Positioning System (OwlPS) project.
* It is subject to the copyright notice and license terms in the
* COPYRIGHT.t2t file found in the top-level directory of this
* distribution and at
* https://code.lm7.fr/mcy/owlps/src/master/COPYRIGHT.t2t
* No part of the OwlPS Project, including this file, may be copied,
* modified, propagated, or distributed except according to the terms
* contained in the COPYRIGHT.t2t file; the COPYRIGHT.t2t file must be
* distributed along with this file, either separately or by replacing
* this notice by the COPYRIGHT.t2t file's contents.
*/
#include "inputudpsocket.hh"
#include "posexcept.hh"
#include "stock.hh"
#include "configuration.hh"
#include <owlps.h>
#include <iostream>
#include <cstdio> // For perror()
#include <unistd.h> // For close()
#include <arpa/inet.h> // For inet_ntop()
#include <unordered_map>
using namespace std ;
/* *** Constructors *** */
InputUDPSocket::InputUDPSocket(const uint_fast16_t _port):
listening_port(_port)
{
if (! init_socket())
throw error_opening_input_file("UDP socket") ;
}
InputUDPSocket::~InputUDPSocket()
{
close_socket() ;
}
/* *** Operations *** */
/**
* @returns `true` if the socket were successfully opened.
* @returns `false` in case of error.
*/
bool InputUDPSocket::init_socket()
{
sockfd = owl_create_udp_listening_socket(listening_port) ;
return sockfd >= 0 ;
}
/**
* Normally, the socket is closed automatically by the destructor. Use
* this if you want to close the socket prematurely.
* #sockfd is set to -1, even in case of error.
* @returns `true` if the socket were successfully closed or were not
* opened.
* @returns `false` in case of error.
*/
bool InputUDPSocket::close_socket()
{
if (sockfd >= 0)
{
if (close(sockfd))
{
perror("Cannot close UDP socket") ;
return false ;
}
sockfd = -1 ;
}
return true ;
}
/**
* This function fills the current Request from the UDP input socket.
*
* @returns `true` if #current_request was correctly filled.
* @returns `false` in case of error (socket not opened, bad packet
* format).
*/
bool InputUDPSocket::fill_current_request()
{
if (eof())
return false ;
// Read the main request data
owl_request request ;
ssize_t nread = recvfrom(sockfd, &request, sizeof(request), 0,
nullptr, nullptr) ;
if (nread <= 0)
{
cerr << "InputUDPSocket: No request received!" << endl ;
return false ;
}
// Endianess conversions
request.nb_packets = ntohs(request.nb_packets) ;
owl_ntoh_timestamp(&request.request_time) ;
request.nb_info = ntohs(request.nb_info) ;
request.x_position = owl_ntohf(request.x_position) ;
request.y_position = owl_ntohf(request.y_position) ;
request.z_position = owl_ntohf(request.z_position) ;
// Mobile MAC
string mac_mobile(
owl_mac_bytes_to_string(request.mobile_mac_addr_bytes)) ;
PosUtil::to_upper(mac_mobile) ;
// Mobile IP
char ip_mobile_str[INET_ADDRSTRLEN] ;
inet_ntop(AF_INET, &request.mobile_ip_addr_bytes,
ip_mobile_str, INET_ADDRSTRLEN) ;
string ip_mobile(ip_mobile_str) ;
PosUtil::to_upper(ip_mobile) ;
// Display request's info
if (Configuration::is_configured("verbose"))
cerr
<< "*** Request received from the aggregator ***"
<< "\n\tType: " << static_cast<uint_fast16_t>(request.type)
<< "\n\tNumber of packets: " << request.nb_packets
<< "\n\tMobile MAC: " << mac_mobile
<< "\n\tMobile IP: " << ip_mobile
<< "\n\tRequest timestamp: " << Timestamp(request.request_time)
<< "\n\tPosition X: " << request.x_position
<< "\n\tPosition Y: " << request.y_position
<< "\n\tPosition Z: " << request.z_position
<< "\n\tDirection: "
<< static_cast<uint_fast16_t>(request.direction)
<< "\n\tNumber of SS: " << request.nb_info
<< '\n' ;
// Mobile MAC (continued)
if (! Configuration::bool_value("positioning.accept-new-mobiles") &&
! Stock::mobile_exists(mac_mobile))
{
if (Configuration::is_configured("verbose"))
cerr << "Dropping request from unknown mobile "
<< mac_mobile << ".\n" ;
return false ;
}
const Mobile &mobile = Stock::find_create_mobile(mac_mobile) ;
// Update the mobile's IP address:
const_cast<Mobile&>(mobile).set_ip_addr(ip_mobile) ;
current_request->set_mobile(&mobile) ;
// Request type
current_request->set_type(request.type) ;
// Number of packets
current_request->set_nb_packets(request.nb_packets) ;
// Timestamp
current_request->set_time_sent(Timestamp(request.request_time)) ;
// Position
Point3D position(request.x_position,
request.y_position,
request.z_position) ;
// Read the {MAC_CP;SS} couples (request_info)
unordered_map<string, Measurement> measurements ;
owl_request_info request_info ;
for (int i = 0 ; i < request.nb_info ; ++i)
{
nread = recvfrom(sockfd, &request_info, sizeof(request_info), 0,
nullptr, nullptr) ;
if (nread <= 0)
{
cerr << "No request info received!" << endl ;
return false ;
}
pkt_id_t packet_id = ntohs(request_info.packet_id) ;
string mac_cp(
owl_mac_bytes_to_string(request_info.cp_mac_addr_bytes)) ;
PosUtil::to_upper(mac_cp) ;
owl_ntoh_timestamp(&request_info.capture_time) ;
Timestamp capture_time(request_info.capture_time) ;
ss_t ss = request_info.ss_dbm ;
if (Configuration::is_configured("verbose"))
cerr
<< "\t* Packet received from the aggregator:"
<< "\n\t\tPacket number: " << packet_id
<< "\n\t\tCP MAC: " << mac_cp
<< "\n\t\tCapture timestamp: " << capture_time
<< "\n\t\tSignal: "
<< ss << " dBm"
<< '\n' ;
if (! Configuration::bool_value("positioning.accept-new-cps") &&
! Stock::cp_exists(mac_cp))
{
if (Configuration::is_configured("verbose"))
cerr << "Dropping packet from unknown CP "
<< mac_cp << ".\n" ;
continue ;
}
const CapturePoint &cp = Stock::find_create_cp(mac_cp) ;
measurements[mac_cp].set_cp(&cp) ;
measurements[mac_cp].add_ss(packet_id, ss) ;
}
if (measurements.empty())
return false ;
current_request->set_measurements(measurements) ;
// Calibration request?
fill_calibration_request_data(
mac_mobile, position, request.direction, request.type) ;
return true ;
}