init
This commit is contained in:
commit
72f97117e7
318
aiken.c
Normal file
318
aiken.c
Normal file
@ -0,0 +1,318 @@
|
|||||||
|
#include <stdlib.h>
|
||||||
|
#include <limits.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <getopt.h>
|
||||||
|
#include <errno.h>
|
||||||
|
#include <time.h>
|
||||||
|
#include <sys/stat.h>
|
||||||
|
#include <sys/socket.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
#include <netinet/in.h>
|
||||||
|
#include <arpa/inet.h>
|
||||||
|
|
||||||
|
#define PORT_UDP 46226
|
||||||
|
#define PORT_TCP 46227
|
||||||
|
|
||||||
|
#define BUF_SIZE 1024
|
||||||
|
|
||||||
|
char *basename(const char *path) {
|
||||||
|
char *res = strrchr(path, '/');
|
||||||
|
return (res == NULL) ? path : (res + 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
void token_generate(char *dst) {
|
||||||
|
sprintf(dst, "%04x", (uint16_t) time(NULL));
|
||||||
|
}
|
||||||
|
|
||||||
|
void share(const char *path) {
|
||||||
|
FILE *file = fopen(path, "rb");
|
||||||
|
if (file == NULL) {
|
||||||
|
perror("Failed to open file");
|
||||||
|
exit(errno);
|
||||||
|
}
|
||||||
|
char token[5];
|
||||||
|
int udp_sock = socket(AF_INET, SOCK_DGRAM, 0), sock_option = 1;
|
||||||
|
if (udp_sock < 0) {
|
||||||
|
perror("Failed to initialize UDP socket");
|
||||||
|
fclose(file);
|
||||||
|
exit(errno);
|
||||||
|
}
|
||||||
|
if (setsockopt(udp_sock, SOL_SOCKET, SO_REUSEADDR, &sock_option, sizeof(sock_option)) < 0) {
|
||||||
|
perror("Failed to set UDP socket options");
|
||||||
|
exit(errno);
|
||||||
|
}
|
||||||
|
struct sockaddr_in listen_addr, receiver_addr;
|
||||||
|
socklen_t addr_len = sizeof(receiver_addr);
|
||||||
|
bzero(&listen_addr, sizeof(listen_addr));
|
||||||
|
listen_addr.sin_family = AF_INET;
|
||||||
|
listen_addr.sin_addr.s_addr = INADDR_ANY;
|
||||||
|
listen_addr.sin_port = htons(PORT_UDP);
|
||||||
|
if (bind(udp_sock, (struct sockaddr *) &listen_addr, sizeof(listen_addr)) < 0) {
|
||||||
|
perror("Failed to bind UDP socket");
|
||||||
|
fclose(file);
|
||||||
|
close(udp_sock);
|
||||||
|
exit(errno);
|
||||||
|
}
|
||||||
|
token_generate(token);
|
||||||
|
printf("Successfully shared!\nToken: %s\n\n", token);
|
||||||
|
char buffer[BUF_SIZE];
|
||||||
|
char *filename = basename(path);
|
||||||
|
int tcp_server_sock, tcp_client_sock;
|
||||||
|
printf("Waiting for request...\n");
|
||||||
|
// recieve request from receiver
|
||||||
|
if (recvfrom(udp_sock, buffer, 4, 0, (struct sockaddr *) &receiver_addr, &addr_len) < 4) {
|
||||||
|
printf("Got invalid request from %s\n", inet_ntoa(receiver_addr.sin_addr));
|
||||||
|
fclose(file);
|
||||||
|
close(udp_sock);
|
||||||
|
exit(errno);
|
||||||
|
}
|
||||||
|
// check request token
|
||||||
|
if (memcmp(buffer, token, 4) != 0) {
|
||||||
|
printf("Got invalid token from %s\n", inet_ntoa(receiver_addr.sin_addr));
|
||||||
|
// send NO packet which tells that request is rejected
|
||||||
|
if (sendto(udp_sock, "NO", 2, 0, (struct sockaddr *) &receiver_addr, addr_len) < 2) {
|
||||||
|
perror("Failed to send NO packet");
|
||||||
|
}
|
||||||
|
fclose(file);
|
||||||
|
close(udp_sock);
|
||||||
|
exit(errno);
|
||||||
|
}
|
||||||
|
printf("Got request from %s\n", inet_ntoa(receiver_addr.sin_addr));
|
||||||
|
// create a TCP socket to send file
|
||||||
|
tcp_server_sock = socket(AF_INET, SOCK_STREAM, 0);
|
||||||
|
if (tcp_server_sock < 0) {
|
||||||
|
perror("Failed to initialize TCP socket");
|
||||||
|
fclose(file);
|
||||||
|
close(udp_sock);
|
||||||
|
exit(errno);
|
||||||
|
}
|
||||||
|
if (setsockopt(tcp_server_sock, SOL_SOCKET, SO_REUSEADDR, &sock_option, sizeof(sock_option)) < 0) {
|
||||||
|
perror("Failed to set TCP socket options");
|
||||||
|
exit(errno);
|
||||||
|
}
|
||||||
|
listen_addr.sin_port = htons(PORT_TCP);
|
||||||
|
if (bind(tcp_server_sock, (struct sockaddr *) &listen_addr, addr_len) < 0) {
|
||||||
|
perror("Failed to bind TCP socket");
|
||||||
|
fclose(file);
|
||||||
|
close(udp_sock);
|
||||||
|
close(tcp_server_sock);
|
||||||
|
exit(errno);
|
||||||
|
}
|
||||||
|
if (listen(tcp_server_sock, 1) < 0) {
|
||||||
|
perror("Failed to listen to TCP socket");
|
||||||
|
fclose(file);
|
||||||
|
close(udp_sock);
|
||||||
|
close(tcp_server_sock);
|
||||||
|
exit(errno);
|
||||||
|
}
|
||||||
|
// send OK packet which tells that we are ready to send file
|
||||||
|
if (sendto(udp_sock, "OK", 2, 0, (struct sockaddr *) &receiver_addr, addr_len) < 2) {
|
||||||
|
perror("Failed to send OK packet");
|
||||||
|
fclose(file);
|
||||||
|
close(udp_sock);
|
||||||
|
close(tcp_server_sock);
|
||||||
|
exit(errno);
|
||||||
|
}
|
||||||
|
close(udp_sock);
|
||||||
|
printf("Waiting for reciever connection...\n");
|
||||||
|
// wait for receiver connection
|
||||||
|
if ((tcp_client_sock = accept(tcp_server_sock, (struct sockaddr *) &receiver_addr, &addr_len)) < 0) {
|
||||||
|
perror("Failed to accept receiver connection");
|
||||||
|
fclose(file);
|
||||||
|
close(tcp_server_sock);
|
||||||
|
exit(errno);
|
||||||
|
}
|
||||||
|
printf("Receiver connected, sending...\n");
|
||||||
|
// send file name
|
||||||
|
if (send(tcp_client_sock, filename, strlen(filename) + 1, 0) < strlen(filename) + 1) {
|
||||||
|
perror("Failed to send file name");
|
||||||
|
fclose(file);
|
||||||
|
close(tcp_client_sock);
|
||||||
|
close(tcp_server_sock);
|
||||||
|
exit(errno);
|
||||||
|
}
|
||||||
|
// send file data
|
||||||
|
ssize_t rcount;
|
||||||
|
while ((rcount = fread(buffer, 1, BUF_SIZE, file)) > 0) {
|
||||||
|
if (send(tcp_client_sock, buffer, rcount, 0) != rcount) {
|
||||||
|
perror("Failed to send file data");
|
||||||
|
fclose(file);
|
||||||
|
close(tcp_client_sock);
|
||||||
|
close(tcp_server_sock);
|
||||||
|
exit(errno);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
close(tcp_client_sock);
|
||||||
|
close(tcp_server_sock);
|
||||||
|
printf("Sending complete!\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
void get(const char *token, const char *path) {
|
||||||
|
size_t abspath_len = strlen(path) + 1;
|
||||||
|
if (abspath_len > PATH_MAX) {
|
||||||
|
printf("Invalid path: too long\n");
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
char abspath[PATH_MAX];
|
||||||
|
strcpy(abspath, path);
|
||||||
|
char fname[FILENAME_MAX];
|
||||||
|
size_t fname_len;
|
||||||
|
struct stat stat_buf;
|
||||||
|
int udp_sock = socket(AF_INET, SOCK_DGRAM, 0);
|
||||||
|
if (udp_sock < 0) {
|
||||||
|
perror("Failed to initialize UDP socket");
|
||||||
|
exit(errno);
|
||||||
|
}
|
||||||
|
int sock_option = 1;
|
||||||
|
if (setsockopt(udp_sock, SOL_SOCKET, SO_BROADCAST, &sock_option, sizeof(sock_option)) < 0) {
|
||||||
|
perror("Failed to set UDP socket options");
|
||||||
|
exit(errno);
|
||||||
|
}
|
||||||
|
struct sockaddr_in broadcast_addr, server_addr;
|
||||||
|
socklen_t addr_len = sizeof(server_addr);
|
||||||
|
bzero(&broadcast_addr, sizeof(broadcast_addr));
|
||||||
|
broadcast_addr.sin_family = AF_INET;
|
||||||
|
broadcast_addr.sin_addr.s_addr = INADDR_BROADCAST;
|
||||||
|
broadcast_addr.sin_port = htons(PORT_UDP);
|
||||||
|
server_addr.sin_family = AF_INET;
|
||||||
|
server_addr.sin_addr.s_addr = INADDR_ANY;
|
||||||
|
server_addr.sin_port = htons(PORT_TCP);
|
||||||
|
int tcp_sock = socket(AF_INET, SOCK_STREAM, 0);
|
||||||
|
if (tcp_sock < 0) {
|
||||||
|
perror("Failed to initialize TCP socket");
|
||||||
|
exit(errno);
|
||||||
|
}
|
||||||
|
if (setsockopt(tcp_sock, SOL_SOCKET, SO_REUSEADDR, &sock_option, sizeof(sock_option)) < 0) {
|
||||||
|
perror("Failed to set TCP socket options");
|
||||||
|
exit(errno);
|
||||||
|
}
|
||||||
|
if (bind(tcp_sock, (struct sockaddr *) &server_addr, addr_len) < 0) {
|
||||||
|
perror("Failed to bind TCP socket");
|
||||||
|
exit(errno);
|
||||||
|
}
|
||||||
|
char buffer[BUF_SIZE];
|
||||||
|
printf("Searching server... ");
|
||||||
|
// send broadcast with the token
|
||||||
|
if (sendto(udp_sock, token, strlen(token), 0, (struct sockaddr *) &broadcast_addr, sizeof(broadcast_addr)) < strlen(token)) {
|
||||||
|
perror("Failed to broadcast request");
|
||||||
|
exit(errno);
|
||||||
|
}
|
||||||
|
// receive response from server
|
||||||
|
if (recvfrom(udp_sock, (void *) buffer, 2, 0, (struct sockaddr *) &server_addr, &addr_len) < 2) {
|
||||||
|
perror("Failed to receive response");
|
||||||
|
exit(errno);
|
||||||
|
}
|
||||||
|
// check server response
|
||||||
|
printf("Server found, response: %.2s\n", buffer);
|
||||||
|
if (memcmp(buffer, "OK", 2) != 0) {
|
||||||
|
if (memcmp(buffer, "NO", 2) != 0) {
|
||||||
|
printf("Error: Got unexpected response from server\n");
|
||||||
|
} else {
|
||||||
|
printf("Request was rejected by server\n");
|
||||||
|
}
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
// connect to server
|
||||||
|
server_addr.sin_port = htons(PORT_TCP);
|
||||||
|
if (connect(tcp_sock, (struct sockaddr *) &server_addr, addr_len) < 0) {
|
||||||
|
perror("Failed to connect to server");
|
||||||
|
exit(errno);
|
||||||
|
}
|
||||||
|
printf("Getting file...\n");
|
||||||
|
// get file name
|
||||||
|
ssize_t rcount;
|
||||||
|
if ((rcount = read(tcp_sock, buffer, BUF_SIZE)) <= 0) {
|
||||||
|
perror("Failed to get file name");
|
||||||
|
exit(errno);
|
||||||
|
}
|
||||||
|
fname_len = strlen(buffer) + 1;
|
||||||
|
strcpy(fname, buffer);
|
||||||
|
// if path is a directory, append file name to it
|
||||||
|
if (stat(abspath, &stat_buf) == 0 && S_ISDIR(stat_buf.st_mode)) {
|
||||||
|
abspath[abspath_len - 1] = '/';
|
||||||
|
abspath[abspath_len] = 0;
|
||||||
|
strcat(abspath, fname);
|
||||||
|
}
|
||||||
|
FILE *file = fopen(abspath, "wb");
|
||||||
|
if (file == NULL) {
|
||||||
|
perror("Failed to open file");
|
||||||
|
exit(errno);
|
||||||
|
}
|
||||||
|
if (rcount > fname_len) {
|
||||||
|
fwrite(buffer + fname_len, 1, rcount - fname_len, file);
|
||||||
|
}
|
||||||
|
// get file data
|
||||||
|
for (;;) {
|
||||||
|
rcount = read(tcp_sock, buffer, BUF_SIZE);
|
||||||
|
if (rcount == 0) {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
if (rcount < 0) {
|
||||||
|
perror("Failed to get file");
|
||||||
|
exit(errno);
|
||||||
|
}
|
||||||
|
fwrite(buffer, 1, rcount, file);
|
||||||
|
}
|
||||||
|
fclose(file);
|
||||||
|
printf("Success!\nFile path: %s\n", realpath(abspath, abspath));
|
||||||
|
}
|
||||||
|
|
||||||
|
int main(int argc, char **argv) {
|
||||||
|
if (argc == 1) {
|
||||||
|
printf("Use '-h' to view help.\n");
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
char *path = NULL,
|
||||||
|
*token = NULL;
|
||||||
|
int mode_share;
|
||||||
|
int opt;
|
||||||
|
while ((opt = getopt(argc, argv, "hs:g:o:V")) != -1) {
|
||||||
|
switch (opt) {
|
||||||
|
case 'h':
|
||||||
|
printf(
|
||||||
|
"(C) Masahiko AMANO aka H1K0, 2024—present\n"
|
||||||
|
"(https://github.com/H1K0/aiken)\n\n"
|
||||||
|
"Usage:\n"
|
||||||
|
" aiken <options> [arguments]\n\n"
|
||||||
|
"Options:\n"
|
||||||
|
" -h Print this help and exit\n"
|
||||||
|
" -s <path> Share the file of the specified <path>\n"
|
||||||
|
" -g <token> Get the file by its token\n"
|
||||||
|
" -o <path> Specify output path when getting file\n"
|
||||||
|
" (defaults to current working directory and\n"
|
||||||
|
" the initial file name)\n"
|
||||||
|
" -V Print version info and exit\n"
|
||||||
|
);
|
||||||
|
return 0;
|
||||||
|
case 'V':
|
||||||
|
printf("Aiken 0.1\n");
|
||||||
|
return 0;
|
||||||
|
case 's':
|
||||||
|
mode_share = 1;
|
||||||
|
path = realpath(optarg, path);
|
||||||
|
if (path == NULL) {
|
||||||
|
perror("Invalid share path");
|
||||||
|
return errno;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case 'g':
|
||||||
|
mode_share = 0;
|
||||||
|
token = optarg;
|
||||||
|
break;
|
||||||
|
case 'o':
|
||||||
|
if (mode_share) {
|
||||||
|
fprintf(stderr, "Warning: The option '-o' is ignored when sharing.\n");
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
path = optarg;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (mode_share) {
|
||||||
|
share(path);
|
||||||
|
} else {
|
||||||
|
get(token, path);
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
Loading…
x
Reference in New Issue
Block a user