#include <stdio.h>

#include <stdlib.h>

#include <memory.h>

#include <errno.h>

#include <sys/types.h>

#include <winsock2.h>





#include <openssl/rsa.h>       /* SSLeay stuff */

#include <openssl/crypto.h>

#include <openssl/x509.h>

#include <openssl/pem.h>

#include <openssl/ssl.h>

#include <openssl/err.h>



#pragma comment(lib, "ssleay32.lib")

#pragma comment(lib, "libeay32.lib")





/*所有需要的参数信息都在此处以#define的形式提供*/ 

#define CERTF "server.crt" /*服务端的证书(需经CA签名)*/ 

#define KEYF  "server.key" /*服务端的私钥(建议加密存储)*/ 

#define CACERT "ca.crt" /*CA 的证书*/ 

#define PORT 443 /*准备绑定的端口*/ 



#define CHK_NULL(x) if ((x)==NULL) exit (1)

#define CHK_ERR(err,s) if ((err)==-1) { perror(s); exit(1); }

#define CHK_SSL(err) if ((err)==-1) { ERR_print_errors_fp(stderr); exit(2); }



void main ()

{

  int err;

  int listen_sd;

  int sd;

  int client_len;

  SSL_CTX* ctx;

  SSL*     ssl;

  X509*    client_cert;

  char*    str;

  char     buf [4096];

  SSL_METHOD *meth;

  WSADATA wsaData;

  struct sockaddr_in sa_serv;

  struct sockaddr_in sa_cli;



  if(WSAStartup(MAKEWORD(2,2),&wsaData) != 0)

  { 

	  printf("WSAStartup()fail:%d\n",GetLastError()); 

	  return; 

  } 



  /* SSL preliminaries. We keep the certificate and key with the context. */

  SSL_load_error_strings(); /*为打印调试信息作准备*/ 

  OpenSSL_add_ssl_algorithms(); /*初始化*/ 



  meth = SSLv3_server_method(); /*采用什么协议(SSLv2/SSLv3/TLSv1)在此指定,TLSv1_server_method,SSLv23_server_method()*/ 



  ctx = SSL_CTX_new (meth);

  //CHK_NULL(ctx);

  if (!ctx) 

  {

    printf("创建SSL_CTX失败!\n");

    exit(2);

  }



  //设置为要求强制校验对方（客户端）证书SSL_VERIFY_PEER|SSL_VERIFY_FAIL_IF_NO_PEER_CERT

  SSL_CTX_set_verify(ctx,SSL_VERIFY_NONE,NULL); /*验证与否SSL_VERIFY_PEER*/ 

  //SSL_CTX_load_verify_locations(ctx,CACERT,NULL); /*若验证,则放置CA证书*/

  

  if (SSL_CTX_use_certificate_file(ctx, CERTF, SSL_FILETYPE_PEM) <= 0) 

  {

    printf("加载证书失败!\n");

    exit(3);

  }



  if (SSL_CTX_use_PrivateKey_file(ctx, KEYF, SSL_FILETYPE_PEM) <= 0)

  {

    printf("加载私钥失败!\n");

    exit(4);

  }

  

  if (!SSL_CTX_check_private_key(ctx)) {

    printf("密钥证书不匹配！\n");

    exit(5);

  }



  SSL_CTX_set_cipher_list(ctx,"RC4-MD5"); 

  SSL_CTX_set_mode(ctx, SSL_MODE_AUTO_RETRY);



  /* ----------------------------------------------- */

  /* Prepare TCP socket for receiving connections */

  /*开始正常的TCP socket过程.................................*/ 

  printf("Begin TCP socket...\n"); 

 



  listen_sd = socket (AF_INET, SOCK_STREAM, 0);   

  CHK_ERR(listen_sd, "socket");

  

  memset (&sa_serv, '\0', sizeof(sa_serv));

  sa_serv.sin_family      = AF_INET;

  sa_serv.sin_addr.s_addr = INADDR_ANY;

  sa_serv.sin_port        = htons (PORT);          /* Server Port number */

  

  err = bind(listen_sd, (struct sockaddr*) &sa_serv, sizeof (sa_serv));                   

  CHK_ERR(err, "bind");

	     

  /*接受TCP链接*/ 

  err = listen (listen_sd, 5);                    

  CHK_ERR(err, "listen");

  

  client_len = sizeof(sa_cli);

  sd = accept (listen_sd, (struct sockaddr*) &sa_cli, (int*)&client_len);

  CHK_ERR(sd, "accept");

  closesocket (listen_sd);



  printf ("Connection from %s, port %d\n",

	  inet_ntoa(sa_cli.sin_addr), sa_cli.sin_port);



  

  /* ----------------------------------------------- */

  /*TCP连接已建立,进行服务端的SSL过程. */ 

  printf("Begin server side SSL\n"); 



  ssl = SSL_new (ctx);  

  CHK_NULL(ssl);

  SSL_set_fd (ssl, sd);

  err = SSL_accept (ssl);

  printf("SSL_accept finished\n"); 

  CHK_SSL(err);

  

  /*打印所有加密算法的信息(可选)*/ 

  printf ("SSL connection using %s\n", SSL_get_cipher (ssl));

  

  /*得到客户端的证书并打印些信息(可选) */ 

  /*client_cert = SSL_get_peer_certificate (ssl);

  if (client_cert != NULL) 

  {

    printf ("Client certificate:\n");

    

    str = X509_NAME_oneline (X509_get_subject_name (client_cert), 0, 0);

    CHK_NULL(str);

    printf ("\t subject: %s\n", str);

    OPENSSL_free (str);

    

    str = X509_NAME_oneline (X509_get_issuer_name  (client_cert), 0, 0);

    CHK_NULL(str);

    printf ("\t issuer: %s\n", str);

    OPENSSL_free (str);

    

    //如不再需要,需将证书释放

    X509_free (client_cert);

  } 

  else

    printf ("Client does not have certificate.\n");*/



  /* 数据交换开始,用SSL_write,SSL_read代替write,read */ 

  int inlen = 0, nread = 0;

  for ( nread = 0; nread < sizeof(buf); nread += inlen )

  {

	  inlen = SSL_read(ssl, buf+nread,sizeof(buf)-nread);

	  CHK_SSL(inlen);

	  if ( inlen <=0 ) break;//出错或客户端无数据

	  fwrite(buf+nread, 1,inlen, stdout);//将接收到的信息打印到标准输出

  }

  

  buf[nread] = '\0';

  printf ("Got %d chars:'%s'\n", nread, buf);

  

  err = SSL_write (ssl, "I hear you.", strlen("I hear you.")); 

  CHK_SSL(err);



  SSL_shutdown(ssl);//关闭SSL连接



  closesocket (sd);

  SSL_free (ssl);

  SSL_CTX_free (ctx);

}

/* EOF - serv.cpp */