59 lines
1.4 KiB
C
59 lines
1.4 KiB
C
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
#include <arpa/inet.h>
|
|
|
|
#include "rep.h"
|
|
|
|
void
|
|
handle_client(int client_socket)
|
|
{
|
|
char buffer[BUFFER_SIZE] = {0};
|
|
read(client_socket, buffer, BUFFER_SIZE);
|
|
|
|
printf("Domain query: %s\n", buffer);
|
|
|
|
for (int i = 0; records[i].domain[0] != '\0'; i++) {
|
|
if (strcmp(records[i].domain, buffer) == 0) {
|
|
write(client_socket, records[i].ip, strlen(records[i].ip));
|
|
write(client_socket, "\n", 1);
|
|
close(client_socket);
|
|
return;
|
|
}
|
|
}
|
|
|
|
// If domain not found
|
|
char *not_found = "Domain not found";
|
|
write(client_socket, not_found, strlen(not_found));
|
|
write(client_socket, "\n", 1);
|
|
close(client_socket);
|
|
}
|
|
|
|
int
|
|
main()
|
|
{
|
|
int server_fd, client_socket;
|
|
struct sockaddr_in address;
|
|
int opt = 1;
|
|
int addrlen = sizeof(address);
|
|
|
|
server_fd = socket(AF_INET, SOCK_STREAM, 0);
|
|
setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR | SO_REUSEPORT, &opt, sizeof(opt));
|
|
|
|
address.sin_family = AF_INET;
|
|
address.sin_addr.s_addr = INADDR_ANY;
|
|
address.sin_port = htons(RNS_PORT);
|
|
|
|
bind(server_fd, (struct sockaddr *)&address, sizeof(address));
|
|
listen(server_fd, 3);
|
|
|
|
while (1) {
|
|
client_socket = accept(server_fd, (struct sockaddr *)&address, (socklen_t *)&addrlen);
|
|
if (client_socket >= 0) {
|
|
handle_client(client_socket);
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|