#pragma once
#include <WinSock2.h>
#pragma comment(lib, "ws2_32.lib")
 
class MySocket
{
protected:
  SOCKET m_hSocket;
public:
 
  // 獲取對端Socket用戶IP端口等
  BOOL GetPeerName(char* rSocketAddress, UINT& rSocketPort)
  {
    sockaddr_in name = { AF_INET };
    int lenname = sizeof(name);
    if (getpeername(m_hSocket, (sockaddr*)&name, &lenname) < 0)
      return false;
    strcpy(rSocketAddress, inet_ntoa(name.sin_addr));
    rSocketPort = htons(name.sin_port);
    return true;
  }
 
  // 獲取本機(jī)Socket用戶IP端口等
  BOOL GetSockName(char* rSocketAddress, UINT& rSocketPort)
  {
    sockaddr_in name = { AF_INET };
    int lenname = sizeof(name);
    if (getsockname(m_hSocket, (sockaddr*)&name, &lenname) < 0)
      return false;
    strcpy(rSocketAddress, inet_ntoa(name.sin_addr));
    rSocketPort = htons(name.sin_port);
    return true;
  }
 
  // 獲取當(dāng)前用戶SocketID
  BOOL GetSocketID()
  {
    return m_hSocket;
  }
 
  // 創(chuàng)建套接字
  BOOL Create(UINT nSocketPort = 0, int nSockType = SOCK_STREAM, LPCTSTR lpszSocketAddress = NULL)
  {
 
    // 創(chuàng)建套接字
    m_hSocket = socket(AF_INET, nSockType, 0);
    if (m_hSocket == INVALID_SOCKET)
      return false;
 
    // 設(shè)置IP地址和端口
    sockaddr_in sa = { AF_INET };
    sa.sin_port = htons(nSocketPort);
    if (lpszSocketAddress)
      sa.sin_addr.s_addr = inet_addr(lpszSocketAddress);
 
    // 綁定套接字和IP地址端口
    return !bind(m_hSocket, (sockaddr*)&sa, sizeof(sa));
  }
 
  // 接受客戶請求
  BOOL Accept(MySocket& rConnectedSock, LPSTR szIp = NULL, UINT* nPort = NULL)
  {
    sockaddr_in sa = { AF_INET };
    int nLen = sizeof(sa);
    rConnectedSock.m_hSocket = accept(this->m_hSocket, (sockaddr*)&sa, &nLen);
    if (rConnectedSock.m_hSocket == INVALID_SOCKET)
      return false;
    if (szIp)
      strcpy(szIp, inet_ntoa(sa.sin_addr));
    if (nPort)
      *nPort = htons(sa.sin_port);
    return true;
  }
 
  // 連接服務(wù)端
  BOOL Connection(LPCSTR lpszHostAddress, UINT nPort)
  {
    sockaddr_in sa = { AF_INET };
    sa.sin_port = htons(nPort);
    sa.sin_addr.s_addr = inet_addr(lpszHostAddress);
    return !connect(m_hSocket, (sockaddr*)&sa, sizeof(sa));
  }
 
  // 偵聽
  BOOL Listen(int nConnectionBacklog = 5)
  {
    return !listen(m_hSocket, nConnectionBacklog);
  }
 
  // 逐條發(fā)送
  int Send(const void* lpBuf, int nBufLen, int nFlags = 0)
  {
    return send(m_hSocket, (LPCSTR)lpBuf, nBufLen, nFlags);
  }
 
  // 發(fā)送整個緩沖區(qū)
  int SendTo(const void* lpBuf, int nBufLen, UINT nHostPort, LPCSTR lpszHostAddress = NULL,
    int nFlags = 0)
  {
    sockaddr_in to = { AF_INET };
    to.sin_port = htons(nHostPort);
    to.sin_addr.s_addr = inet_addr(lpszHostAddress);
    return sendto(m_hSocket, (LPCSTR)lpBuf, nBufLen, nFlags, (sockaddr*)&to, sizeof(to));
  }
 
  // 逐條接收
  int Receive(void* lpBuf, int nBufLen, int nFlags = 0)
  {
    return recv(m_hSocket, (LPTSTR)lpBuf, nBufLen, nFlags);
  }
 
  // 接收整個緩沖區(qū)
  int ReceiveFrom(void* lpBuf, int nBufLen, char* rSocketAddress, UINT& rSocketPort, int nFlags = 0)
  {
    sockaddr_in from = { AF_INET };
    int lenFrom = sizeof(from);
    int n = recvfrom(m_hSocket, (LPSTR)lpBuf, nBufLen, nFlags, (sockaddr*)&from, &lenFrom);
    strcpy(rSocketAddress, inet_ntoa(from.sin_addr));
    rSocketPort = htons(from.sin_port);
    return n;
  }
 
  // 關(guān)閉套接字
  void Close()
  {
    closesocket(m_hSocket);
    m_hSocket = INVALID_SOCKET;
  }
  MySocket()
  {
    WSADATA wsaData;
    WSAStartup(0x0202, &wsaData);
    m_hSocket = INVALID_SOCKET;
 
  }
  ~MySocket()
  {
    Close();
  }
};

#define _CRT_SECURE_NO_WARNINGS
#define _WINSOCK_DEPRECATED_NO_WARNINGS
 
#include <iostream>
#include <process.h>
#include "MySocket.hpp"
 
using namespace std;
 
void ClientPro(void* ptr)
{
    // 初始化
    MySocket* pSock = (MySocket*)ptr;
    MySocket server_socket = *pSock;
    server_socket.Send((const char *)"Welcome to LyServer", 19);
 
    // 獲取客戶端信息
    char sIp[20];
    UINT nPort;
    server_socket.GetPeerName(sIp, nPort);
 
    while (true)
    {
        char szBuffer[4096] = { 0 };
 
        // 接收客戶返回消息
        int ref = server_socket.Receive(szBuffer, sizeof(szBuffer));
        if (ref <= 0)
        {
            std::cout << "客戶: " << sIp << ":" << nPort << " [已斷開]" << std::endl;
            break;
        }
 
        std::cout << "地址: " << sIp << ":" << nPort << " 接收命令: " << szBuffer << std::endl;
 
        // 選擇不同的命令
        if (strcmp(szBuffer, "list\n") == 0)
        {
            std::cout << "輸出文件" << std::endl;
        }
        else if (strcmp(szBuffer, "download\n") == 0)
        {
            std::cout << "下載文件" << std::endl;
        }
        else if (strcmp(szBuffer, "upload\n") == 0)
        {
            std::cout << "上傳文件" << std::endl;
        }
 
        // 返回給客戶端
        server_socket.Send((char*)"ok", 2);
    }
}
 
int main(int argc, char *argv[])
{
    MySocket sock;
    if (!sock.Create(8233, SOCK_STREAM, "127.0.0.1"))
    {
        return -1;
    }
 
    // 獲取本機(jī)信息
    char sSevIp[20];
    UINT nSevPort;
    sock.GetSockName(sSevIp, nSevPort);
    std::cout << "服務(wù)端: " << sSevIp << ":" << nSevPort << " 服務(wù)器啟動成功" << std::endl;
 
    sock.Listen(5);
 
    // 獲取客戶端信息
    char sIp[20];
    UINT nPort;
 
    MySocket ptr;
    while (true)
    {
        // 當(dāng)有新用戶進(jìn)來自動創(chuàng)建一個線程來維持會話
        sock.Accept(ptr, sIp, &nPort);
        std::cout << "客戶: " << sIp << ":" << nPort << " [已登錄]" << std::endl;
 
        // 多線程
        _beginthread(ClientPro, 0, &ptr);
    }
    return 0;
}

#define _CRT_SECURE_NO_WARNINGS
#define _WINSOCK_DEPRECATED_NO_WARNINGS
#include <iostream>
#include "MySocket.hpp"
 
using namespace std;
 
int main(int argc, char* argv[])
{
  MySocket sock;
  if (!sock.Create(0, SOCK_STREAM))
  {
    return -1;
  }
 
  // 獲取本機(jī)信息
  char sClientIp[20];
  UINT nClientPort;
  sock.GetSockName(sClientIp, nClientPort);
  std::cout << "服務(wù)端: " << sClientIp << ":" << nClientPort << " 服務(wù)器啟動成功" << std::endl;
 
  if (!sock.Connection("127.0.0.1", 8233))
  {
    cout << "連接服務(wù)器失敗" << GetLastError() << endl;
    return -1;
  }
 
  char szBuffer[4096] = { 0 };
  int ref = sock.Receive(szBuffer, sizeof(szBuffer));
  szBuffer[ref] = 0;
  std::cout << "服務(wù)端回應(yīng): " << szBuffer << std::endl;
 
  while (true)
  {
 
  // 循環(huán)接受輸入
  input:
    memset(szBuffer, 0, 4096);
    std::cout << "Input CMD > ";
 
    // 接收輸入命令
    int inputLine = 0;
    while ((szBuffer[inputLine++] = getchar()) != '\n');
    if (strlen(szBuffer) == 1)
      goto input;
 
    // 發(fā)送數(shù)據(jù)
    sock.Send(szBuffer, 4096, 0);
 
    // 接收回顯
    memset(szBuffer, 0, 4096);
    sock.Receive(szBuffer, 4096, 0);
    std::cout << "服務(wù)端回顯: " << szBuffer << std::endl;
 
  }
  sock.Close();
  return 0;
}

#define _CRT_SECURE_NO_WARNINGS
#define _WINSOCK_DEPRECATED_NO_WARNINGS
 
#include <iostream>
#include <process.h>
#include <vector>
#include "MySocket.hpp"
 
using namespace std;
 
// 登錄狀態(tài)記錄
typedef struct
{
  char UserName[32];
  int SocketID;
}loginPool;
 
// ------------------------------------------------------------------------
// 用戶登錄驗證代碼部分
 
std::vector<loginPool> login_pool_vect;
 
// 檢查用戶ID是否存在與容器內(nèi),如果存在則返回用戶名
bool is_login(std::vector<loginPool> &ptr, int socket_id)
{
  for (int x = 0; x < ptr.size(); x++)
  {
    if (ptr[x].SocketID == socket_id)
    {
      return true;
    }
  }
  return false;
}
 
// 用戶登錄驗證
bool login(char *username, char *password, int socket_id)
{
  if ((strcmp(username, "lyshark") == 0) && (strcmp(password, "123123") == 0))
  {
    // 如果在則增加一個socket登錄標(biāo)志
    loginPool pool_ptr;
    pool_ptr.SocketID = socket_id;
    strcpy(pool_ptr.UserName, "lyshark");
 
    login_pool_vect.push_back(pool_ptr);
    return true;
  }
  else if ((strcmp(username, "admin") == 0) && (strcmp(password, "123456") == 0))
  {
    // 如果在則增加一個socket登錄標(biāo)志
    loginPool pool_ptr;
    pool_ptr.SocketID = socket_id;
    strcpy(pool_ptr.UserName, "lyshark");
 
    login_pool_vect.push_back(pool_ptr);
    return true;
  }
  return false;
}
 
// 根據(jù)傳入ID從容器內(nèi)彈出一個節(jié)點
bool logout(std::vector<loginPool> &ptr, int socket_id)
{
  for (vector<loginPool>::iterator it = ptr.begin(); it != ptr.end(); it++)
  {
    if (it->SocketID == socket_id)
    {
      // 彈出指定結(jié)構(gòu)體
      ptr.erase(it);
      return true;
    }
  }
  return false;
}
 
// ------------------------------------------------------------------------
// 響應(yīng)客戶端的子線程(主要功能實現(xiàn)部分)
void ClientPro(void* ptr)
{
  // 初始化
  MySocket* pSock = (MySocket*)ptr;
  MySocket server_socket = *pSock;
  server_socket.Send((const char *)"Welcome to LyShark Mini Server", 31);
 
  // 獲取客戶端信息
  char sIp[20];
  UINT nPort;
  server_socket.GetPeerName(sIp, nPort);
 
  while (true)
  {
    char szBuffer[4096] = { 0 };
    int sid = pSock->GetSocketID();
 
    int ref = server_socket.Receive(szBuffer, sizeof(szBuffer));
    if (ref <= 0)
    {
      logout(login_pool_vect, sid);
      std::cout << "客戶: " << sIp << ":" << nPort << " [已斷開]" << std::endl;
      break;
    }
 
    std::cout << "地址: " << sIp << ":" << nPort << " 接收命令: " << szBuffer << std::endl;
 
    // 用戶登錄
    if (strcmp(szBuffer, "login\n") == 0)
    {
      char recv_username[32] = { 0 };
      char recv_password[32] = { 0 };
 
      // 接收用戶名和密碼
      pSock->Receive(recv_username, 32, 0);
      pSock->Receive(recv_password, 32, 0);
 
      // 驗證登錄狀態(tài)
      bool login_flag = login(recv_username, recv_password, sid);
      if (login_flag == TRUE)
      {
        std::cout << "用戶: " << recv_username << " 已登錄" << std::endl;
        pSock->Send("已登錄", sizeof("已登錄"), 0);
      }
      else
      {
        pSock->Send("賬號或密碼錯誤", sizeof("賬號或密碼錯誤"), 0);
      }
    }
 
    // 用戶登出
    else if (strcmp(szBuffer, "logout\n") == 0)
    {
      // 驗證是否登錄成功
      int login_flag = is_login(login_pool_vect, sid);
      if (login_flag == TRUE)
      {
        std::cout << "用戶已登出" << std::endl;
        logout(login_pool_vect, sid);
        pSock->Send("用戶已登出", sizeof("用戶已登出"), 0);
      }
      else
      {
        std::cout << "請先登錄" << std::endl;
        pSock->Send("請先登錄", sizeof("請先登錄"), 0);
      }
    }
 
    // 遍歷本機(jī)文件
    else if (strcmp(szBuffer, "list\n") == 0)
    {
      // 驗證是否登錄成功
      int login_flag = is_login(login_pool_vect, sid);
      if (login_flag == TRUE)
      {
        std::cout << "用戶已登錄,輸出本機(jī)文件" << std::endl;
        pSock->Send("認(rèn)證通過", sizeof("認(rèn)證通過"), 0);
 
        // 循環(huán)輸出數(shù)據(jù)包
        for (int x = 0; x < 10; x++)
        {
          char sz[1024] = { 0 };
          sprintf(sz, "count -> %d", x);
          pSock->Send(sz, sizeof(sz), 0);
        }
      }
      else
      {
        std::cout << "請先登錄" << std::endl;
        pSock->Send("請先登錄", sizeof("請先登錄"), 0);
      }
    }
  }
}
 
int main(int argc, char *argv[])
{
  MySocket sock;
  if (!sock.Create(8233, SOCK_STREAM, "127.0.0.1"))
  {
    return -1;
  }
 
  // 獲取本機(jī)信息
  char sSevIp[20];
  UINT nSevPort;
  sock.GetSockName(sSevIp, nSevPort);
  std::cout << "服務(wù)端: " << sSevIp << ":" << nSevPort << " 服務(wù)器啟動成功" << std::endl;
 
  sock.Listen(5);
 
  // 獲取客戶端信息
  char sIp[20];
  UINT nPort;
 
  MySocket ptr;
  while (true)
  {
    sock.Accept(ptr, sIp, &nPort);
    std::cout << "客戶: " << sIp << ":" << nPort << " [已登錄]" << std::endl;
 
    // 多線程
    _beginthread(ClientPro, 0, &ptr);
  }
  return 0;
}

#define _CRT_SECURE_NO_WARNINGS
#define _WINSOCK_DEPRECATED_NO_WARNINGS
#include <iostream>
#include "MySocket.hpp"
 
using namespace std;
 
int main(int argc, char* argv[])
{
  MySocket sock;
  if (!sock.Create(0, SOCK_STREAM))
  {
    return -1;
  }
 
  // 獲取本機(jī)信息
  char sClientIp[20];
  UINT nClientPort;
  sock.GetSockName(sClientIp, nClientPort);
 
  if (!sock.Connection("127.0.0.1", 8233))
  {
    cout << "連接服務(wù)器失敗" << GetLastError() << endl;
    return -1;
  }
 
  char szBuffer[4096] = { 0 };
  int ref = sock.Receive(szBuffer, sizeof(szBuffer));
  szBuffer[ref] = 0;
  std::cout << "服務(wù)端回應(yīng): " << szBuffer << std::endl;
 
  while (true)
  {
  input:
    memset(szBuffer, 0, 4096);
    std::cout << "CMD > ";
 
    // 發(fā)送命令
    int inputLine = 0;
    while ((szBuffer[inputLine++] = getchar()) != '\n');
    if (strlen(szBuffer) == 1)
      goto input;
 
    // 執(zhí)行登錄
    if (strcmp(szBuffer, "login\n") == 0)
    {
      // 發(fā)送命令
      sock.Send(szBuffer, 4096, 0);
 
      char input_username[32] = { 0 };
      char input_password[32] = { 0 };
 
      // 發(fā)送用戶名
      printf("用戶名: ");
      scanf("%s", &input_username);
      sock.Send(input_username, 32, 0);
 
      // 發(fā)送密碼
      printf("密碼: ");
      scanf("%s", &input_password);
      sock.Send(input_password, 32, 0);
 
      // 獲取登錄狀態(tài)
      char recv_message[64] = { 0 };
      sock.Receive(recv_message, 64, 0);
      std::cout << recv_message << std::endl;
    }
 
    // 登出用戶
    else if (strcmp(szBuffer, "logout\n") == 0)
    {
      // 發(fā)送命令
      sock.Send(szBuffer, 4096, 0);
 
      // 獲取返回消息
      char recv_message[64] = { 0 };
      sock.Receive(recv_message, 64, 0);
      std::cout << recv_message << std::endl;
    }
 
    // 遍歷本機(jī)文件
    else if (strcmp(szBuffer, "list\n") == 0)
    {
      // 發(fā)送命令
      sock.Send(szBuffer, 4096, 0);
 
      // 獲取返回消息
      char recv_message[64] = { 0 };
      sock.Receive(recv_message, 64, 0);
      std::cout << recv_message << std::endl;
 
      if (strcmp(recv_message, "請先登錄") == 0)
      {
        goto input;
      }
 
      // 循環(huán)接收數(shù)據(jù)包
      for (int x = 0; x < 10; x++)
      {
        char sz[1024] = { 0 };
        sock.Receive(sz, 1024, 0);
        std::cout << sz << std::endl;
      }
    }
  }
  sock.Close();
  return 0;
}