#include <stdlib.h>
#include <sys/types.h>
#include <stdio.h>
#include <sys/socket.h>
#include <netinet/
in.h>
#include <
string.h>
#include <getopt.h>
void get_remote_ip_and_port(
struct sockaddr_in* s_add,
int* ip,
int* port) {
if (!s_add || !ip || !port)
return;
*port = ntohs(s_add->sin_port);
*ip = s_add->sin_addr.s_addr;
}
void get_local_ip_and_port(
int fd,
int* ip,
int* port) {
if (fd <= 0 || !ip || !port)
return;
struct sockaddr_in s_add;
int sin_size =
sizeof(s_add);
if(getsockname(fd, (
struct sockaddr *)&s_add, &sin_size) == 0 && s_add.sin_family == AF_INET && sin_size ==
sizeof(s_add)) {
*port = ntohs(s_add.sin_port);
*ip = s_add.sin_addr.s_addr;
}
}
char* get_ip_str(
int ip,
char* buf) {
inet_ntop(AF_INET, &ip, buf, INET_ADDRSTRLEN);
return buf;
}
void usage(
char* argv0) {
fprintf(stdout,
"Usage: %s [-p <port>] [-o output_file] [-h]\n\n"
" -p <port> local port used to listen, by default, 8888\n"
" -o <output_file> output file used to record the data received from client, by default, dump to ./x.in\n"
" -h print this usage\n"
"\n", argv0);
}
int main(
int argc,
char* argv[]) {
int sfp = 0;
int nfp = 0;
struct sockaddr_in s_add, c_add;
int sin_size = 0;
int server_port = 8888;
int remote_port = 0;
int remote_ip = 0;
char remote_ip_str[INET_ADDRSTRLEN];
int local_port = 0;
int local_ip = 0;
char local_ip_str[INET_ADDRSTRLEN];
char output_file[256];
sprintf(output_file, "%s", "./x.in");
FILE* ofp = 0;
char c = '\0';
struct option long_options[] = {
{"help", 0, 0, 'h'},
{0, 0, 0, 0}
};
while (-1 != (c = getopt_long(argc, argv, ":p:o:h", long_options, NULL))) {
switch (c) {
case 'p':
server_port = atoi(optarg);
break;
case 'o':
sprintf(output_file, "%s", optarg);
break;
case 'h':
case '?':
//an option character that was not in optstring
case ':':
//an option with a missing argument
default:
usage(argv[0]);
return -1;
}
}
sfp = socket(AF_INET, SOCK_STREAM, 0);
if (-1 == sfp) {
printf("socket fail ! \n");
return -1;
}
printf("socket ok !\n");
bzero(&s_add,
sizeof(
struct sockaddr_in));
s_add.sin_family = AF_INET;
s_add.sin_addr.s_addr = htonl(INADDR_ANY);
s_add.sin_port = htons(server_port);
if (-1 == bind(sfp, (
struct sockaddr *) (&s_add),
sizeof(
struct sockaddr))) {
printf("bind fail !\n");
return -1;
}
printf("bind to %d ok!\n", server_port);
if (-1 == listen(sfp, 5)) {
printf("listen fail !\n");
return -1;
}
printf("listen ok\n\n");
ofp = fopen(output_file, "wrb");
while (1) {
sin_size =
sizeof(
struct sockaddr_in);
nfp = accept(sfp, (
struct sockaddr *) (&c_add), &sin_size);
if (-1 == nfp) {
printf("accept fail !\n");
return -1;
}
printf("accept ok!\n");
get_remote_ip_and_port(&c_add, &remote_ip, &remote_port);
get_local_ip_and_port(nfp, &local_ip, &local_port);
printf("establish connect %s:%d -> %s:%d\n", get_ip_str(remote_ip, remote_ip_str), remote_port, get_ip_str(local_ip, local_ip_str), local_port);
fprintf(ofp, "----------------- establish connect %s:%d -> %s:%d -----------------{\n", get_ip_str(remote_ip, remote_ip_str), remote_port, get_ip_str(local_ip, local_ip_str), local_port);
char buffer[1024];
int recbytes = 0;
int total_bytes = 0;
while ((recbytes = read(nfp, buffer, 1024)) > 0) {
total_bytes += recbytes;
fwrite(buffer, 1, recbytes, ofp);
}
fprintf(ofp, "\n}---------------- receive %d bytes ------------------\n\n", total_bytes);
printf("receive over!\n\n");
fflush(stdout);
fflush(ofp);
close(nfp);
}
fclose(ofp);
close(sfp);
return 0;
}