161 lines
2.9 KiB
C
161 lines
2.9 KiB
C
#include <sys/socket.h>
|
|
#include <sys/types.h>
|
|
#include <netinet/in.h>
|
|
#include <sys/stat.h>
|
|
#include <fcntl.h>
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
#include <arpa/inet.h>
|
|
|
|
|
|
#include "../src/ipc.h"
|
|
#include "../src/utils.h"
|
|
|
|
/**
|
|
* tcp connection to localhost port argv[1]
|
|
* message 1: pong
|
|
* message 2: echo with libipc format
|
|
*
|
|
* [messagetype | len | usertype | payload ]
|
|
* 1 B | 4 B | 1 B | ]
|
|
*/
|
|
|
|
|
|
int connection(int port)
|
|
{
|
|
int sockfd;
|
|
struct sockaddr_in server;
|
|
socklen_t addrlen;
|
|
|
|
// socket factory
|
|
if((sockfd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) == -1)
|
|
{
|
|
perror("socket");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
// init remote addr structure and other params
|
|
server.sin_family = AF_INET;
|
|
server.sin_port = htons(port);
|
|
addrlen = sizeof(struct sockaddr_in);
|
|
|
|
// get addr from command line and convert it
|
|
if(inet_pton(AF_INET, "127.0.0.1", &server.sin_addr) <= 0)
|
|
{
|
|
perror("inet_pton");
|
|
close(sockfd);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
printf("Trying to connect to the remote host\n");
|
|
if(connect(sockfd, (struct sockaddr *) &server, addrlen) == -1)
|
|
{
|
|
perror("connect");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
printf("Connection OK\n");
|
|
|
|
return sockfd;
|
|
}
|
|
|
|
|
|
void send_receive (int sockfd) {
|
|
unsigned char buf[BUFSIZ];
|
|
memset (buf, 0, BUFSIZ);
|
|
|
|
// first, send service name "pong"
|
|
// send string
|
|
if(send(sockfd, "pong", 4, 0) == -1) {
|
|
perror("send pong");
|
|
close(sockfd);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
printf ("message 'pong' sent\n");
|
|
|
|
memset (buf, 0, BUFSIZ);
|
|
|
|
int paylen;
|
|
paylen = recv(sockfd, buf, BUFSIZ, 0);
|
|
if (paylen <= 0) {
|
|
printf ("cannot connect to networkd\n");
|
|
exit (EXIT_FAILURE);
|
|
}
|
|
printf ("should receive 'OK': %*.s\n", paylen, buf);
|
|
|
|
memset (buf, 0, BUFSIZ);
|
|
|
|
buf[0] = MSG_TYPE_DATA;
|
|
|
|
// uint32_t v = htonl(6);
|
|
// memcpy (buf+1, &v, sizeof (uint32_t));
|
|
|
|
uint32_t v = 6;
|
|
uint32_t net_paylen = htonl (v);
|
|
memcpy (buf+1, &net_paylen, sizeof (uint32_t));
|
|
|
|
buf[5] = 0;
|
|
memcpy (buf+6, "coucou", 6);
|
|
|
|
print_hexa ("SENT MESSAGE", buf, 12);
|
|
|
|
// 2 | 6 | 0 | "coucou"
|
|
// 1 B | 4 B | 1 | 6 B
|
|
|
|
if(send(sockfd, buf, 12, 0) == -1) {
|
|
perror("send coucou");
|
|
close(sockfd);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
printf ("message 'coucou' sent\n");
|
|
|
|
memset (buf, 0, BUFSIZ);
|
|
|
|
paylen = recv (sockfd, buf, BUFSIZ, 0);
|
|
if(paylen < 0) {
|
|
perror("recv a message");
|
|
close(sockfd);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
if (paylen > 0) {
|
|
print_hexa ("RECEIVED MESSAGE", buf, paylen);
|
|
}
|
|
else {
|
|
fprintf (stderr, "error: disconnection from the server\n");
|
|
exit (EXIT_FAILURE);
|
|
}
|
|
|
|
#if 0
|
|
// send string
|
|
if(sendto(sockfd, , ), 0) == -1)
|
|
{
|
|
perror("sendto");
|
|
close(sockfd);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
#endif
|
|
}
|
|
|
|
|
|
int main(int argc, char * argv[])
|
|
{
|
|
int port = 9000;
|
|
if (argc > 1) {
|
|
port = atoi (argv[1]);
|
|
}
|
|
|
|
int sockfd = connection (port);
|
|
|
|
send_receive (sockfd);
|
|
|
|
printf("Disconnection\n");
|
|
|
|
// close the socket
|
|
close(sockfd);
|
|
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|