FreeBASIC и сети

#ifndef unicode
    #define unicode
#endif
#include once "windows.bi"
#include once "win\winsock2.bi"
#include once "win\ws2tcpip.bi"
 
' Соединиться с сервером и вернуть сокет
DECLARE FUNCTION ConnectToServer(BYVAL sServer AS WString Ptr, BYVAL Port AS WString Ptr, BYVAL LocalAddress AS WString Ptr, BYVAL LocalPort AS WString Ptr)AS SOCKET
 
' Создать прослушивающий сокет, привязанный к адресу
DECLARE FUNCTION CreateSocketAndListen(BYVAL LocalAddress AS WString Ptr, BYVAL LocalPort AS WString Ptr)AS SOCKET
 
' Закрывает сокет
DECLARE SUB CloseSocketConnection(BYVAL mSock AS SOCKET)
 
' Создать сокет, привязанный к адресу
DECLARE FUNCTION CreateSocketAndBind(BYVAL sServer AS WString Ptr, BYVAL Port AS WString Ptr)AS SOCKET
 
' Разрешение доменного имени
DECLARE FUNCTION ResolveHost(BYVAL sServer AS WString Ptr, BYVAL Port AS WString Ptr)AS addrinfoW Ptr

#include once "Network.bi"
 
SUB CloseSocketConnection(BYVAL mSock AS SOCKET)
    Shutdown(mSock, 2)
    closesocket(mSock)
END SUB
 
FUNCTION ResolveHost(BYVAL sServer AS WString Ptr, BYVAL Port AS WString Ptr)AS addrinfoW Ptr
    DIM hints AS addrinfoW
    ' Если стоит AF_UNSPEC, то неважно, IPv4 или IPv6
    hints.ai_family = AF_UNSPEC ' AF_INET или AF_INET6
    hints.ai_socktype = SOCK_STREAM
    hints.ai_protocol = IPPROTO_TCP
    
    DIM pResult AS addrinfoW Ptr = 0
    IF GetAddrInfoW(sServer, Port, @hints, @pResult) = 0 THEN
        RETURN pResult
    END IF
    RETURN 0
END FUNCTION
 
FUNCTION CreateSocketAndBind(BYVAL sServer AS WString Ptr, BYVAL Port AS WString Ptr)AS SOCKET
    ' Открыть сокет
    DIM iSocket AS SOCKET = socket_(AF_UNSPEC, SOCK_STREAM, IPPROTO_TCP)
    IF iSocket <> INVALID_SOCKET THEN
        ' Привязать адрес к сокету
        DIM localIpList AS addrinfoW Ptr = ResolveHost(sServer, Port)
        IF localIpList <> 0 THEN
            ' Обойти список адресов и сделать привязку
            DIM pPtr AS addrinfoW Ptr = localIpList
            DIM BindResult AS INTEGER = ANY
            DO
                BindResult = bind(iSocket, Cast(LPSOCKADDR, pPtr->ai_addr), pPtr->ai_addrlen)
                IF BindResult = 0 THEN
                    ' Привязано
                    EXIT DO
                END IF
                pPtr = pPtr->ai_next
            LOOP UNTIL pPtr = 0
            ' Очистка
            FreeAddrInfoW(localIpList)
            ' Привязались к адресу
            IF BindResult = 0 THEN
                RETURN iSocket
            END IF
        END IF
        CloseSocketConnection(iSocket)
    END IF
    RETURN INVALID_SOCKET
END FUNCTION
 
FUNCTION CreateSocketAndListen(BYVAL LocalAddress AS WString Ptr, BYVAL LocalPort AS WString Ptr)AS SOCKET
    ' Открыть сокет
    DIM iSocket AS SOCKET = CreateSocketAndBind(LocalAddress, LocalPort)
    IF iSocket <> INVALID_SOCKET THEN
        ' Начать прослушивание
        IF listen(iSocket, 1) <> SOCKET_ERROR THEN
            RETURN iSocket
        END IF
        CloseSocketConnection(iSocket)
    END IF
    RETURN INVALID_SOCKET
END FUNCTION
 
FUNCTION ConnectToServer(BYVAL sServer AS WString Ptr, BYVAL Port AS WString Ptr, BYVAL LocalAddress AS WString Ptr, BYVAL LocalPort AS WString Ptr)AS SOCKET
    ' Открыть сокет
    DIM iSocket AS SOCKET = CreateSocketAndBind(LocalAddress, LocalPort)
    IF iSocket <> INVALID_SOCKET THEN
        ' Привязать адрес к сокету
        DIM localIpList AS addrinfoW Ptr = ResolveHost(sServer, Port)
        IF localIpList <> 0 THEN
            ' Обойти список адресов и сделать привязку
            DIM pPtr AS addrinfoW Ptr = localIpList
            DIM ConnectResult AS INTEGER = ANY
            DO
                ConnectResult = connect(iSocket, Cast(LPSOCKADDR, pPtr->ai_addr), pPtr->ai_addrlen)
                IF ConnectResult = 0 THEN
                    ' Соединено
                    EXIT DO
                END IF
                pPtr = pPtr->ai_next
            LOOP UNTIL pPtr = 0
            ' Очистка
            FreeAddrInfoW(localIpList)
            ' Соединение установлено
            IF ConnectResult = 0 THEN
                RETURN iSocket
            END IF
        END IF
        CloseSocketConnection(iSocket)
    END IF
    RETURN INVALID_SOCKET
END FUNCTION

#include once "Network.bi"
 
' IP‐адрес, с которого сервер будет принимать соединения
CONST LocalAddress = "0.0.0.0"
CONST LocalPort = "8080"
CONST BufferSize AS INTEGER = 4096
 
DIM objWsaData AS WSAData = ANY
IF WSAStartup(MAKEWORD(2, 2), @objWsaData) <> NO_ERROR THEN
    ' Ошибка инициализации
    END(1)
END IF
 
' Создать прослушивающий сокет
DIM ListenSocket AS SOCKET = CreateSocketAndListen(LocalAddress, LocalPort)
IF ListenSocket = INVALID_SOCKET THEN
    ' Ошибка создания прослушивающего сокета
    WSACleanup()
    END(2)
END IF
 
' Принять соединение
DIM RemoteAddress AS SOCKADDR_IN = ANY
DIM RemoteAddressLength AS LONG = SizeOf(RemoteAddress)
DIM ClientSocket AS SOCKET = accept(ListenSocket, CPtr(SOCKADDR Ptr, @RemoteAddress), @RemoteAddressLength)
 
DO UNTIL ClientSocket = INVALID_SOCKET
    DIM buffer AS ZString * BufferSize = ANY
    ' Принять данные от клиента
    DIM intReceivedBytesCount AS INTEGER = recv(ClientSocket, @buffer, BufferSize, 0)
    IF intReceivedBytesCount > 0 OrElse intReceivedBytesCount <> SOCKET_ERROR THEN
        ' Отправить обратно их же
        DIM intSendBytesCount AS INTEGER = send(ClientSocket, @buffer, intReceivedBytesCount, 0)
    END IF
    ' Закрыть
    CloseSocketConnection(ClientSocket)
    ' Принять вновь
    ClientSocket = accept(ListenSocket, CPtr(SOCKADDR Ptr, @RemoteAddress), @RemoteAddressLength)
LOOP
 
' Закрыть
CloseSocketConnection(ListenSocket)
WSACleanup()

#include once "Network.bi"
 
' Адрес сервера
CONST ServerAddress = "127.0.0.1"
' Порт сервера
CONST ServerPort = "8080"
' Адрес сетевой карты, с которого будет идти соединение к серверу
CONST LocalBindAddress = "0.0.0.0"
CONST LocalBindPort = "0"
 
CONST BufferSize AS INTEGER = 4096
 
DIM objWsaData AS WSAData = ANY
IF WSAStartup(MAKEWORD(2, 2), @objWsaData) <> NO_ERROR THEN
    ' Ошибка инициализации
    END(1)
END IF
 
' Соединиться с сервером
DIM ClientSocket AS SOCKET = ConnectToServer(@ServerAddress, @ServerPort, @LocalBindAddress, @LocalBindPort)
IF ClientSocket = INVALID_SOCKET THEN
    PRINT "Ошибка"
ELSE
    ' Отправить строку "Привет, мир!" на сервер
    CONST HelloWorld = "Привет, мир!"
    send(ClientSocket, CPtr(ZString Ptr, @HelloWorld), (LEN(HelloWorld) + 1) * SizeOf(WString), 0)
    
    ' Принять данные с сервера
    DIM buffer AS WString * BufferSize = ANY
    DIM intBytesCount AS INTEGER = recv(ClientSocket, CPtr(ZString Ptr, @buffer), BufferSize, 0)
    ' Напечатать
    PRINT buffer
END IF
 
' Закрыть
CloseSocketConnection(ClientSocket)
WSACleanup()