You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
244 lines
7.2 KiB
244 lines
7.2 KiB
/* $Id: natpmpc.c,v 1.13 2012/08/21 17:23:38 nanard Exp $ */ |
|
/* libnatpmp |
|
Copyright (c) 2007-2011, Thomas BERNARD |
|
All rights reserved. |
|
|
|
Redistribution and use in source and binary forms, with or without |
|
modification, are permitted provided that the following conditions are met: |
|
|
|
* Redistributions of source code must retain the above copyright notice, |
|
this list of conditions and the following disclaimer. |
|
* Redistributions in binary form must reproduce the above copyright notice, |
|
this list of conditions and the following disclaimer in the documentation |
|
and/or other materials provided with the distribution. |
|
* The name of the author may not be used to endorse or promote products |
|
derived from this software without specific prior written permission. |
|
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" |
|
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
|
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
|
ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE |
|
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR |
|
CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF |
|
SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS |
|
INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN |
|
CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) |
|
ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE |
|
POSSIBILITY OF SUCH DAMAGE. |
|
*/ |
|
#include <stdio.h> |
|
#include <errno.h> |
|
#include <string.h> |
|
#if defined(_MSC_VER) |
|
#if _MSC_VER >= 1400 |
|
#define strcasecmp _stricmp |
|
#else |
|
#define strcasecmp stricmp |
|
#endif |
|
#else |
|
#include <unistd.h> |
|
#endif |
|
#ifdef WIN32 |
|
#include <winsock2.h> |
|
#else |
|
#include <netinet/in.h> |
|
#include <arpa/inet.h> |
|
#endif |
|
#include "natpmp.h" |
|
|
|
void usage(FILE * out, const char * argv0) |
|
{ |
|
fprintf(out, "Usage :\n"); |
|
fprintf(out, " %s [options]\n", argv0); |
|
fprintf(out, "\tdisplay the public IP address.\n"); |
|
fprintf(out, " %s -h\n", argv0); |
|
fprintf(out, "\tdisplay this help screen.\n"); |
|
fprintf(out, " %s [options] -a <public port> <private port> <protocol> [lifetime]\n", argv0); |
|
fprintf(out, "\tadd a port mapping.\n"); |
|
fprintf(out, "\nOption available :\n"); |
|
fprintf(out, " -g ipv4address\n"); |
|
fprintf(out, "\tforce the gateway to be used as destination for NAT-PMP commands.\n"); |
|
fprintf(out, "\n In order to remove a mapping, set it with a lifetime of 0 seconds.\n"); |
|
fprintf(out, " To remove all mappings for your machine, use 0 as private port and lifetime.\n"); |
|
} |
|
|
|
/* sample code for using libnatpmp */ |
|
int main(int argc, char * * argv) |
|
{ |
|
natpmp_t natpmp; |
|
natpmpresp_t response; |
|
int r; |
|
int sav_errno; |
|
struct timeval timeout; |
|
fd_set fds; |
|
int i; |
|
int protocol = 0; |
|
uint16_t privateport = 0; |
|
uint16_t publicport = 0; |
|
uint32_t lifetime = 3600; |
|
int command = 0; |
|
int forcegw = 0; |
|
in_addr_t gateway = 0; |
|
struct in_addr gateway_in_use; |
|
|
|
#ifdef WIN32 |
|
WSADATA wsaData; |
|
int nResult = WSAStartup(MAKEWORD(2,2), &wsaData); |
|
if(nResult != NO_ERROR) |
|
{ |
|
fprintf(stderr, "WSAStartup() failed.\n"); |
|
return -1; |
|
} |
|
#endif |
|
|
|
/* argument parsing */ |
|
for(i=1; i<argc; i++) { |
|
if(argv[i][0] == '-') { |
|
switch(argv[i][1]) { |
|
case 'h': |
|
usage(stdout, argv[0]); |
|
return 0; |
|
case 'g': |
|
forcegw = 1; |
|
if(argc < i + 1) { |
|
fprintf(stderr, "Not enough arguments for option -%c\n", argv[i][1]); |
|
return 1; |
|
} |
|
gateway = inet_addr(argv[++i]); |
|
break; |
|
case 'a': |
|
command = 'a'; |
|
if(argc < i + 4) { |
|
fprintf(stderr, "Not enough arguments for option -%c\n", argv[i][1]); |
|
return 1; |
|
} |
|
i++; |
|
if(1 != sscanf(argv[i], "%hu", &publicport)) { |
|
fprintf(stderr, "%s is not a correct 16bits unsigned integer\n", argv[i]); |
|
return 1; |
|
} |
|
i++; |
|
if(1 != sscanf(argv[i], "%hu", &privateport)) { |
|
fprintf(stderr, "%s is not a correct 16bits unsigned integer\n", argv[i]); |
|
return 1; |
|
} |
|
i++; |
|
if(0 == strcasecmp(argv[i], "tcp")) |
|
protocol = NATPMP_PROTOCOL_TCP; |
|
else if(0 == strcasecmp(argv[i], "udp")) |
|
protocol = NATPMP_PROTOCOL_UDP; |
|
else { |
|
fprintf(stderr, "%s is not a valid protocol\n", argv[i]); |
|
return 1; |
|
} |
|
if(argc > i + 1) { |
|
if(1 != sscanf(argv[i+1], "%u", &lifetime)) { |
|
fprintf(stderr, "%s is not a correct 32bits unsigned integer\n", argv[i]); |
|
} else { |
|
i++; |
|
} |
|
} |
|
break; |
|
default: |
|
fprintf(stderr, "Unknown option %s\n", argv[i]); |
|
usage(stderr, argv[0]); |
|
return 1; |
|
} |
|
} else { |
|
fprintf(stderr, "Unknown option %s\n", argv[i]); |
|
usage(stderr, argv[0]); |
|
return 1; |
|
} |
|
} |
|
|
|
/* initnatpmp() */ |
|
r = initnatpmp(&natpmp, forcegw, gateway); |
|
printf("initnatpmp() returned %d (%s)\n", r, r?"FAILED":"SUCCESS"); |
|
if(r<0) |
|
return 1; |
|
|
|
gateway_in_use.s_addr = natpmp.gateway; |
|
printf("using gateway : %s\n", inet_ntoa(gateway_in_use)); |
|
|
|
/* sendpublicaddressrequest() */ |
|
r = sendpublicaddressrequest(&natpmp); |
|
printf("sendpublicaddressrequest returned %d (%s)\n", |
|
r, r==2?"SUCCESS":"FAILED"); |
|
if(r<0) |
|
return 1; |
|
|
|
do { |
|
FD_ZERO(&fds); |
|
FD_SET(natpmp.s, &fds); |
|
getnatpmprequesttimeout(&natpmp, &timeout); |
|
r = select(FD_SETSIZE, &fds, NULL, NULL, &timeout); |
|
if(r<0) { |
|
fprintf(stderr, "select()"); |
|
return 1; |
|
} |
|
r = readnatpmpresponseorretry(&natpmp, &response); |
|
sav_errno = errno; |
|
printf("readnatpmpresponseorretry returned %d (%s)\n", |
|
r, r==0?"OK":(r==NATPMP_TRYAGAIN?"TRY AGAIN":"FAILED")); |
|
if(r<0 && r!=NATPMP_TRYAGAIN) { |
|
#ifdef ENABLE_STRNATPMPERR |
|
fprintf(stderr, "readnatpmpresponseorretry() failed : %s\n", |
|
strnatpmperr(r)); |
|
#endif |
|
fprintf(stderr, " errno=%d '%s'\n", |
|
sav_errno, strerror(sav_errno)); |
|
} |
|
} while(r==NATPMP_TRYAGAIN); |
|
if(r<0) |
|
return 1; |
|
|
|
/* TODO : check that response.type == 0 */ |
|
printf("Public IP address : %s\n", inet_ntoa(response.pnu.publicaddress.addr)); |
|
printf("epoch = %u\n", response.epoch); |
|
|
|
if(command == 'a') { |
|
/* sendnewportmappingrequest() */ |
|
r = sendnewportmappingrequest(&natpmp, protocol, |
|
privateport, publicport, |
|
lifetime); |
|
printf("sendnewportmappingrequest returned %d (%s)\n", |
|
r, r==12?"SUCCESS":"FAILED"); |
|
if(r < 0) |
|
return 1; |
|
|
|
do { |
|
FD_ZERO(&fds); |
|
FD_SET(natpmp.s, &fds); |
|
getnatpmprequesttimeout(&natpmp, &timeout); |
|
select(FD_SETSIZE, &fds, NULL, NULL, &timeout); |
|
r = readnatpmpresponseorretry(&natpmp, &response); |
|
printf("readnatpmpresponseorretry returned %d (%s)\n", |
|
r, r==0?"OK":(r==NATPMP_TRYAGAIN?"TRY AGAIN":"FAILED")); |
|
} while(r==NATPMP_TRYAGAIN); |
|
if(r<0) { |
|
#ifdef ENABLE_STRNATPMPERR |
|
fprintf(stderr, "readnatpmpresponseorretry() failed : %s\n", |
|
strnatpmperr(r)); |
|
#endif |
|
return 1; |
|
} |
|
|
|
printf("Mapped public port %hu protocol %s to local port %hu " |
|
"liftime %u\n", |
|
response.pnu.newportmapping.mappedpublicport, |
|
response.type == NATPMP_RESPTYPE_UDPPORTMAPPING ? "UDP" : |
|
(response.type == NATPMP_RESPTYPE_TCPPORTMAPPING ? "TCP" : |
|
"UNKNOWN"), |
|
response.pnu.newportmapping.privateport, |
|
response.pnu.newportmapping.lifetime); |
|
printf("epoch = %u\n", response.epoch); |
|
} |
|
|
|
r = closenatpmp(&natpmp); |
|
printf("closenatpmp() returned %d (%s)\n", r, r==0?"SUCCESS":"FAILED"); |
|
if(r<0) |
|
return 1; |
|
|
|
return 0; |
|
} |
|
|
|
|