Importation du code depuis GitHub
This commit is contained in:
gasc
commit
fca1caf5ea
4 changed files with 1043 additions and 0 deletions
3
README.md
Normal file
3
README.md
Normal file
|
|
@ -0,0 +1,3 @@
|
||||||
|
# Tsocks (Computer science lab)
|
||||||
|
Here I share my own implementation of [Tsocks](http://tsocks.sourceforge.net/) in C carried out as part of the computer science work at INSA Toulouse.
|
||||||
|
|
||||||
274
tsock_v1_gasc_m.c
Normal file
274
tsock_v1_gasc_m.c
Normal file
|
|
@ -0,0 +1,274 @@
|
||||||
|
/* librairie standard ... */
|
||||||
|
#include <stdlib.h>
|
||||||
|
/* pour getopt */
|
||||||
|
#include <unistd.h>
|
||||||
|
/* déclaration des types de base */
|
||||||
|
#include <sys/types.h>
|
||||||
|
/* constantes relatives aux domaines, types et protocoles */
|
||||||
|
#include <sys/socket.h>
|
||||||
|
/* constantes et structures propres au domaine UNIX */
|
||||||
|
#include <sys/un.h>
|
||||||
|
/* constantes et structures propres au domaine INTERNET */
|
||||||
|
#include <netinet/in.h>
|
||||||
|
/* structures retournées par les fonctions de gestion de la base de
|
||||||
|
données du réseau */
|
||||||
|
#include <netdb.h>
|
||||||
|
/* pour les entrées/sorties */
|
||||||
|
#include <stdio.h>
|
||||||
|
/* pour la gestion des erreurs */
|
||||||
|
#include <errno.h>
|
||||||
|
|
||||||
|
/*
|
||||||
|
TP AIPS
|
||||||
|
Version 1 - Fonctionnelle
|
||||||
|
*/
|
||||||
|
|
||||||
|
///////////////////////////////////////////////////////////
|
||||||
|
|
||||||
|
// Fonctions d'erreurs
|
||||||
|
/// Afficher l'usage et quitte
|
||||||
|
void err_args() {
|
||||||
|
printf("[!] Usage: cmd [-u][-p|-s][-n ##][-l ##]\n");
|
||||||
|
exit(EXIT_FAILURE);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Affiche erreur socket
|
||||||
|
void err_socks(){
|
||||||
|
printf("\n[-] Erreur lors de la création/affectation du socket");
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Affiche erreur connexion
|
||||||
|
void err_conn(){
|
||||||
|
printf("\n[-] Erreur lors de la connexion serveur/client");
|
||||||
|
}
|
||||||
|
|
||||||
|
///////////////////////////////////////////////////////////
|
||||||
|
|
||||||
|
|
||||||
|
// Given functions
|
||||||
|
void construire_message(char *message, char motif, int lg) {
|
||||||
|
int i;
|
||||||
|
for (i=0;i<lg;i++) message[i] = motif;
|
||||||
|
}
|
||||||
|
|
||||||
|
void afficher_message(char *message, int lg) {
|
||||||
|
int i;
|
||||||
|
printf("[+] Message construit : ");
|
||||||
|
for (i=0;i<lg;i++) printf("%c", message[i]);
|
||||||
|
printf("\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
///////////////////////////UDP/////////////////////////////
|
||||||
|
|
||||||
|
int source_UDP(char *HOSTNAME, int PORT, int nb_message, int l_msg) {
|
||||||
|
// Création du socket UDP
|
||||||
|
printf("[+] Création du socket local...");
|
||||||
|
int s_udp = socket(AF_INET,SOCK_DGRAM, 0);
|
||||||
|
|
||||||
|
if (s_udp == -1){
|
||||||
|
err_socks();
|
||||||
|
return 1;
|
||||||
|
} else printf("Ok!\n");
|
||||||
|
|
||||||
|
|
||||||
|
// Création du socket distant
|
||||||
|
printf("[+] Création du socket distant...");
|
||||||
|
struct sockaddr_in servaddr;
|
||||||
|
memset(&servaddr, 0, sizeof(servaddr));
|
||||||
|
struct hostent *hp ;
|
||||||
|
|
||||||
|
// Résolution du nom
|
||||||
|
if ((hp = gethostbyname(HOSTNAME)) == NULL) {
|
||||||
|
err_conn();
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Creation de la struct du socket distant
|
||||||
|
servaddr.sin_family = AF_INET;
|
||||||
|
memcpy((char*)&(servaddr.sin_addr.s_addr), hp->h_addr, hp->h_length);
|
||||||
|
servaddr.sin_port = PORT;
|
||||||
|
printf("Ok!\n");
|
||||||
|
|
||||||
|
printf("[+] Envoi des messages à %s:%d...\n", HOSTNAME, PORT);
|
||||||
|
|
||||||
|
// TEST
|
||||||
|
char *pmesg = malloc(l_msg*sizeof(char));
|
||||||
|
char c = 'a';
|
||||||
|
|
||||||
|
// Création des messages à envoyer et envoi
|
||||||
|
for (int i = 1; i <= nb_message; i++){
|
||||||
|
construire_message(pmesg, c, l_msg);
|
||||||
|
afficher_message(pmesg, l_msg);
|
||||||
|
sendto(s_udp,
|
||||||
|
pmesg,
|
||||||
|
strlen(pmesg),
|
||||||
|
0,
|
||||||
|
(const struct sockaddr *) &servaddr,
|
||||||
|
sizeof(servaddr)
|
||||||
|
);
|
||||||
|
c++;
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int puit_UDP(int PORT, int nb_message, int l_msg) {
|
||||||
|
// Création du socket UDP
|
||||||
|
printf("[+] Création du socket local...");
|
||||||
|
int s_udp = socket(AF_INET,SOCK_DGRAM, 0);
|
||||||
|
|
||||||
|
if (s_udp == -1){
|
||||||
|
err_socks();
|
||||||
|
return 1;
|
||||||
|
} else printf("Ok!\n");
|
||||||
|
|
||||||
|
struct sockaddr_in servaddr, cliaddr;
|
||||||
|
memset(&servaddr, 0, sizeof(servaddr));
|
||||||
|
memset(&cliaddr, 0, sizeof(cliaddr));
|
||||||
|
|
||||||
|
servaddr.sin_family = AF_INET;
|
||||||
|
servaddr.sin_addr.s_addr = INADDR_ANY;
|
||||||
|
servaddr.sin_port = PORT;
|
||||||
|
|
||||||
|
if ( bind(s_udp, (const struct sockaddr *)&servaddr, sizeof(servaddr)) < 0 ) {
|
||||||
|
err_socks();
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
printf("[+] Bind ok !\n");
|
||||||
|
|
||||||
|
int n;
|
||||||
|
int len;
|
||||||
|
int msg_count = 0;
|
||||||
|
char buffer[l_msg];
|
||||||
|
|
||||||
|
printf("[+] Waiting for messages...\n");
|
||||||
|
|
||||||
|
while (msg_count >= nb_message || nb_message != -1){
|
||||||
|
n = recvfrom(s_udp, (char *)buffer, l_msg,
|
||||||
|
MSG_WAITALL, (struct sockaddr *) &cliaddr,
|
||||||
|
&len);
|
||||||
|
buffer[n] = '\0';
|
||||||
|
printf("[>] Reception du message n°%d : %s\n", msg_count, buffer);
|
||||||
|
|
||||||
|
n = 0;
|
||||||
|
msg_count++;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
////////////////////////////TCP/////////////////////////////
|
||||||
|
|
||||||
|
int source_TCP(char *HOSTNAME, int PORT, int nb_message, int l_msg){
|
||||||
|
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
int puit_TCP(int PORT, int nb_message, int l_msg){
|
||||||
|
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
///////////////////////////////////////////////////////////
|
||||||
|
|
||||||
|
/// Main entry
|
||||||
|
int main (int argc, char **argv)
|
||||||
|
{
|
||||||
|
int c;
|
||||||
|
extern char *optarg;
|
||||||
|
extern int optind;
|
||||||
|
int nb_message = -1; /* Nb de messages à envoyer ou à recevoir, par défaut : 10 en émission, infini en réception */
|
||||||
|
int source = -1; /* 0 = puits, 1 = source */
|
||||||
|
int tcp = 1; /* 0 = UDP, 1 = TCP */
|
||||||
|
int l_msg = 30; /* longueur du message, en octet. Defaut : 30. */
|
||||||
|
|
||||||
|
while ((c = getopt(argc, argv, "pn:sul:")) != -1) {
|
||||||
|
switch (c) {
|
||||||
|
case 'p':
|
||||||
|
if (source == 1) {
|
||||||
|
err_args();
|
||||||
|
}
|
||||||
|
source = 0;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case 's':
|
||||||
|
if (source == 0) {
|
||||||
|
err_args();
|
||||||
|
}
|
||||||
|
source = 1;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case 'n':
|
||||||
|
nb_message = atoi(optarg);
|
||||||
|
break;
|
||||||
|
|
||||||
|
case 'u':
|
||||||
|
tcp = 0;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case 'l':
|
||||||
|
l_msg = atoi(optarg);
|
||||||
|
|
||||||
|
default:
|
||||||
|
err_args();
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
int PORT = atoi(argv[argc-1]);
|
||||||
|
//PORT = htons(PORT);
|
||||||
|
|
||||||
|
if (source == -1) {
|
||||||
|
err_args();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (tcp)
|
||||||
|
printf("[+] On utilise le protocol TCP\n");
|
||||||
|
else
|
||||||
|
printf("[+] On utilise le protocol UDP\n");
|
||||||
|
|
||||||
|
|
||||||
|
char* HOSTNAME = "";
|
||||||
|
if (source == 1){
|
||||||
|
printf("[+] On est dans la source\n");
|
||||||
|
HOSTNAME = argv[argc-2];
|
||||||
|
}
|
||||||
|
else
|
||||||
|
printf("[+] On est dans le puit\n");
|
||||||
|
|
||||||
|
if (nb_message != -1) {
|
||||||
|
if (source == 1)
|
||||||
|
printf("[+] Nombre de tampons à envoyer : %d\n", nb_message);
|
||||||
|
else
|
||||||
|
printf("[+] Nombre de tampons à recevoir : %d\n", nb_message);
|
||||||
|
} else {
|
||||||
|
if (source == 1) {
|
||||||
|
nb_message = 10 ;
|
||||||
|
printf("[~] Nombre de tampons à envoyer = 10 par défaut\n");
|
||||||
|
}
|
||||||
|
else
|
||||||
|
printf("[+] Nombre de tampons à envoyer = infini\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (l_msg == 30)
|
||||||
|
printf("[~] Longueur des messages à envoyer/recevoir : 30 par défaut\n");
|
||||||
|
else
|
||||||
|
printf("[+] Nombre de tampons à envoyer/recevoir : %d \n", l_msg);
|
||||||
|
|
||||||
|
int s = 1;
|
||||||
|
|
||||||
|
// Choose the right function
|
||||||
|
if (tcp){
|
||||||
|
if (source) s = source_TCP(HOSTNAME, PORT, nb_message, l_msg);
|
||||||
|
else s = puit_TCP(PORT, nb_message, l_msg);
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
if (source) s = source_UDP(HOSTNAME, PORT, nb_message, l_msg);
|
||||||
|
else s = puit_UDP(PORT, nb_message, l_msg);
|
||||||
|
}
|
||||||
|
|
||||||
|
return s;
|
||||||
|
}
|
||||||
|
|
||||||
379
tsock_v2_gasc_m.c
Normal file
379
tsock_v2_gasc_m.c
Normal file
|
|
@ -0,0 +1,379 @@
|
||||||
|
/* librairie standard ... */
|
||||||
|
#include <stdlib.h>
|
||||||
|
/* pour getopt */
|
||||||
|
#include <unistd.h>
|
||||||
|
/* déclaration des types de base */
|
||||||
|
#include <sys/types.h>
|
||||||
|
/* constantes relatives aux domaines, types et protocoles */
|
||||||
|
#include <sys/socket.h>
|
||||||
|
/* constantes et structures propres au domaine UNIX */
|
||||||
|
#include <sys/un.h>
|
||||||
|
/* constantes et structures propres au domaine INTERNET */
|
||||||
|
#include <netinet/in.h>
|
||||||
|
/* structures retournées par les fonctions de gestion de la base de
|
||||||
|
données du réseau */
|
||||||
|
#include <netdb.h>
|
||||||
|
/* pour les entrées/sorties */
|
||||||
|
#include <stdio.h>
|
||||||
|
/* pour la gestion des erreurs */
|
||||||
|
#include <errno.h>
|
||||||
|
|
||||||
|
/*
|
||||||
|
TP AIPS
|
||||||
|
Version 2 - FONCTIONNELLE
|
||||||
|
*/
|
||||||
|
|
||||||
|
///////////////////////////////////////////////////////////
|
||||||
|
|
||||||
|
// Fonctions d'erreurs
|
||||||
|
/// Afficher l'usage et quitte
|
||||||
|
void err_args() {
|
||||||
|
printf("[!] Usage: cmd [-u][-p|-s][-n ##][-l ##]\n");
|
||||||
|
exit(EXIT_FAILURE);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Affiche erreur socket
|
||||||
|
void err_socks(){
|
||||||
|
printf("\n[-] Erreur lors de la création/affectation du socket\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Affiche erreur connexion
|
||||||
|
void err_conn(){
|
||||||
|
printf("\n[-] Erreur lors de la connexion serveur/client\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
///////////////////////////////////////////////////////////
|
||||||
|
|
||||||
|
|
||||||
|
// Given functions
|
||||||
|
void construire_message(char *message, char motif, int lg) {
|
||||||
|
int i;
|
||||||
|
for (i=0;i<lg;i++) message[i] = motif;
|
||||||
|
}
|
||||||
|
|
||||||
|
void afficher_message(char *message, int lg) {
|
||||||
|
int i;
|
||||||
|
printf("[+] Message construit : ");
|
||||||
|
for (i=0;i<lg;i++) printf("%c", message[i]);
|
||||||
|
printf("\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
///////////////////////////UDP/////////////////////////////
|
||||||
|
|
||||||
|
int source_UDP(char *HOSTNAME, int PORT, int nb_message, int l_msg) {
|
||||||
|
// Création du socket UDP
|
||||||
|
printf("[+] Création du socket local...");
|
||||||
|
int s_udp = socket(AF_INET,SOCK_DGRAM, 0);
|
||||||
|
|
||||||
|
if (s_udp == -1){
|
||||||
|
err_socks();
|
||||||
|
return 1;
|
||||||
|
} else printf("Ok!\n");
|
||||||
|
|
||||||
|
|
||||||
|
// Création du socket distant
|
||||||
|
printf("[+] Création du socket distant...");
|
||||||
|
struct sockaddr_in servaddr;
|
||||||
|
memset(&servaddr, 0, sizeof(servaddr));
|
||||||
|
struct hostent *hp ;
|
||||||
|
|
||||||
|
// Résolution du nom
|
||||||
|
if ((hp = gethostbyname(HOSTNAME)) == NULL) {
|
||||||
|
err_conn();
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Creation de la struct du socket distant
|
||||||
|
servaddr.sin_family = AF_INET;
|
||||||
|
memcpy((char*)&(servaddr.sin_addr.s_addr), hp->h_addr, hp->h_length);
|
||||||
|
servaddr.sin_port = PORT;
|
||||||
|
printf("Ok!\n");
|
||||||
|
|
||||||
|
printf("[+] Envoi des messages à %s:%d...\n", HOSTNAME, PORT);
|
||||||
|
|
||||||
|
// TEST
|
||||||
|
char *pmesg = malloc(l_msg*sizeof(char));
|
||||||
|
char c = 'a';
|
||||||
|
|
||||||
|
// Création des messages à envoyer et envoi
|
||||||
|
for (int i = 1; i <= nb_message; i++){
|
||||||
|
construire_message(pmesg, c, l_msg);
|
||||||
|
afficher_message(pmesg, l_msg);
|
||||||
|
sendto(s_udp,
|
||||||
|
pmesg,
|
||||||
|
strlen(pmesg),
|
||||||
|
0,
|
||||||
|
(const struct sockaddr *) &servaddr,
|
||||||
|
sizeof(servaddr)
|
||||||
|
);
|
||||||
|
c++;
|
||||||
|
if (c >= '{') c = 'a';
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int puit_UDP(int PORT, int nb_message, int l_msg) {
|
||||||
|
// Création du socket UDP
|
||||||
|
printf("[+] Création du socket local...");
|
||||||
|
int s_udp = socket(AF_INET,SOCK_DGRAM, 0);
|
||||||
|
|
||||||
|
if (s_udp == -1){
|
||||||
|
err_socks();
|
||||||
|
return 1;
|
||||||
|
} else printf("Ok!\n");
|
||||||
|
|
||||||
|
struct sockaddr_in servaddr, cliaddr;
|
||||||
|
memset(&servaddr, 0, sizeof(servaddr));
|
||||||
|
memset(&cliaddr, 0, sizeof(cliaddr));
|
||||||
|
|
||||||
|
servaddr.sin_family = AF_INET;
|
||||||
|
servaddr.sin_addr.s_addr = INADDR_ANY;
|
||||||
|
servaddr.sin_port = PORT;
|
||||||
|
|
||||||
|
if ( bind(s_udp, (const struct sockaddr *)&servaddr, sizeof(servaddr)) < 0 ) {
|
||||||
|
err_socks();
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
printf("[+] Bind ok !\n");
|
||||||
|
|
||||||
|
int n;
|
||||||
|
int len;
|
||||||
|
int msg_count = 0;
|
||||||
|
char buffer[l_msg];
|
||||||
|
|
||||||
|
printf("[+] Waiting for messages...\n");
|
||||||
|
|
||||||
|
while (msg_count >= nb_message || nb_message != -1){
|
||||||
|
n = recvfrom(s_udp, (char *)buffer, l_msg,
|
||||||
|
MSG_WAITALL, (struct sockaddr *) &cliaddr,
|
||||||
|
&len);
|
||||||
|
buffer[n] = '\0';
|
||||||
|
printf("[>] Reception du message n°%d : %s\n", msg_count, buffer);
|
||||||
|
|
||||||
|
n = 0;
|
||||||
|
msg_count++;
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
////////////////////////////TCP/////////////////////////////
|
||||||
|
|
||||||
|
int source_TCP(char *HOSTNAME, int PORT, int nb_message, int l_msg){
|
||||||
|
// Création du socket TCP
|
||||||
|
printf("[+] Création du socket local...");
|
||||||
|
int s_tcp = socket(AF_INET,SOCK_STREAM, 0);
|
||||||
|
|
||||||
|
if (s_tcp == -1){
|
||||||
|
err_socks();
|
||||||
|
return 1;
|
||||||
|
} else printf("Ok!\n");
|
||||||
|
|
||||||
|
// Création du socket distant
|
||||||
|
printf("[+] Création du socket distant...");
|
||||||
|
struct sockaddr_in servaddr;
|
||||||
|
memset(&servaddr, 0, sizeof(servaddr));
|
||||||
|
struct hostent *hp ;
|
||||||
|
|
||||||
|
// Résolution du nom
|
||||||
|
if ((hp = gethostbyname(HOSTNAME)) == NULL) {
|
||||||
|
err_conn();
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Creation de la struct du socket distant
|
||||||
|
servaddr.sin_family = AF_INET;
|
||||||
|
memcpy((char*)&(servaddr.sin_addr.s_addr), hp->h_addr, hp->h_length);
|
||||||
|
servaddr.sin_port = PORT;
|
||||||
|
printf("Ok!\n");
|
||||||
|
|
||||||
|
if (connect(s_tcp, (const struct sockaddr *)&servaddr, sizeof(servaddr)) < 0){
|
||||||
|
err_conn();
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
printf("[+] Envoi des messages...\n");
|
||||||
|
|
||||||
|
// TEST
|
||||||
|
char *pmesg = malloc(l_msg*sizeof(char));
|
||||||
|
char c = 'a';
|
||||||
|
|
||||||
|
// Création des messages à envoyer et envoi
|
||||||
|
for (int i = 1; i <= nb_message; i++){
|
||||||
|
construire_message(pmesg, c, l_msg);
|
||||||
|
afficher_message(pmesg, l_msg);
|
||||||
|
if (write(s_tcp, pmesg, l_msg) < 0){
|
||||||
|
err_conn();
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
c++;
|
||||||
|
if (c >= '{') c = 'a';
|
||||||
|
}
|
||||||
|
|
||||||
|
close(s_tcp);
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int puit_TCP(int PORT, int nb_message, int l_msg){
|
||||||
|
// Création du socket TCP
|
||||||
|
printf("[+] Création du socket local...");
|
||||||
|
int s_tcp = socket(AF_INET,SOCK_STREAM, 0);
|
||||||
|
|
||||||
|
if (s_tcp == -1){
|
||||||
|
err_socks();
|
||||||
|
return 1;
|
||||||
|
} else printf("Ok!\n");
|
||||||
|
|
||||||
|
struct sockaddr_in servaddr, cliaddr;
|
||||||
|
memset(&servaddr, 0, sizeof(servaddr));
|
||||||
|
memset(&cliaddr, 0, sizeof(cliaddr));
|
||||||
|
|
||||||
|
servaddr.sin_family = AF_INET;
|
||||||
|
servaddr.sin_addr.s_addr = INADDR_ANY;
|
||||||
|
servaddr.sin_port = PORT;
|
||||||
|
|
||||||
|
if (bind(s_tcp, (struct sockaddr *)&servaddr, sizeof(servaddr)) != 0) {
|
||||||
|
err_socks();
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
printf("[+] Bind ok !\n");
|
||||||
|
|
||||||
|
if (listen(s_tcp, 5) != 0){
|
||||||
|
err_conn();
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
printf("[+] Listening on port %d\n", PORT);
|
||||||
|
|
||||||
|
int s_conn, l_recpt, msg_count = 1;
|
||||||
|
int lg_cliaddr = sizeof(cliaddr);
|
||||||
|
|
||||||
|
if ((s_conn = accept(s_tcp, (struct sockaddr *)&cliaddr, &lg_cliaddr)) < 0){
|
||||||
|
err_conn();
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
printf("[+] Connection accepted\n");
|
||||||
|
|
||||||
|
|
||||||
|
int n;
|
||||||
|
char buffer[l_msg];
|
||||||
|
|
||||||
|
printf("[+] Waiting for messages...\n");
|
||||||
|
|
||||||
|
while (msg_count <= nb_message || nb_message == -1){
|
||||||
|
if ((l_recpt = read(s_conn, buffer, sizeof(buffer))) < 0){
|
||||||
|
err_conn();
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
if (l_recpt > 0){
|
||||||
|
printf("[>] Reception du message n°%d : %s\n", msg_count, buffer);
|
||||||
|
|
||||||
|
n = 0;
|
||||||
|
msg_count++;
|
||||||
|
bzero(buffer, l_msg);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
///////////////////////////////////////////////////////////
|
||||||
|
|
||||||
|
/// Main entry
|
||||||
|
int main (int argc, char **argv)
|
||||||
|
{
|
||||||
|
int c;
|
||||||
|
extern char *optarg;
|
||||||
|
extern int optind;
|
||||||
|
int nb_message = -1; /* Nb de messages à envoyer ou à recevoir, par défaut : 10 en émission, infini en réception */
|
||||||
|
int source = -1; /* 0 = puits, 1 = source */
|
||||||
|
int tcp = 1; /* 0 = UDP, 1 = TCP */
|
||||||
|
int l_msg = 30; /* longueur du message, en octet. Defaut : 30. */
|
||||||
|
|
||||||
|
while ((c = getopt(argc, argv, "pn:su")) != -1) {
|
||||||
|
switch (c) {
|
||||||
|
case 'p':
|
||||||
|
if (source == 1) {
|
||||||
|
err_args();
|
||||||
|
}
|
||||||
|
source = 0;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case 's':
|
||||||
|
if (source == 0) {
|
||||||
|
err_args();
|
||||||
|
}
|
||||||
|
source = 1;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case 'n':
|
||||||
|
nb_message = atoi(optarg);
|
||||||
|
break;
|
||||||
|
|
||||||
|
case 'u':
|
||||||
|
tcp = 0;
|
||||||
|
break;
|
||||||
|
|
||||||
|
default:
|
||||||
|
err_args();
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
int PORT = atoi(argv[argc-1]);
|
||||||
|
//PORT = htons(PORT);
|
||||||
|
|
||||||
|
if (source == -1) {
|
||||||
|
err_args();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (tcp)
|
||||||
|
printf("[+] On utilise le protocol TCP\n");
|
||||||
|
else
|
||||||
|
printf("[+] On utilise le protocol UDP\n");
|
||||||
|
|
||||||
|
|
||||||
|
char* HOSTNAME = "";
|
||||||
|
if (source == 1){
|
||||||
|
printf("[+] On est dans la source\n");
|
||||||
|
HOSTNAME = argv[argc-2];
|
||||||
|
}
|
||||||
|
else
|
||||||
|
printf("[+] On est dans le puit\n");
|
||||||
|
|
||||||
|
if (nb_message != -1) {
|
||||||
|
if (source == 1)
|
||||||
|
printf("[+] Nombre de tampons à envoyer : %d\n", nb_message);
|
||||||
|
else
|
||||||
|
printf("[+] Nombre de tampons à recevoir : %d\n", nb_message);
|
||||||
|
} else {
|
||||||
|
if (source == 1) {
|
||||||
|
nb_message = 10 ;
|
||||||
|
printf("[~] Nombre de tampons à envoyer : 10 par défaut\n");
|
||||||
|
}
|
||||||
|
else
|
||||||
|
printf("[+] Nombre de tampons à envoyer : infini\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (l_msg == 30)
|
||||||
|
printf("[~] Longueur des messages à envoyer/recevoir : 30 par défaut\n");
|
||||||
|
else
|
||||||
|
printf("[+] Nombre de tampons à envoyer/recevoir : %d \n", l_msg);
|
||||||
|
|
||||||
|
int s = 1;
|
||||||
|
|
||||||
|
// Choose the right function
|
||||||
|
if (tcp){
|
||||||
|
if (source) s = source_TCP(HOSTNAME, PORT, nb_message, l_msg);
|
||||||
|
else s = puit_TCP(PORT, nb_message, l_msg);
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
if (source) s = source_UDP(HOSTNAME, PORT, nb_message, l_msg);
|
||||||
|
else s = puit_UDP(PORT, nb_message, l_msg);
|
||||||
|
}
|
||||||
|
|
||||||
|
return s;
|
||||||
|
}
|
||||||
|
|
||||||
387
tsock_v3_gasc_m.c
Normal file
387
tsock_v3_gasc_m.c
Normal file
|
|
@ -0,0 +1,387 @@
|
||||||
|
/* librairie standard ... */
|
||||||
|
#include <stdlib.h>
|
||||||
|
/* pour getopt */
|
||||||
|
#include <unistd.h>
|
||||||
|
/* déclaration des types de base */
|
||||||
|
#include <sys/types.h>
|
||||||
|
/* constantes relatives aux domaines, types et protocoles */
|
||||||
|
#include <sys/socket.h>
|
||||||
|
/* constantes et structures propres au domaine UNIX */
|
||||||
|
#include <sys/un.h>
|
||||||
|
/* constantes et structures propres au domaine INTERNET */
|
||||||
|
#include <netinet/in.h>
|
||||||
|
/* structures retournées par les fonctions de gestion de la base de
|
||||||
|
données du réseau */
|
||||||
|
#include <netdb.h>
|
||||||
|
/* pour les entrées/sorties */
|
||||||
|
#include <stdio.h>
|
||||||
|
/* pour la gestion des erreurs */
|
||||||
|
#include <errno.h>
|
||||||
|
|
||||||
|
/*
|
||||||
|
TP AIPS
|
||||||
|
Version 3 alpha - NON COMPLETE
|
||||||
|
*/
|
||||||
|
|
||||||
|
///////////////////////////////////////////////////////////
|
||||||
|
|
||||||
|
// Fonctions d'erreurs
|
||||||
|
/// Afficher l'usage et quitte
|
||||||
|
void err_args() {
|
||||||
|
printf("[!] Usage: cmd [-u][-p|-s][-n ##][-l ##]\n");
|
||||||
|
exit(EXIT_FAILURE);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Affiche erreur socket
|
||||||
|
void err_socks(){
|
||||||
|
printf("\n[-] Erreur lors de la création/affectation du socket\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Affiche erreur connexion
|
||||||
|
void err_conn(){
|
||||||
|
printf("\n[-] Erreur lors de la connexion serveur/client\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
///////////////////////////////////////////////////////////
|
||||||
|
|
||||||
|
|
||||||
|
// Given functions
|
||||||
|
void construire_message(char *message, char motif, int lg, int numero) {
|
||||||
|
int i;
|
||||||
|
char str[5];
|
||||||
|
for (i=0;i<5;++i) str[i] = '-';
|
||||||
|
sprintf(str, "%5d", numero);
|
||||||
|
for (i=0;i<lg;i++) message[i] = motif;
|
||||||
|
for (i=0;i<5;++i) message[i] = str[i];
|
||||||
|
}
|
||||||
|
|
||||||
|
void afficher_message(char *message, int lg) {
|
||||||
|
int i;
|
||||||
|
printf("[+] Message construit : ");
|
||||||
|
for (i=0;i<lg;i++) printf("%c", message[i]);
|
||||||
|
printf("\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
///////////////////////////UDP/////////////////////////////
|
||||||
|
|
||||||
|
int source_UDP(char *HOSTNAME, int PORT, int nb_message, int l_msg) {
|
||||||
|
// Création du socket UDP
|
||||||
|
printf("[+] Création du socket local...");
|
||||||
|
int s_udp = socket(AF_INET,SOCK_DGRAM, 0);
|
||||||
|
|
||||||
|
if (s_udp == -1){
|
||||||
|
err_socks();
|
||||||
|
return 1;
|
||||||
|
} else printf("Ok!\n");
|
||||||
|
|
||||||
|
|
||||||
|
// Création du socket distant
|
||||||
|
printf("[+] Création du socket distant...");
|
||||||
|
struct sockaddr_in servaddr;
|
||||||
|
memset(&servaddr, 0, sizeof(servaddr));
|
||||||
|
struct hostent *hp ;
|
||||||
|
|
||||||
|
// Résolution du nom
|
||||||
|
if ((hp = gethostbyname(HOSTNAME)) == NULL) {
|
||||||
|
err_conn();
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Creation de la struct du socket distant
|
||||||
|
servaddr.sin_family = AF_INET;
|
||||||
|
memcpy((char*)&(servaddr.sin_addr.s_addr), hp->h_addr_list[0], hp->h_length);
|
||||||
|
servaddr.sin_port = PORT;
|
||||||
|
printf("Ok!\n");
|
||||||
|
|
||||||
|
printf("[+] Envoi des messages à %s:%d...\n", HOSTNAME, PORT);
|
||||||
|
|
||||||
|
// TEST
|
||||||
|
char *pmesg = malloc(l_msg*sizeof(char));
|
||||||
|
char c = 'a';
|
||||||
|
|
||||||
|
// Création des messages à envoyer et envoi
|
||||||
|
for (int i = 1; i <= nb_message; i++){
|
||||||
|
construire_message(pmesg, c, l_msg, i);
|
||||||
|
afficher_message(pmesg, l_msg);
|
||||||
|
sendto(s_udp,
|
||||||
|
pmesg,
|
||||||
|
strlen(pmesg),
|
||||||
|
0,
|
||||||
|
(const struct sockaddr *) &servaddr,
|
||||||
|
sizeof(servaddr)
|
||||||
|
);
|
||||||
|
c++;
|
||||||
|
if (c >= '{') c = 'a';
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int puit_UDP(int PORT, int nb_message, int l_msg) {
|
||||||
|
// Création du socket UDP
|
||||||
|
printf("[+] Création du socket local...");
|
||||||
|
int s_udp = socket(AF_INET,SOCK_DGRAM, 0);
|
||||||
|
|
||||||
|
if (s_udp == -1){
|
||||||
|
err_socks();
|
||||||
|
return 1;
|
||||||
|
} else printf("Ok!\n");
|
||||||
|
|
||||||
|
struct sockaddr_in servaddr, cliaddr;
|
||||||
|
memset(&servaddr, 0, sizeof(servaddr));
|
||||||
|
memset(&cliaddr, 0, sizeof(cliaddr));
|
||||||
|
|
||||||
|
servaddr.sin_family = AF_INET;
|
||||||
|
servaddr.sin_addr.s_addr = INADDR_ANY;
|
||||||
|
servaddr.sin_port = PORT;
|
||||||
|
|
||||||
|
if ( bind(s_udp, (const struct sockaddr *)&servaddr, sizeof(servaddr)) < 0 ) {
|
||||||
|
err_socks();
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
printf("[+] Bind ok !\n");
|
||||||
|
|
||||||
|
int n;
|
||||||
|
int len;
|
||||||
|
int msg_count = 0;
|
||||||
|
char buffer[l_msg];
|
||||||
|
|
||||||
|
printf("[+] Waiting for messages...\n");
|
||||||
|
|
||||||
|
while (msg_count >= nb_message || nb_message != -1){
|
||||||
|
n = recvfrom(s_udp, (char *)buffer, l_msg,
|
||||||
|
MSG_WAITALL, (struct sockaddr *) &cliaddr,
|
||||||
|
&len);
|
||||||
|
buffer[n] = '\0';
|
||||||
|
printf("[>] Reception du message n°%d : %s\n", msg_count, buffer);
|
||||||
|
|
||||||
|
n = 0;
|
||||||
|
msg_count++;
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
////////////////////////////TCP/////////////////////////////
|
||||||
|
|
||||||
|
int source_TCP(char *HOSTNAME, int PORT, int nb_message, int l_msg){
|
||||||
|
// Création du socket TCP
|
||||||
|
printf("[+] Création du socket local...");
|
||||||
|
int s_tcp = socket(AF_INET,SOCK_STREAM, 0);
|
||||||
|
|
||||||
|
if (s_tcp == -1){
|
||||||
|
err_socks();
|
||||||
|
return 1;
|
||||||
|
} else printf("Ok!\n");
|
||||||
|
|
||||||
|
// Création du socket distant
|
||||||
|
printf("[+] Création du socket distant...");
|
||||||
|
struct sockaddr_in servaddr;
|
||||||
|
memset(&servaddr, 0, sizeof(servaddr));
|
||||||
|
struct hostent *hp ;
|
||||||
|
|
||||||
|
// Résolution du nom
|
||||||
|
if ((hp = gethostbyname(HOSTNAME)) == NULL) {
|
||||||
|
err_conn();
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Creation de la struct du socket distant
|
||||||
|
servaddr.sin_family = AF_INET;
|
||||||
|
memcpy((char*)&(servaddr.sin_addr.s_addr), hp->h_addr_list[0], hp->h_length);
|
||||||
|
servaddr.sin_port = PORT;
|
||||||
|
printf("Ok!\n");
|
||||||
|
|
||||||
|
if (connect(s_tcp, (const struct sockaddr *)&servaddr, sizeof(servaddr)) < 0){
|
||||||
|
err_conn();
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
printf("[+] Envoi des messages...\n");
|
||||||
|
|
||||||
|
// TEST
|
||||||
|
char *pmesg = malloc(l_msg*sizeof(char));
|
||||||
|
char c = 'a';
|
||||||
|
|
||||||
|
// Création des messages à envoyer et envoi
|
||||||
|
for (int i = 1; i <= nb_message; i++){
|
||||||
|
construire_message(pmesg, c, l_msg, i);
|
||||||
|
afficher_message(pmesg, l_msg);
|
||||||
|
if (write(s_tcp, pmesg, l_msg) < 0){
|
||||||
|
err_conn();
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
c++;
|
||||||
|
if (c >= '{') c = 'a';
|
||||||
|
}
|
||||||
|
|
||||||
|
close(s_tcp);
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int puit_TCP(int PORT, int nb_message, int l_msg){
|
||||||
|
// Création du socket TCP
|
||||||
|
printf("[+] Création du socket local...");
|
||||||
|
int s_tcp = socket(AF_INET,SOCK_STREAM, 0);
|
||||||
|
|
||||||
|
if (s_tcp == -1){
|
||||||
|
err_socks();
|
||||||
|
return 1;
|
||||||
|
} else printf("Ok!\n");
|
||||||
|
|
||||||
|
struct sockaddr_in servaddr, cliaddr;
|
||||||
|
memset(&servaddr, 0, sizeof(servaddr));
|
||||||
|
memset(&cliaddr, 0, sizeof(cliaddr));
|
||||||
|
|
||||||
|
servaddr.sin_family = AF_INET;
|
||||||
|
servaddr.sin_addr.s_addr = INADDR_ANY;
|
||||||
|
servaddr.sin_port = PORT;
|
||||||
|
|
||||||
|
if (bind(s_tcp, (struct sockaddr *)&servaddr, sizeof(servaddr)) != 0) {
|
||||||
|
err_socks();
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
printf("[+] Bind ok !\n");
|
||||||
|
|
||||||
|
if (listen(s_tcp, 5) != 0){
|
||||||
|
err_conn();
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
printf("[+] Listening on port %d\n", PORT);
|
||||||
|
|
||||||
|
int s_conn, l_recpt, msg_count = 1;
|
||||||
|
int lg_cliaddr = sizeof(cliaddr);
|
||||||
|
|
||||||
|
if ((s_conn = accept(s_tcp, (struct sockaddr *)&cliaddr, &lg_cliaddr)) < 0){
|
||||||
|
err_conn();
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
printf("[+] Connection accepted\n");
|
||||||
|
|
||||||
|
|
||||||
|
int n;
|
||||||
|
char buffer[l_msg];
|
||||||
|
|
||||||
|
printf("[+] Waiting for messages...\n");
|
||||||
|
|
||||||
|
while (msg_count <= nb_message || nb_message == -1){
|
||||||
|
if ((l_recpt = read(s_conn, buffer, sizeof(buffer))) < 0){
|
||||||
|
err_conn();
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
if (l_recpt > 0){
|
||||||
|
printf("[>] Reception du message n°%d : %s\n", msg_count, buffer);
|
||||||
|
|
||||||
|
n = 0;
|
||||||
|
msg_count++;
|
||||||
|
bzero(buffer, l_msg);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
///////////////////////////////////////////////////////////
|
||||||
|
|
||||||
|
/// Main entry
|
||||||
|
int main (int argc, char **argv)
|
||||||
|
{
|
||||||
|
int c;
|
||||||
|
extern char *optarg;
|
||||||
|
extern int optind;
|
||||||
|
int nb_message = -1; /* Nb de messages à envoyer ou à recevoir, par défaut : 10 en émission, infini en réception */
|
||||||
|
int source = -1; /* 0 = puits, 1 = source */
|
||||||
|
int tcp = 1; /* 0 = UDP, 1 = TCP */
|
||||||
|
int l_msg = 30; /* longueur du message, en octet. Defaut : 30. */
|
||||||
|
|
||||||
|
while ((c = getopt(argc, argv, "pn:sul:")) != -1) {
|
||||||
|
switch (c) {
|
||||||
|
case 'p':
|
||||||
|
if (source == 1) {
|
||||||
|
err_args();
|
||||||
|
}
|
||||||
|
source = 0;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case 's':
|
||||||
|
if (source == 0) {
|
||||||
|
err_args();
|
||||||
|
}
|
||||||
|
source = 1;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case 'n':
|
||||||
|
nb_message = atoi(optarg);
|
||||||
|
break;
|
||||||
|
|
||||||
|
case 'u':
|
||||||
|
tcp = 0;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case 'l':
|
||||||
|
l_msg = atoi(optarg) + 5;
|
||||||
|
break;
|
||||||
|
|
||||||
|
default:
|
||||||
|
err_args();
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
int PORT = atoi(argv[argc-1]);
|
||||||
|
//PORT = htons(PORT);
|
||||||
|
|
||||||
|
if (source == -1) {
|
||||||
|
err_args();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (tcp)
|
||||||
|
printf("[+] On utilise le protocol TCP\n");
|
||||||
|
else
|
||||||
|
printf("[+] On utilise le protocol UDP\n");
|
||||||
|
|
||||||
|
|
||||||
|
char* HOSTNAME = "";
|
||||||
|
if (source == 1){
|
||||||
|
printf("[+] On est dans la source\n");
|
||||||
|
HOSTNAME = argv[argc-2];
|
||||||
|
}
|
||||||
|
else
|
||||||
|
printf("[+] On est dans le puit\n");
|
||||||
|
|
||||||
|
if (nb_message != -1) {
|
||||||
|
if (source == 1)
|
||||||
|
printf("[+] Nombre de tampons à envoyer : %d\n", nb_message);
|
||||||
|
else
|
||||||
|
printf("[+] Nombre de tampons à recevoir : %d\n", nb_message);
|
||||||
|
} else {
|
||||||
|
if (source == 1) {
|
||||||
|
nb_message = 10 ;
|
||||||
|
printf("[~] Nombre de tampons à envoyer : 10 par défaut\n");
|
||||||
|
}
|
||||||
|
else
|
||||||
|
printf("[+] Nombre de tampons à envoyer : infini\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (l_msg == 30)
|
||||||
|
printf("[~] Longueur des messages à envoyer/recevoir : 30 par défaut\n");
|
||||||
|
else
|
||||||
|
printf("[+] Nombre de tampons à envoyer/recevoir : %d \n", l_msg);
|
||||||
|
|
||||||
|
int s = 1;
|
||||||
|
|
||||||
|
// Choose the right function
|
||||||
|
if (tcp){
|
||||||
|
if (source) s = source_TCP(HOSTNAME, PORT, nb_message, l_msg);
|
||||||
|
else s = puit_TCP(PORT, nb_message, l_msg);
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
if (source) s = source_UDP(HOSTNAME, PORT, nb_message, l_msg);
|
||||||
|
else s = puit_UDP(PORT, nb_message, l_msg);
|
||||||
|
}
|
||||||
|
|
||||||
|
return s;
|
||||||
|
}
|
||||||
|
|
||||||
Loading…
Reference in a new issue