Forum Klix.ba

Code: Select all

// file_transfer.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <winsock2.h>

#define BUFFER_SIZE 1024

void error(const char *msg) {
    perror(msg);
    exit(EXIT_FAILURE);
}

int main(int argc, char *argv[]) {
    if (argc != 5) {
        fprintf(stderr, "Usage: %s <'s'/'r'> <port> <ip_address> <filename>\n", argv[0]);
        exit(EXIT_FAILURE);
    }

    WSADATA wsa;
    if (WSAStartup(MAKEWORD(2, 2), &wsa) != 0) {
        error("Failed to initialize Winsock");
    }

    char mode = argv[1][0];
    int port = atoi(argv[2]);
    const char *ip_address = argv[3];
    const char *filename = argv[4];

    if (mode == 's') {
        // Client (Send file)
        FILE *file = fopen(filename, "rb");
        if (!file) {
            error("Error opening file");
        }

        SOCKET sock = socket(AF_INET, SOCK_STREAM, 0);
        if (sock == INVALID_SOCKET) {
            error("Error creating socket");
        }

        struct sockaddr_in server_addr;
        server_addr.sin_family = AF_INET;
        server_addr.sin_port = htons(port);
        server_addr.sin_addr.s_addr = inet_addr(ip_address);

        if (connect(sock, (struct sockaddr *)&server_addr, sizeof(server_addr)) == SOCKET_ERROR) {
            error("Error connecting to server");
        }

        char buffer[BUFFER_SIZE];
        int nread;
        while ((nread = fread(buffer, 1, BUFFER_SIZE, file)) > 0) {
            send(sock, buffer, nread, 0);
        }

        fclose(file);
        closesocket(sock);
    } else if (mode == 'r') {
        // Server (Receive file)
        FILE *file = fopen(filename, "wb");
        if (!file) {
            error("Error creating file");
        }

        SOCKET server_sock = socket(AF_INET, SOCK_STREAM, 0);
        if (server_sock == INVALID_SOCKET) {
            error("Error creating server socket");
        }

        struct sockaddr_in server_addr, client_addr;
        server_addr.sin_family = AF_INET;
        server_addr.sin_port = htons(port);
        server_addr.sin_addr.s_addr = INADDR_ANY;

        if (bind(server_sock, (struct sockaddr *)&server_addr, sizeof(server_addr)) == SOCKET_ERROR) {
            error("Error binding socket");
        }

        listen(server_sock, 1);
        socklen_t client_len = sizeof(client_addr);
        SOCKET client_sock = accept(server_sock, (struct sockaddr *)&client_addr, &client_len);
        if (client_sock == INVALID_SOCKET) {
            error("Error accepting client connection");
        }

        char buffer[BUFFER_SIZE];
        int nread;
        while ((nread = recv(client_sock, buffer, BUFFER_SIZE, 0)) > 0) {
            fwrite(buffer, 1, nread, file);
        }

        fclose(file);
        closesocket(client_sock);
        closesocket(server_sock);
    } else {
        fprintf(stderr, "Invalid command. Use 's' for send or 'r' for receive.\n");
        exit(EXIT_FAILURE);
    }

    WSACleanup();
    return 0;
}

Code: Select all

g++ -o file_transfer.exe file_transfer.c -lws2_32 -static

Code: Select all

file_transfer.exe s <port> <ip_address> <filename>  # Send mode
file_transfer.exe r <port> <ip_address> <filename>  # Receive mode

Code: Select all

#define _WIN32_WINNT 0x501
#include <stdio.h>
#include <stdlib.h>
#include <winsock2.h>

#define BUF_SIZE 1024

int send_file(const char *filename, const char *ip, int port) {
    FILE *file = fopen(filename, "rb");
    if (!file) {
        perror("fopen");
        return -1;
    }

    WSADATA wsaData;
    SOCKET sock = INVALID_SOCKET;
    if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) {
        printf("Failed. Error Code : %d", WSAGetLastError());
        return -1;
    }

    sock = socket(AF_INET, SOCK_STREAM, 0);
    struct sockaddr_in server_addr;
    server_addr.sin_family = AF_INET;
    server_addr.sin_port = htons(port);
    server_addr.sin_addr.s_addr = inet_addr(ip);

    if (connect(sock, (struct sockaddr*)&server_addr, sizeof(server_addr)) < 0) {
        perror("connect");
        return -1;
    }

    char buf[BUF_SIZE];
    size_t bytes_read;
    while ((bytes_read = fread(buf, 1, BUF_SIZE, file)) > 0) {
        send(sock, buf, bytes_read, 0);
    }

    fclose(file);
    closesocket(sock);
    WSACleanup();

    return 0;
}

int receive_file(const char *filename, int port) {
    FILE *file = fopen(filename, "wb");
    if (!file) {
        perror("fopen");
        return -1;
    }

    WSADATA wsaData;
    SOCKET sock = INVALID_SOCKET, new_sock;
    if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) {
        printf("Failed. Error Code : %d", WSAGetLastError());
        return -1;
    }

    sock = socket(AF_INET, SOCK_STREAM, 0);
    struct sockaddr_in server_addr;
    server_addr.sin_family = AF_INET;
    server_addr.sin_port = htons(port);
    server_addr.sin_addr.s_addr = INADDR_ANY;

    if (bind(sock, (struct sockaddr*)&server_addr, sizeof(server_addr)) < 0) {
        perror("bind");
        return -1;
    }

    listen(sock, 1);

    new_sock = accept(sock, NULL, NULL);
    if (new_sock < 0) {
        perror("accept");
        return -1;
    }

    char buf[BUF_SIZE];
    int bytes_received;
    while ((bytes_received = recv(new_sock, buf, BUF_SIZE, 0)) > 0) {
        fwrite(buf, 1, bytes_received, file);
    }

    fclose(file);
    closesocket(sock);
    closesocket(new_sock);
    WSACleanup();

    return 0;
}

int main(int argc, char *argv[]) {
    if (argc != 5) {
        fprintf(stderr, "Usage: %s <s/r> <port> <ip> <filename>\n", argv[0]);
        return -1;
    }

    const char *mode = argv[1];
    int port = atoi(argv[2]);
    const char *ip = argv[3];
    const char *filename = argv[4];

    if (mode[0] == 's') {
        return send_file(filename, ip, port);
    } else if (mode[0] == 'r') {
        return receive_file(filename, port);
    } else {
        fprintf(stderr, "Invalid mode. Use 's' to send or 'r' to receive.\n");
        return -1;
    }
}