Separate modes of tcp handlers
This commit is contained in:
parent
43d4bfca2f
commit
02ae085aed
25
README.md
25
README.md
@ -1,9 +1,15 @@
|
|||||||
# KQUEUE Utilization
|
# Kernel Queue (kqueue) Utilization
|
||||||
|
|
||||||
This is your lame attempt to understand how those Network Virtual Functions sofware works as well to sharpen your understanding on C++ and C programming language.
|
This is your lame attempt to understand how those Network Virtual Functions sofware works as well to sharpen your understanding on C++ and C programming language.
|
||||||
|
|
||||||
|
## Why not [epoll]() or [select]()
|
||||||
|
|
||||||
|
Using epoll or select would make your system spend most of its time looking at and fiddling sith the data structures and has very little time for anything else.
|
||||||
|
|
||||||
|
|
||||||
|
## Notes
|
||||||
|
|
||||||
|
|
||||||
## Hints
|
|
||||||
|
|
||||||
- Compile using `g++` for the file watcher
|
- Compile using `g++` for the file watcher
|
||||||
|
|
||||||
@ -13,8 +19,8 @@ The `kqueue_socket.c` is my attempt on create a socket watcher timer program usi
|
|||||||
|
|
||||||
- [KQUEUE NETBSD GUIDE](https://wiki.netbsd.org/tutorials/kqueue_tutorial/)
|
- [KQUEUE NETBSD GUIDE](https://wiki.netbsd.org/tutorials/kqueue_tutorial/)
|
||||||
|
|
||||||
```
|
```C
|
||||||
|
// --------------------------------------------
|
||||||
struct kevent my_event;
|
struct kevent my_event;
|
||||||
struct kevent my_change;
|
struct kevent my_change;
|
||||||
|
|
||||||
@ -23,13 +29,20 @@ The `kqueue_socket.c` is my attempt on create a socket watcher timer program usi
|
|||||||
for(;;){
|
for(;;){
|
||||||
|
|
||||||
...;
|
...;
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
```
|
```
|
||||||
|
|
||||||
|
### TCP 01
|
||||||
|
|
||||||
|
- Pointer of kevent
|
||||||
|
|
||||||
|
### TCP 02
|
||||||
|
|
||||||
|
- Array of kevent
|
||||||
|
|
||||||
## Goal
|
## Goal
|
||||||
|
|
||||||
- Understand the difference between c++ and C
|
- Understand the difference between c++ and C
|
||||||
- Understand how strcuct works and the difference between it on c++ and C
|
- Understand how strcuct works and the difference between it on c++ and C
|
||||||
- Understand
|
- Understand
|
||||||
|
BIN
bin/tcp_sock
Executable file
BIN
bin/tcp_sock
Executable file
Binary file not shown.
106
src/ktcp_socket_01.c
Normal file
106
src/ktcp_socket_01.c
Normal file
@ -0,0 +1,106 @@
|
|||||||
|
#include <stdio.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <netinet/in.h>
|
||||||
|
#include <sys/socket.h>
|
||||||
|
#include <sys/event.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
|
||||||
|
int main() {
|
||||||
|
int socket_listen_fd,
|
||||||
|
port_num = 1339,
|
||||||
|
client_len,
|
||||||
|
socket_connection_fd,
|
||||||
|
kq,
|
||||||
|
new_events;
|
||||||
|
|
||||||
|
struct kevent change_event[4], event[4];
|
||||||
|
struct kevent evList[32];
|
||||||
|
|
||||||
|
struct sockaddr_in serv_addr, client_addr;
|
||||||
|
|
||||||
|
if (((socket_listen_fd = socket(AF_INET, SOCK_STREAM, 0)) < 0))
|
||||||
|
{
|
||||||
|
perror("Failed creating socket");
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
bzero((char *)&serv_addr, sizeof(serv_addr));
|
||||||
|
serv_addr.sin_family = AF_INET;
|
||||||
|
serv_addr.sin_addr.s_addr = INADDR_ANY;
|
||||||
|
serv_addr.sin_port = htons(port_num);
|
||||||
|
|
||||||
|
if (bind(socket_listen_fd, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) < 0)
|
||||||
|
{
|
||||||
|
perror("Failed binding socket");
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
listen(socket_listen_fd, 3);
|
||||||
|
client_len = sizeof(client_addr);
|
||||||
|
|
||||||
|
// kqueue SETUP
|
||||||
|
|
||||||
|
kq = kqueue();
|
||||||
|
|
||||||
|
EV_SET(change_event, socket_listen_fd, EVFILT_READ, EV_ADD | EV_ENABLE, 0, 0, 0);
|
||||||
|
if (kevent(kq, change_event, 1, NULL, 0, NULL) == -1)
|
||||||
|
{
|
||||||
|
perror("Failed create kevent");
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
for (;;)
|
||||||
|
{
|
||||||
|
|
||||||
|
new_events = kevent(kq, NULL, 0, event, 1, NULL);
|
||||||
|
|
||||||
|
// printf("New events -> %d\n", new_events);
|
||||||
|
|
||||||
|
if (new_events == -1)
|
||||||
|
{
|
||||||
|
perror("Failed create new kevent");
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
for (int i = 0; new_events > i; i++)
|
||||||
|
{
|
||||||
|
printf("Amount of new events : %d\n", new_events);
|
||||||
|
int event_fd = event[i].ident;
|
||||||
|
|
||||||
|
if (event[i].flags & EV_EOF)
|
||||||
|
{
|
||||||
|
printf("Client disconnected...\n");
|
||||||
|
close(event_fd);
|
||||||
|
}
|
||||||
|
|
||||||
|
else if (event_fd == socket_listen_fd)
|
||||||
|
{
|
||||||
|
printf("New connection coming in...\n");
|
||||||
|
|
||||||
|
socket_connection_fd = accept(event_fd, (struct sockaddr *)&client_addr, (socklen_t *)&client_len);
|
||||||
|
if (socket_connection_fd == -1)
|
||||||
|
{
|
||||||
|
perror("Failed to accept connection...");
|
||||||
|
}
|
||||||
|
|
||||||
|
EV_SET(change_event, socket_connection_fd, EVFILT_READ, EV_ADD, 0, 0, NULL);
|
||||||
|
if (kevent(kq, change_event, 1, NULL, 0, NULL) < 0)
|
||||||
|
{
|
||||||
|
perror("Kevent failed to maintain connection");
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
else if (event[i].filter == EVFILT_READ)
|
||||||
|
{
|
||||||
|
char buf[1024];
|
||||||
|
size_t bytes_read = recv(event_fd, buf, sizeof(buf), 0);
|
||||||
|
printf("Reading %zu bytes\n", bytes_read);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
}
|
116
src/ktcp_socket_02.c
Normal file
116
src/ktcp_socket_02.c
Normal file
@ -0,0 +1,116 @@
|
|||||||
|
#include <stdio.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <netinet/in.h>
|
||||||
|
#include <sys/socket.h>
|
||||||
|
#include <sys/event.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
|
||||||
|
int main() {
|
||||||
|
int socket_listen_fd,
|
||||||
|
port_num = 2339,
|
||||||
|
client_len,
|
||||||
|
socket_connection_fd,
|
||||||
|
kq,
|
||||||
|
new_events;
|
||||||
|
|
||||||
|
struct kevent change_event, event;
|
||||||
|
struct kevent evList[32];
|
||||||
|
|
||||||
|
struct sockaddr_in serv_addr, client_addr;
|
||||||
|
|
||||||
|
if (((socket_listen_fd = socket(AF_INET, SOCK_STREAM, 0)) < 0))
|
||||||
|
{
|
||||||
|
perror("Failed creating socket");
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
bzero((char *)&serv_addr, sizeof(serv_addr));
|
||||||
|
serv_addr.sin_family = AF_INET;
|
||||||
|
serv_addr.sin_addr.s_addr = INADDR_ANY;
|
||||||
|
serv_addr.sin_port = htons(port_num);
|
||||||
|
|
||||||
|
if (bind(socket_listen_fd, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) < 0)
|
||||||
|
{ perror("Failed binding socket"); exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
listen(socket_listen_fd, 3);
|
||||||
|
client_len = sizeof(client_addr);
|
||||||
|
|
||||||
|
// kqueue SETUP
|
||||||
|
|
||||||
|
kq = kqueue();
|
||||||
|
|
||||||
|
EV_SET(&change_event, socket_listen_fd, EVFILT_READ, EV_ADD | EV_ENABLE, 0, 0, 0);
|
||||||
|
if (kevent(kq, &change_event, 1, NULL, 0, NULL) == -1)
|
||||||
|
{
|
||||||
|
perror("Failed create kevent");
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
for (;;)
|
||||||
|
{
|
||||||
|
|
||||||
|
new_events = kevent(kq, NULL, 0, evList, 1, NULL);
|
||||||
|
|
||||||
|
// printf("New events -> %d\n", new_events);
|
||||||
|
|
||||||
|
if (new_events == -1)
|
||||||
|
{
|
||||||
|
perror("Failed create new kevent");
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
for (int i = 0; new_events > i; i++)
|
||||||
|
{
|
||||||
|
printf("Amount of new events : %d\n", new_events);
|
||||||
|
int event_fd = evList[i].ident;
|
||||||
|
|
||||||
|
if (evList[i].flags & EV_EOF)
|
||||||
|
{
|
||||||
|
printf("Client disconnected...\n");
|
||||||
|
close(event_fd);
|
||||||
|
}
|
||||||
|
|
||||||
|
else if (event_fd == socket_listen_fd)
|
||||||
|
{
|
||||||
|
printf("New connection coming in...\n");
|
||||||
|
|
||||||
|
socket_connection_fd = accept(event_fd, (struct sockaddr *)&client_addr, (socklen_t *)&client_len);
|
||||||
|
if (socket_connection_fd == -1)
|
||||||
|
{
|
||||||
|
perror("Failed to accept connection...");
|
||||||
|
}
|
||||||
|
|
||||||
|
EV_SET(&change_event, socket_connection_fd, EVFILT_READ, EV_ADD, 0, 0, NULL);
|
||||||
|
if (kevent(kq, &change_event, 1, NULL, 0, NULL) < 0)
|
||||||
|
{
|
||||||
|
perror("Kevent failed to maintain connection");
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
else if (evList[i].filter == EVFILT_READ)
|
||||||
|
{
|
||||||
|
char buf[1024];
|
||||||
|
size_t bytes_read = recv(event_fd, buf, sizeof(buf), 0);
|
||||||
|
printf("--------------------------------------------------\n");
|
||||||
|
printf("Reading %zu bytes\n", bytes_read);
|
||||||
|
printf("Incoming msg : %s", buf);
|
||||||
|
printf("--------------------------------------------------\n");
|
||||||
|
|
||||||
|
char s_buf[1024];
|
||||||
|
int len;
|
||||||
|
char *message = "[+] Hello from kTCP\n";
|
||||||
|
|
||||||
|
strcpy(s_buf, message);
|
||||||
|
|
||||||
|
send(event_fd, s_buf, strlen(s_buf), 0);
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
}
|
125
src/wtf.c
Normal file
125
src/wtf.c
Normal file
@ -0,0 +1,125 @@
|
|||||||
|
#include <stdio.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <netinet/in.h>
|
||||||
|
#include <sys/socket.h>
|
||||||
|
#include <sys/event.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
|
||||||
|
int main()
|
||||||
|
{
|
||||||
|
// All needed variables.
|
||||||
|
int socket_listen_fd,
|
||||||
|
portno = 1815,
|
||||||
|
client_len,
|
||||||
|
socket_connection_fd,
|
||||||
|
kq,
|
||||||
|
new_events;
|
||||||
|
struct kevent change_event[4],
|
||||||
|
event[4];
|
||||||
|
struct sockaddr_in serv_addr,
|
||||||
|
client_addr;
|
||||||
|
|
||||||
|
// Create socket.
|
||||||
|
if (((socket_listen_fd = socket(AF_INET, SOCK_STREAM, 0)) < 0))
|
||||||
|
{
|
||||||
|
perror("ERROR opening socket");
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create socket structure and bind to ip address.
|
||||||
|
bzero((char *)&serv_addr, sizeof(serv_addr));
|
||||||
|
serv_addr.sin_family = AF_INET;
|
||||||
|
serv_addr.sin_addr.s_addr = INADDR_ANY;
|
||||||
|
serv_addr.sin_port = htons(portno);
|
||||||
|
|
||||||
|
if (bind(socket_listen_fd, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) < 0)
|
||||||
|
{
|
||||||
|
perror("Error binding socket");
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Start listening.
|
||||||
|
listen(socket_listen_fd, 3);
|
||||||
|
client_len = sizeof(client_addr);
|
||||||
|
|
||||||
|
// Prepare the kqueue.
|
||||||
|
kq = kqueue();
|
||||||
|
|
||||||
|
// Create event 'filter', these are the events we want to monitor.
|
||||||
|
// Here we want to monitor: socket_listen_fd, for the events: EVFILT_READ
|
||||||
|
// (when there is data to be read on the socket), and perform the following
|
||||||
|
// actions on this kevent: EV_ADD and EV_ENABLE (add the event to the kqueue
|
||||||
|
// and enable it).
|
||||||
|
EV_SET(change_event, socket_listen_fd, EVFILT_READ, EV_ADD | EV_ENABLE, 0, 0, 0);
|
||||||
|
|
||||||
|
// Register kevent with the kqueue.
|
||||||
|
if (kevent(kq, change_event, 1, NULL, 0, NULL) == -1)
|
||||||
|
{
|
||||||
|
perror("kevent");
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Actual event loop.
|
||||||
|
for (;;)
|
||||||
|
{
|
||||||
|
// Check for new events, but do not register new events with
|
||||||
|
// the kqueue. Hence the 2nd and 3rd arguments are NULL, 0.
|
||||||
|
// Only handle 1 new event per iteration in the loop; 5th
|
||||||
|
// argument is 1.
|
||||||
|
new_events = kevent(kq, NULL, 0, event, 1, NULL);
|
||||||
|
if (new_events == -1)
|
||||||
|
{
|
||||||
|
perror("kevent");
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
for (int i = 0; new_events > i; i++)
|
||||||
|
{
|
||||||
|
printf("amount of new events: %d\n", new_events);
|
||||||
|
int event_fd = event[i].ident;
|
||||||
|
|
||||||
|
// When the client disconnects an EOF is sent. By closing the file
|
||||||
|
// descriptor the event is automatically removed from the kqueue.
|
||||||
|
if (event[i].flags & EV_EOF)
|
||||||
|
{
|
||||||
|
printf("Client has disconnected\n");
|
||||||
|
close(event_fd);
|
||||||
|
}
|
||||||
|
// If the new event's file descriptor is the same as the listening
|
||||||
|
// socket's file descriptor, we are sure that a new client wants
|
||||||
|
// to connect to our socket.
|
||||||
|
else if (event_fd == socket_listen_fd)
|
||||||
|
{
|
||||||
|
printf("New connection coming in...\n");
|
||||||
|
|
||||||
|
// Incoming socket connection on the listening socket.
|
||||||
|
// Create a new socket for the actual connection to client.
|
||||||
|
socket_connection_fd = accept(event_fd, (struct sockaddr *)&client_addr, (socklen_t *)&client_len);
|
||||||
|
if (socket_connection_fd == -1)
|
||||||
|
{
|
||||||
|
perror("Accept socket error");
|
||||||
|
}
|
||||||
|
|
||||||
|
// Put this new socket connection also as a 'filter' event
|
||||||
|
// to watch in kqueue, so we can now watch for events on this
|
||||||
|
// new socket.
|
||||||
|
EV_SET(change_event, socket_connection_fd, EVFILT_READ, EV_ADD, 0, 0, NULL);
|
||||||
|
if (kevent(kq, change_event, 1, NULL, 0, NULL) < 0)
|
||||||
|
{
|
||||||
|
perror("kevent error");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
else if (event[i].filter & EVFILT_READ)
|
||||||
|
{
|
||||||
|
// Read bytes from socket
|
||||||
|
char buf[1024];
|
||||||
|
size_t bytes_read = recv(event_fd, buf, sizeof(buf), 0);
|
||||||
|
printf("read %zu bytes\n", bytes_read);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user