ntpclient.c

来自「一个类似windows」· C语言 代码 · 共 88 行

#include <timedate.h>

SOCKET Sock;
SOCKADDR_IN myAddr, ntpAddr;

BOOL InitialiseConnection()
{
    WSADATA wsaData;
    INT Ret;

    Ret = WSAStartup(MAKEWORD(2, 2),
                     &wsaData);
    if (Ret != 0)
        return FALSE;


    Sock = WSASocket(AF_INET,
                     SOCK_DGRAM,
                     IPPROTO_UDP,
                     NULL,
                     0,
                     WSA_FLAG_OVERLAPPED);
    if (Sock == INVALID_SOCKET )
        return FALSE;

    /* setup client socket */
    ZeroMemory(&myAddr, sizeof(myAddr));
    myAddr.sin_family = AF_INET;
    myAddr.sin_port = htons(IPPORT_TIMESERVER);
    myAddr.sin_addr.s_addr = INADDR_ANY;

    Ret = bind(Sock,
               (SOCKADDR *)&myAddr,
               sizeof(SOCKADDR));
    if (Ret == SOCKET_ERROR)
        return FALSE;

    /* setup server socket */
    ZeroMemory(&ntpAddr, sizeof(ntpAddr));
    ntpAddr.sin_family = AF_INET;
    ntpAddr.sin_port = htons(IPPORT_TIMESERVER);
    ntpAddr.sin_addr.s_addr = INADDR_ANY;

    return TRUE;
}

VOID DestroyConnection()
{
    WSACleanup();
}

/* send some data to wake the server up */
BOOL SendData()
{
    CHAR Packet[] = "";
    INT Ret;

    Ret = sendto(Sock,
                 Packet,
                 sizeof(Packet),
                 0,
                 (SOCKADDR *)&myAddr,
                 sizeof(SOCKADDR_IN));

    if (Ret == SOCKET_ERROR)
        return FALSE;

    return TRUE;
}


BOOL RecieveData(CHAR *Buf)
{
    INT Ret;
    INT Size = sizeof(SOCKADDR_IN);

    Ret = recvfrom(Sock,
                   Buf,
                   BUFSIZE,
                   0,
                   (SOCKADDR *)&ntpAddr,
                   &Size);
    if (Ret == SOCKET_ERROR)
        return FALSE;

    return TRUE;
}