OpenSSL socket 服务端

这是我从网上找来改好的，需要添加 include 和 lib 路径

需要自己手动生成 pem 文件，在注释里面有命行，自己动手。

#define WIN32_LEAN_AND_MEAN

#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <sys/types.h>
#include <Windows.h>
#include <WinSock2.h>
#include <MSWSock.h>
#include <WS2tcpip.h>
#include <openssl/ssl.h>
#include <openssl/err.h>
#pragma comment(lib, "ws2_32.lib")
#pragma comment(lib, "libssl.lib")
#pragma comment(lib, "libcrypto.lib")
#pragma comment(lib, "Crypt32.lib")

// openssl 生成公钥和私钥的方法：
// openssl genrsa -out privkey.pem 2048 
// openssl req -config openssl.cnf -new -x509 -key privkey.pem -out cacert.pem -days 1095

#define MAXBUF 1024

/************关于本文档*********************************************
 ************filename: ssl-server.c
 *************purpose: 演示利用 OpenSSL 库进行基于 IP层的 SSL 加密通讯的方法，这是服务器端例子
 **********************************************************************************/

#define bzero ZeroMemory
#define close closesocket

SSL_CTX *ctx;
bool InitSslAll(char* publicPem, char* privatePem)
{
    /* SSL 库初始化 */
    SSL_library_init();

    /* 载入所有 SSL 算法 */
    OpenSSL_add_all_algorithms();

    /* 载入所有 SSL 错误消息 */
    SSL_load_error_strings();

    /* 以 SSL V2 和 V3 标准兼容方式产生一个 SSL_CTX ，即 SSL Content Text */
    ctx = SSL_CTX_new(SSLv23_server_method());

    /* 也可以用 SSLv2_server_method() 或 SSLv3_server_method() 单独表示 V2 或 V3标准 */
    if (ctx == NULL) {
        ERR_print_errors_fp(stdout);
        return false;
    }

    /* 载入用户的数字证书， 此证书用来发送给客户端。 证书里包含有公钥 */
    if (SSL_CTX_use_certificate_file(ctx, publicPem, SSL_FILETYPE_PEM) <= 0)
    {
        ERR_print_errors_fp(stdout);
        return false;
    }

    /* 载入用户私钥 */
    if (SSL_CTX_use_PrivateKey_file(ctx, privatePem, SSL_FILETYPE_PEM) <= 0)
    {
        ERR_print_errors_fp(stdout);
        return false;
    }

    /* 检查用户私钥是否正确 */
    if (!SSL_CTX_check_private_key(ctx))
    {
        ERR_print_errors_fp(stdout);
        return false;
    }
    return true;
}

int main(int argc, char **argv){
    SOCKET sockfd, new_fd;
    socklen_t len;
    struct sockaddr_in my_addr, their_addr;
    unsigned int myport, lisnum;
    char buf[MAXBUF + 1];

    if (argv[1])
        myport = atoi(argv[1]);
    else
        myport = 7838;
    if (argv[2])
        lisnum = atoi(argv[2]);
    else
        lisnum = 2;

    InitSslAll(argv[4], argv[5]);

    WSADATA wsaData;
    int iResult = 0;
    iResult = WSAStartup(MAKEWORD(2, 2), &wsaData);
    if (iResult != NO_ERROR) {
        wprintf(L"Error at WSAStartup()
");
        return 1;
    }

    /* 开启一个 socket 监听 */
    if ((sockfd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) == -1)
    {
        perror("socket");
        exit(1);
    } else
        printf("socket created
");

    bzero(&my_addr, sizeof(my_addr));
    my_addr.sin_family = PF_INET;
    my_addr.sin_port = htons(myport);
    if (argv[3])
            inet_pton(AF_INET, argv[3], &my_addr.sin_addr.s_addr);
    else
        my_addr.sin_addr.s_addr = INADDR_ANY;

    if (bind(sockfd, (struct sockaddr *)&my_addr, sizeof(struct sockaddr))        == -1)
    {
        printf("bind error:%d", WSAGetLastError());
        exit(1);
    }
    else
        printf("binded
");

    if (listen(sockfd, lisnum) == -1)
    {
        printf("listen error:%d", WSAGetLastError());
        exit(1);
    } else
        printf("begin listen
");

        while (1)
        {
            SSL *ssl;
            len = sizeof(struct sockaddr);

            /* 等待客户端连上来 */
            if ((new_fd =accept(sockfd, (struct sockaddr *) &their_addr,&len)) == -1)
            {
                printf("accept error:%d", WSAGetLastError());
                exit(errno);
            }
            else
            {
                char addr[64] = { 0 };

                inet_ntop(AF_INET, &their_addr.sin_addr, addr, 64);
                printf("server: got connection from %s, port %d, socket %d
"
                    , addr,    ntohs(their_addr.sin_port), new_fd);
            }


            /* 基于 ctx 产生一个新的 SSL */
            ssl = SSL_new(ctx);

            /* 将连接用户的 socket 加入到 SSL */

            SSL_set_fd(ssl, new_fd);

            /* 建立 SSL 连接 */

            if (SSL_accept(ssl) == -1)
            {
                perror("accept");
                close(new_fd);
                break;
            }

            /* 开始处理每个新连接上的数据收发 */

            bzero(buf, MAXBUF + 1);
            strcpy_s(buf, MAXBUF, "server->client");

            /* 发消息给客户端 */
            len = SSL_write(ssl, buf, strlen(buf));
            if (len <= 0) {
                char err[256] = { 0 };
                strerror_s(err, errno);
                printf("消息'%s'发送失败！错误代码是%d，错误信息是'%s'
",
                    buf, errno, err);
                goto finish;
            } else
                printf("消息'%s'发送成功，共发送了%d个字节！
",
                    buf, len);
            bzero(buf, MAXBUF + 1);

            /* 接收客户端的消息 */
            len = SSL_read(ssl, buf, MAXBUF);
            if (len > 0)
                printf("接收消息成功:'%s'，共%d个字节的数据
",
                    buf, len);
            else
            {
                char err[256] = { 0 };
                strerror_s(err, errno);
                printf("消息接收失败！错误代码是%d，错误信息是'%s'
",
                    errno, err);
            }


            /* 处理每个新连接上的数据收发结束 */
        finish:

            /* 关闭 SSL 连接 */
            SSL_shutdown(ssl);
            /* 释放 SSL */
            SSL_free(ssl);

            /* 关闭 socket */

            close(new_fd);
        }

    /* 关闭监听的 socket */

    close(sockfd);

    /* 释放 CTX */

    SSL_CTX_free(ctx);

    return 0;

}