Added get_port() method to socket_t to learn the listening port.
This commit is contained in:
@@ -1,5 +1,5 @@
|
||||
/*
|
||||
* Copyright (C) 2006-2010 Tobias Brunner
|
||||
* Copyright (C) 2006-2012 Tobias Brunner
|
||||
* Copyright (C) 2005-2010 Martin Willi
|
||||
* Copyright (C) 2006 Daniel Roethlisberger
|
||||
* Copyright (C) 2005 Jan Hutter
|
||||
@@ -67,6 +67,14 @@ struct socket_t {
|
||||
*/
|
||||
status_t (*send) (socket_t *this, packet_t *packet);
|
||||
|
||||
/**
|
||||
* Get the port this socket is listening on.
|
||||
*
|
||||
* @param nat_t TRUE to get the port used to float in case of NAT-T
|
||||
* @return the port
|
||||
*/
|
||||
u_int16_t (*get_port) (socket_t *this, bool nat_t);
|
||||
|
||||
/**
|
||||
* Destroy a socket implementation.
|
||||
*/
|
||||
|
||||
@@ -1,5 +1,5 @@
|
||||
/*
|
||||
* Copyright (C) 2010 Tobias Brunner
|
||||
* Copyright (C) 2010-2012 Tobias Brunner
|
||||
* Hochschule fuer Technik Rapperswil
|
||||
* Copyright (C) 2010 Martin Willi
|
||||
* Copyright (C) 2010 revosec AG
|
||||
@@ -89,6 +89,19 @@ METHOD(socket_manager_t, sender, status_t,
|
||||
return status;
|
||||
}
|
||||
|
||||
METHOD(socket_manager_t, get_port, u_int16_t,
|
||||
private_socket_manager_t *this, bool nat_t)
|
||||
{
|
||||
u_int16_t port = 0;
|
||||
this->lock->read_lock(this->lock);
|
||||
if (this->socket)
|
||||
{
|
||||
port = this->socket->get_port(this->socket, nat_t);
|
||||
}
|
||||
this->lock->unlock(this->lock);
|
||||
return port;
|
||||
}
|
||||
|
||||
static void create_socket(private_socket_manager_t *this)
|
||||
{
|
||||
socket_constructor_t create;
|
||||
@@ -153,6 +166,7 @@ socket_manager_t *socket_manager_create()
|
||||
.public = {
|
||||
.send = _sender,
|
||||
.receive = _receiver,
|
||||
.get_port = _get_port,
|
||||
.add_socket = _add_socket,
|
||||
.remove_socket = _remove_socket,
|
||||
.destroy = _destroy,
|
||||
|
||||
@@ -1,5 +1,5 @@
|
||||
/*
|
||||
* Copyright (C) 2010 Tobias Brunner
|
||||
* Copyright (C) 2010-2012 Tobias Brunner
|
||||
* Hochschule fuer Technik Rapperswil
|
||||
* Copyright (C) 2010 Martin Willi
|
||||
* Copyright (C) 2010 revosec AG
|
||||
@@ -52,6 +52,14 @@ struct socket_manager_t {
|
||||
*/
|
||||
status_t (*send) (socket_manager_t *this, packet_t *packet);
|
||||
|
||||
/**
|
||||
* Get the port the registered socket is listening on.
|
||||
*
|
||||
* @param nat_t TRUE to get the port used to float in case of NAT-T
|
||||
* @return the port, or 0, if no socket is registered
|
||||
*/
|
||||
u_int16_t (*get_port) (socket_manager_t *this, bool nat_t);
|
||||
|
||||
/**
|
||||
* Register a socket constructor.
|
||||
*
|
||||
|
||||
Reference in New Issue
Block a user