OpenSSL库应用的一个例子

小锁

OpenSSL的文档超级不全，找来了这个demo，抛砖引玉！

1. 
   
2. /*  ssl_sockets.c  */
   
3. 
   
4. // This code is provided "as is" with NO WARRANTY expressed or
   
5. // implied. You may use it freely at your own risk.
   
6. 
   
7. #include <sys/socket.h>
   
8. #include <netinet/in.h>
   
9. #include <netdb.h>
   
10. #include <stdio.h>
    
11. #include <openssl/ssl.h>
    
12. #include <assert.h>
    
13. 
    
14. void die(const char *ptr){
    
15.         assert(ptr);
    
16.         puts(ptr);
    
17.         exit(-1);
    
18. }
    
19. 
    
20. int writeToSSLSocket(SSL *ssl, char *request) {
    
21.    int bytesWritten;
    
22. 
    
23.    bytesWritten = SSL_write (ssl, request, strlen(request));
    
24.    return bytesWritten;
    
25. }
    
26. 
    
27. int readn(SSL *ssl, char *buf,int n) {
    
28.    unsigned char *p;
    
29.    int i;
    
30.    int nread;
    
31. 
    
32.    p = (unsigned char *)buf;
    
33.    i = 0;
    
34.    while(i < n) {
    
35.       nread = SSL_read(ssl, p, n-i);
    
36.       if(nread <= 0) return(i);
    
37.       p += nread;
    
38.       i += nread;
    
39.    }
    
40.    return(i);
    
41. }
    
42. 
    
43. void printSSLServerResponse(SSL *ssl) {
    
44.    int bytesRead;
    
45.    int readSize = 4096;
    
46.    char buf[readSize + 2];
    
47. 
    
48.    memset (buf, 0, sizeof(buf));
    
49.    while (bytesRead = readn(ssl, buf, readSize)) {
    
50.       printf(buf);
    
51.       memset (buf, 0, sizeof(buf));
    
52.    }
    
53. }
    
54. 
    
55. int openSocket(char *host, int port) {
    
56.    long ipAddress;
    
57.    struct hostent* hostInfo;
    
58.    struct sockaddr_in sockInfo;
    
59.    int sock;
    
60. 
    
61.    memset(&sockInfo, 0, sizeof(sockInfo));
    
62.    sockInfo.sin_family = AF_INET;
    
63.    sockInfo.sin_port = htons(port);
    
64.    ipAddress = inet_addr(host);
    
65.    if (ipAddress < 0) {
    
66.       hostInfo = gethostbyname(host);
    
67.       ipAddress = *(long *)*hostInfo->h_addr_list;
    
68.    }
    
69.    sockInfo.sin_addr.s_addr = ipAddress;
    
70. 
    
71.    // Open the socket
    
72.    if ((sock = socket(AF_INET, SOCK_STREAM, 0)) < 0)
    
73.       die("Can't open socket\n");
    
74. 
    
75.    // And connect
    
76.    if (connect (sock, (struct sockaddr*)&sockInfo, sizeof(sockInfo)) == -1)
    
77.       die("Cannot connect to socket\n");
    
78. 
    
79.    return sock;
    
80. }
    
81. 
    
82. int SSLsocket(SSL *ssl, char *host, int port) {
    
83.    int sock, result;
    
84. 
    
85.    sock = openSocket(host, port);
    
86.    SSL_set_fd(ssl, sock);
    
87.    result = SSL_connect(ssl);
    
88.    return result;
    
89. }
    

复制代码

1. 
   
2. /* ssl_main.c */
   
3. // This code is provided "as is" with NO WARRANTY expressed or
   
4. // implied. You may use it freely at your own risk.
   
5. 
   
6. #include <stdio.h>
   
7. #include <stdlib.h>
   
8. #include <openssl/ssl.h>
   
9. 
   
10. extern void die(char *s);
    
11. extern char *formRequest();
    
12. extern char *getHost(char *hostname, int *portNum, char *arg);
    
13. extern int SSLsocket(SSL *ssl, char *host, int port);
    
14. extern int writeToSSLSocket(SSL *ssl, char *request);
    
15. extern void printSSLServerResponse(SSL *ssl);
    
16. 
    
17. int main(int argc, char *argv[]) {
    
18.    char hostName[128];
    
19.    int i;
    
20.    int port;
    
21.    int result;
    
22.    char *request;
    
23.    SSL *ssl = NULL;
    
24.    SSL_CTX *ssl_ctx = NULL;
    
25. 
    
26.    if (argc < 2) {
    
27.       die("Usage: paul2 hostname:port");
    
28.    }
    
29. 
    
30.    strcpy(hostName, argv[1]);
    
31.    {
    
32.            char *ptr;
    
33.            ptr = strchr(hostName, ':');
    
34.            if(ptr){
    
35.                    *ptr = '\0';
    
36.                    port = atoi(ptr + 1);
    
37.            }else{
    
38.                    port = 443;
    
39.            }
    
40.    }
    
41. 
    
42.    // init SSL socket
    
43.    SSL_load_error_strings();
    
44.    SSLeay_add_ssl_algorithms();
    
45.    ssl_ctx = SSL_CTX_new(SSLv23_client_method());
    
46.    ssl = SSL_new(ssl_ctx);
    
47.    result = SSLsocket(ssl, hostName, port);
    
48. 
    
49.    char buf[1024];
    
50.    while(fgets(buf, 1024, stdin)){
    
51.                 result = writeToSSLSocket(ssl, buf);
    
52.                 if(strlen(buf) == 1){
    
53.                         buf[0] = '\n';
    
54.                         break;
    
55.                 }
    
56.    }
    
57.    printSSLServerResponse(ssl);
    
58. 
    
59.    SSL_shutdown(ssl);
    
60.    free(ssl);
    
61.    if (ssl_ctx != NULL) SSL_CTX_free(ssl_ctx);
    
62. 
    
63.    return 0;
    
64. }
    

复制代码