#pragma comment(lib,"libssl.lib")
#pragma comment(lib,"libcrypto.lib")

#include<tchar.h>
#include<WinSock2.h>
#include<WS2tcpip.h>
#include<iostream>
#include<openssl\ssl.h>
#pragma comment(lib,"ws2_32.lib")
#pragma comment(lib,"libssl.lib")
#pragma comment(lib,"libcrypto.lib")

CONST INT RECV_SIZE = 8192;

INT _tmain(INT argc, LPTSTR argv[])
{
 //啟動wsa
 WSADATA wsadData;
 WSAStartup(MAKEWORD(2, 2), &wsadData);

 //獲取Host的IP地址等信息
 ADDRINFOT aiHints;
 ZeroMemory(&aiHints, sizeof(ADDRINFOT));
 aiHints.ai_family = AF_INET;
 aiHints.ai_flags = AI_PASSIVE;
 aiHints.ai_protocol = 0;
 aiHints.ai_socktype = SOCK_STREAM;
 std::wstring wstrHost = TEXT("www.baidu.com");
 PADDRINFOT paiResult;
 GetAddrInfo(wstrHost.c_str(), NULL, &aiHints, &paiResult);

 //創建套接字
 SOCKET sSocket = socket(AF_INET, SOCK_STREAM, 0);
 if (sSocket == SOCKET_ERROR)
 {
  std::wcout << "Error socket" << std::endl;
  return -1;
 }

 //連接Host
 SOCKADDR_IN sinHost;
 sinHost.sin_addr = ((LPSOCKADDR_IN)paiResult->ai_addr)->sin_addr;
 sinHost.sin_family = AF_INET;
 sinHost.sin_port = htons(443);
 if (connect(sSocket, (LPSOCKADDR)&sinHost, sizeof(SOCKADDR_IN)) == SOCKET_ERROR)
 {
  std::wcout << "Error connect" << std::endl;
  return -1;
 }

 //初始化OpenSSL庫
 //（雖然不知道為什么，但是不加這三行似乎并不會導致什么問題，在不加這3行的情況下測試了幾個網站并沒有發現任何問題喵）
 SSL_library_init();
 SSLeay_add_ssl_algorithms();
 SSL_load_error_strings();

 //創建SSL會話環境等
 SSL_CTX *pctxSSL = SSL_CTX_new(TLSv1_2_client_method());
 if (pctxSSL == NULL)
 {
  std::wcout << "Error SSL_CTX_new" << std::endl;
  return -1;
 }
 SSL *psslSSL = SSL_new(pctxSSL);
 if (psslSSL == NULL)
 {
  std::wcout << "Error SSL_new" << std::endl;
  return -1;
 }
 SSL_set_fd(psslSSL, sSocket);
 INT iErrorConnect = SSL_connect(psslSSL);
 if (iErrorConnect < 0)
 {
  std::wcout << "Error SSL_connect, iErrorConnect=" << iErrorConnect << std::endl;
  return -1;
 }
 std::wcout << "SSL connection using " << SSL_get_cipher(psslSSL) << std::endl;

 //發包
 std::string strWrite =
  "GET https://www.baidu.com/ HTTP/1.1\r\n"
  "Host: www.baidu.com\r\n"
  "Connection: close\r\n\r\n";
 INT iErrorWrite = SSL_write(psslSSL, strWrite.c_str(), strWrite.length()) < 0;
 if (iErrorWrite < 0)
 {
  std::wcout << "Error SSL_write" << std::endl;
  return -1;
 }

 //收包并輸出
 //這里接受的是char形式的，所以中文會亂碼
 //如果要正常顯示中文，需要再轉換為wchar_t或std::wstring
 LPSTR lpszRead = new CHAR[RECV_SIZE];
 INT iLength=1;
 while (iLength >= 1)
 {
  iLength = SSL_read(psslSSL, lpszRead, RECV_SIZE - 1);
  if (iLength < 0)
  {
   std::wcout << "Error SSL_read" << std::endl;
   delete[] lpszRead;
   return -1;
  }
  lpszRead[iLength] = TEXT('\0');
  std::wcout << lpszRead;
 }
 delete[] lpszRead;

 return 0;
}