C語言socket如何進(jìn)行SSL/TLS加密

小樊
82
2024-10-09 02:02:34
欄目: 云計(jì)算

在C語言中，使用socket進(jìn)行SSL/TLS加密通信需要使用第三方庫，如OpenSSL。下面是一個(gè)簡單的示例，展示了如何使用OpenSSL庫創(chuàng)建一個(gè)SSL/TLS客戶端和服務(wù)器。

  1. 首先，確保已經(jīng)安裝了OpenSSL庫。在大多數(shù)Linux發(fā)行版中，可以使用包管理器安裝，例如：

    sudo apt-get install libssl-dev

  2. 創(chuàng)建一個(gè)SSL/TLS服務(wù)器：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <openssl/ssl.h>
#include <openssl/err.h>

#define PORT 8443
#define BUFFER_SIZE 1024

int main() {
    SSL_CTX *ctx = NULL;
    SSL *ssl = NULL;
    int server_fd, client_fd;
    struct sockaddr_in server_addr, client_addr;
    socklen_t addr_len = sizeof(client_addr);
    char buffer[BUFFER_SIZE];

    // 初始化SSL上下文
    SSL_library_init();
    SSL_load_error_strings();
    OpenSSL_add_all_algorithms();
    ctx = SSL_CTX_new(TLSv1_2_client_method());
    if (!ctx) {
        fprintf(stderr, "Failed to create SSL context\n");
        return 1;
    }

    // 創(chuàng)建socket
    server_fd = socket(AF_INET, SOCK_STREAM, 0);
    if (server_fd < 0) {
        perror("socket");
        return 1;
    }

    // 配置服務(wù)器地址
    memset(&server_addr, 0, sizeof(server_addr));
    server_addr.sin_family = AF_INET;
    server_addr.sin_addr.s_addr = INADDR_ANY;
    server_addr.sin_port = htons(PORT);

    // 綁定并監(jiān)聽socket
    if (bind(server_fd, (struct sockaddr *)&server_addr, sizeof(server_addr)) < 0) {
        perror("bind");
        return 1;
    }
    if (listen(server_fd, 5) < 0) {
        perror("listen");
        return 1;
    }

    printf("Server listening on port %d\n", PORT);

    // 接受客戶端連接
    client_fd = accept(server_fd, (struct sockaddr *)&client_addr, &addr_len);
    if (client_fd < 0) {
        perror("accept");
        return 1;
    }

    // 創(chuàng)建SSL對(duì)象
    ssl = SSL_new(ctx);
    if (!ssl) {
        fprintf(stderr, "Failed to create SSL object\n");
        return 1;
    }

    // 綁定socket到SSL對(duì)象
    SSL_set_fd(ssl, client_fd);

    // 建立SSL連接
    if (SSL_accept(ssl) <= 0) {
        fprintf(stderr, "Failed to establish SSL connection\n");
        return 1;
    }

    // 讀取并打印客戶端發(fā)送的數(shù)據(jù)
    while (1) {
        int len = SSL_read(ssl, buffer, BUFFER_SIZE);
        if (len > 0) {
            printf("Received from client: %s\n", buffer);
        } else if (len == 0) {
            break;
        } else {
            perror("SSL_read");
            break;
        }
    }

    // 清理資源
    SSL_free(ssl);
    SSL_CTX_free(ctx);
    close(client_fd);
    close(server_fd);

    return 0;
}
  1. 創(chuàng)建一個(gè)SSL/TLS客戶端：
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <openssl/ssl.h>
#include <openssl/err.h>

#define PORT 8443
#define BUFFER_SIZE 1024

int main() {
    SSL_CTX *ctx = NULL;
    SSL *ssl = NULL;
    int client_fd;
    struct sockaddr_in server_addr;
    char buffer[BUFFER_SIZE];

    // 初始化SSL上下文
    SSL_library_init();
    SSL_load_error_strings();
    OpenSSL_add_all_algorithms();
    ctx = SSL_CTX_new(TLSv1_2_client_method());
    if (!ctx) {
        fprintf(stderr, "Failed to create SSL context\n");
        return 1;
    }

    // 創(chuàng)建socket
    client_fd = socket(AF_INET, SOCK_STREAM, 0);
    if (client_fd < 0) {
        perror("socket");
        return 1;
    }

    // 配置服務(wù)器地址
    memset(&server_addr, 0, sizeof(server_addr));
    server_addr.sin_family = AF_INET;
    server_addr.sin_addr.s_addr = inet_addr("127.0.0.1");
    server_addr.sin_port = htons(PORT);

    // 連接到服務(wù)器
    if (connect(client_fd, (struct sockaddr *)&server_addr, sizeof(server_addr)) < 0) {
        perror("connect");
        return 1;
    }

    // 創(chuàng)建SSL對(duì)象
    ssl = SSL_new(ctx);
    if (!ssl) {
        fprintf(stderr, "Failed to create SSL object\n");
        return 1;
    }

    // 綁定socket到SSL對(duì)象
    SSL_set_fd(ssl, client_fd);

    // 建立SSL連接
    if (SSL_connect(ssl) <= 0) {
        fprintf(stderr, "Failed to establish SSL connection\n");
        return 1;
    }

    // 向服務(wù)器發(fā)送數(shù)據(jù)
    const char *message = "Hello, server!";
    SSL_write(ssl, message, strlen(message));

    // 讀取并打印服務(wù)器發(fā)送的數(shù)據(jù)
    while (1) {
        int len = SSL_read(ssl, buffer, BUFFER_SIZE);
        if (len > 0) {
            printf("Received from server: %s\n", buffer);
        } else if (len == 0) {
            break;
        } else {
            perror("SSL_read");
            break;
        }
    }

    // 清理資源
    SSL_free(ssl);
    SSL_CTX_free(ctx);
    close(client_fd);

    return 0;
}
  1. 編譯并運(yùn)行服務(wù)器和客戶端：

對(duì)于服務(wù)器：

gcc server.c -o server -lssl -lcrypto
./server

對(duì)于客戶端：

gcc client.c -o client -lssl -lcrypto
./client

現(xiàn)在，服務(wù)器和客戶端應(yīng)該通過SSL/TLS加密進(jìn)行通信。請(qǐng)注意，這個(gè)示例僅用于演示目的，實(shí)際應(yīng)用中可能需要更多的錯(cuò)誤處理和功能。

0 踩