tftpd.h File Reference

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	home
struct	tftpConnType
struct	acknowledgement
struct	message
struct	tftperror
struct	packet
struct	data12
struct	request
struct	data1
struct	data2
struct	data15

Macros
#define	MYBYTE   unsigned char
#define	MYWORD   unsigned short
#define	MYDWORD   unsigned int
#define	my_inet_addr   inet_addr
#define	MAX_SERVERS   8

Functions
bool	detectChange ()
void	closeConn ()
void	getInterfaces (data1 *)
void	runProg ()
void	processRequest (LPVOID lpParam)
char *	myGetToken (char *, MYBYTE)
char *	myTrim (char *, char *)
void	init (void *)
bool	cleanReq (request *)
bool	addServer (MYDWORD *, MYDWORD)
FILE *	openSection (const char *, MYBYTE, char *)
char *	readSection (char *, FILE *)
bool	getSection (const char *, char *, MYBYTE, char *)
bool	isIP (char *s)
char *	myLower (char *)
char *	myUpper (char *)
char *	IP2String (char *, MYDWORD)
MYDWORD *	findServer (MYDWORD *, MYDWORD)
void	printWindowsError ()
void	logMess (request *, MYBYTE)
void	logMess (char *, MYBYTE)

Macro Definition Documentation

MAX_SERVERS

#define MAX_SERVERS   8

Definition at line 34 of file tftpd.h.

my_inet_addr

#define my_inet_addr   inet_addr

Definition at line 33 of file tftpd.h.

MYBYTE

#define MYBYTE   unsigned char

Definition at line 21 of file tftpd.h.

MYDWORD

#define MYDWORD   unsigned int

Definition at line 23 of file tftpd.h.

MYWORD

#define MYWORD   unsigned short

Definition at line 22 of file tftpd.h.

Function Documentation

addServer()

bool addServer	(	MYDWORD *	,
		MYDWORD
	)

Definition at line 2557 of file tftpd.cpp.

{
    for (MYBYTE i = 0; i < MAX_SERVERS; i++)
    {
        if (!ip || array[i] == ip)
            return 0;
        else if (!array[i])
        {
            array[i] = ip;
            return 1;
        }
    }
    return 0;
}

Referenced by getInterfaces().

cleanReq()

bool cleanReq

(

request *

)

Definition at line 1505 of file tftpd.cpp.

{
    //printf("cleaning\n");

    if (req->file)
        fclose(req->file);

    if (!(req->sock == INVALID_SOCKET))
    {
        //printf("Here\n");
        closesocket(req->sock);
    }

    if (req->pkt[0])
        free(req->pkt[0]);

    if (req->pkt[1])
        free(req->pkt[1]);

    WaitForSingleObject(cEvent, INFINITE);
    activeThreads--;
    SetEvent(cEvent);

    //printf("cleaned\n");

    return (totalThreads <= minThreads);
}

Referenced by processRequest().

closeConn()

void closeConn

(

)

Definition at line 481 of file tftpd.cpp.

{
    for (int i = 0; i < MAX_SERVERS && network.tftpConn[i].loaded; i++)
        if (network.tftpConn[i].ready)
            closesocket(network.tftpConn[i].sock);
}

Referenced by init(), runProg(), and ServiceMain().

detectChange()

bool detectChange

(

)

Definition at line 2428 of file tftpd.cpp.

{
    if (!cfig.failureCount)
    {
        if (cfig.ifspecified)
            return false;
    }

    MYDWORD eventWait = UINT_MAX;

    if (cfig.failureCount)
#ifdef __REACTOS__
        eventWait = 10000 * (1 << cfig.failureCount);
#else
        eventWait = 10000 * pow(2, cfig.failureCount);
#endif

    OVERLAPPED overlap;
    MYDWORD ret;
    HANDLE hand = NULL;
    overlap.hEvent = WSACreateEvent();

    ret = NotifyAddrChange(&hand, &overlap);

    if (ret != NO_ERROR)
    {
        if (WSAGetLastError() != WSA_IO_PENDING)
        {
            printf("NotifyAddrChange error...%d\n", WSAGetLastError());
            return true;
        }
    }

    if ( WaitForSingleObject(overlap.hEvent, eventWait) == WAIT_OBJECT_0 )
        WSACloseEvent(overlap.hEvent);

    network.ready = false;

    while (network.busy)
        Sleep(1000);

    if (cfig.failureCount)
    {
        sprintf(logBuff, "Retrying failed Listening Interfaces..");
        logMess(logBuff, 1);
    }
    else
    {
        sprintf(logBuff, "Network changed, re-detecting Interfaces..");
        logMess(logBuff, 1);
    }

    return true;
}

Referenced by init().

findServer()

MYDWORD* findServer	(	MYDWORD *	,
		MYDWORD
	)

Definition at line 2572 of file tftpd.cpp.

{
    if (ip)
    {
        for (MYBYTE i = 0; i < MAX_SERVERS && array[i]; i++)
        {
            if (array[i] == ip)
                return &(array[i]);
        }
    }
    return 0;
}

Referenced by init().

getInterfaces()

void getInterfaces

(

data1 *

)

Definition at line 2519 of file tftpd.cpp.

{
    memset(network, 0, sizeof(data1));

    SOCKET sd = WSASocket(PF_INET, SOCK_DGRAM, 0, 0, 0, 0);

    if (sd == INVALID_SOCKET)
        return;

    INTERFACE_INFO InterfaceList[MAX_SERVERS];
    unsigned long nBytesReturned;

    if (WSAIoctl(sd, SIO_GET_INTERFACE_LIST, 0, 0, &InterfaceList,
                 sizeof(InterfaceList), &nBytesReturned, 0, 0) == SOCKET_ERROR)
        return ;

    int nNumInterfaces = nBytesReturned / sizeof(INTERFACE_INFO);

    for (int i = 0; i < nNumInterfaces; ++i)
    {
        sockaddr_in *pAddress = (sockaddr_in*)&(InterfaceList[i].iiAddress);
        u_long nFlags = InterfaceList[i].iiFlags;

        if (pAddress->sin_addr.s_addr)
        {
            addServer(network->allServers, pAddress->sin_addr.s_addr);

            if (!(nFlags & IFF_POINTTOPOINT) && (nFlags & IFF_UP))
            {
                addServer(network->staticServers, pAddress->sin_addr.s_addr);
            }
        }
    }

    closesocket(sd);
}

Referenced by init().

getSection()

bool getSection	(	const char *	,
		char *	,
		MYBYTE	,
		char *
	)

Definition at line 1533 of file tftpd.cpp.

{
    //printf("%s=%s\n",fileName,sectionName);
    char section[128];
    sprintf(section, "[%s]", sectionName);
    myUpper(section);
    FILE *f = fopen(fileName, "rt");
    char buff[512];
    MYBYTE found = 0;

    if (f)
    {
        while (fgets(buff, 511, f))
        {
            myUpper(buff);
            myTrim(buff, buff);

            if (strstr(buff, section) == buff)
            {
                found++;
                if (found == serial)
                {
                    //printf("%s=%s\n",fileName,sectionName);
                    while (fgets(buff, 511, f))
                    {
                        myTrim(buff, buff);

                        if (strstr(buff, "[") == buff)
                            break;

                        if (((*buff) >= '0' && (*buff) <= '9') || ((*buff) >= 'A' && (*buff) <= 'Z') || ((*buff) >= 'a' && (*buff) <= 'z') || ((*buff) && strchr("/\\?*", (*buff))))
                        {
                            buffer += sprintf(buffer, "%s", buff);
                            buffer++;
                        }
                    }
                    break;
                }
            }
        }
        fclose(f);
    }

    *buffer = 0;
    *(buffer + 1) = 0;
    return (found == serial);
}

init()

void init

(

void *

)

Definition at line 1805 of file tftpd.cpp.

{
    memset(&cfig, 0, sizeof(cfig));

    GetModuleFileName(NULL, extbuff, _MAX_PATH);
    char *fileExt = strrchr(extbuff, '.');
    *fileExt = 0;
    sprintf(iniFile, "%s.ini", extbuff);
    sprintf(lnkFile, "%s.url", extbuff);
    fileExt = strrchr(extbuff, '\\');
    *fileExt = 0;
    fileExt++;
    sprintf(logFile, "%s\\log\\%s%%Y%%m%%d.log", extbuff, fileExt);

    FILE *f = NULL;
    char raw[512];
    char name[512];
    char value[512];

    if (verbatim)
    {
        cfig.logLevel = 2;
        printf("%s\n\n", sVersion);
    }
    else if ((f = openSection("LOGGING", 1, iniFile)))
    {
        cfig.logLevel = 1;
        tempbuff[0] = 0;

        while (readSection(raw, f))
        {
            if (!strcasecmp(raw, "None"))
                cfig.logLevel = 0;
            else if (!strcasecmp(raw, "Errors"))
                cfig.logLevel = 1;
            else if (!strcasecmp(raw, "All"))
                cfig.logLevel = 2;
            else
                sprintf(tempbuff, "Section [LOGGING], Invalid LogLevel: %s", raw);
        }
    }

    if (!verbatim && cfig.logLevel && logFile[0])
    {
        time_t t = time(NULL);
        tm *ttm = localtime(&t);
        loggingDay = ttm->tm_yday;
        strftime(extbuff, sizeof(extbuff), logFile, ttm);

        cfig.logfile = fopen(extbuff, "at");

        if (cfig.logfile)
        {
            WritePrivateProfileString("InternetShortcut","URL", extbuff, lnkFile);
            WritePrivateProfileString("InternetShortcut","IconIndex", "0", lnkFile);
            WritePrivateProfileString("InternetShortcut","IconFile", extbuff, lnkFile);
            sprintf(logBuff, "%s Starting..", sVersion);
            logMess(logBuff, 1);

            if (tempbuff[0])
                logMess(tempbuff, 0);
        }
    }

    MYWORD wVersionRequested = MAKEWORD(1, 1);
    WSAStartup(wVersionRequested, &cfig.wsaData);

    if (cfig.wsaData.wVersion != wVersionRequested)
    {
        sprintf(logBuff, "WSAStartup Error");
        logMess(logBuff, 1);
    }

    if ((f = openSection("HOME", 1, iniFile)))
    {
        while (readSection(raw, f))
        {
            mySplit(name, value, raw, '=');

            if (strlen(value))
            {
                if (!cfig.homes[0].alias[0] && cfig.homes[0].target[0])
                {
                    sprintf(logBuff, "Section [HOME], alias and bare path mixup, entry %s ignored", raw);
                    logMess(logBuff, 1);
                }
                else if (strchr(name, notFileSep) || strchr(name, fileSep) || strchr(name, '>') || strchr(name, '<') || strchr(name, '.'))
                {
                    sprintf(logBuff, "Section [HOME], invalid chars in alias %s, entry ignored", name);
                    logMess(logBuff, 1);
                }
                else if (name[0] && strlen(name) < 64 && value[0])
                {
#ifdef __REACTOS__
                    for (int i = 0; i < MAX_SERVERS; i++)
#else
                    for (int i = 0; i < 8; i++)
#endif
                    {
                        if (cfig.homes[i].alias[0] && !strcasecmp(name, cfig.homes[i].alias))
                        {
                            sprintf(logBuff, "Section [HOME], Duplicate Entry: %s ignored", raw);
                            logMess(logBuff, 1);
                            break;
                        }
                        else if (!cfig.homes[i].alias[0])
                        {
                            strcpy(cfig.homes[i].alias, name);
                            strcpy(cfig.homes[i].target, value);

                            if (cfig.homes[i].target[strlen(cfig.homes[i].target) - 1] != fileSep)
                            {
                                tempbuff[0] = fileSep;
                                tempbuff[1] = 0;
                                strcat(cfig.homes[i].target, tempbuff);
                            }

                            break;
                        }
                    }
                }
                else
                {
                    sprintf(logBuff, "Section [HOME], alias %s too large", name);
                    logMess(logBuff, 1);
                }
            }
            else if (!cfig.homes[0].alias[0] && !cfig.homes[0].target[0])
            {
                strcpy(cfig.homes[0].target, name);

                if (cfig.homes[0].target[strlen(cfig.homes[0].target) - 1] != fileSep)
                {
                    tempbuff[0] = fileSep;
                    tempbuff[1] = 0;
                    strcat(cfig.homes[0].target, tempbuff);
                }
            }
            else if (cfig.homes[0].alias[0])
            {
                sprintf(logBuff, "Section [HOME], alias and bare path mixup, entry %s ignored", raw);
                logMess(logBuff, 1);
            }
            else if (cfig.homes[0].target[0])
            {
                sprintf(logBuff, "Section [HOME], Duplicate Path: %s ignored", raw);
                logMess(logBuff, 1);
            }
            else
            {
                sprintf(logBuff, "Section [HOME], missing = sign, Invalid Entry: %s ignored", raw);
                logMess(logBuff, 1);
            }
        }
    }

    if (!cfig.homes[0].target[0])
    {
        GetModuleFileName(NULL, cfig.homes[0].target, UCHAR_MAX);
        char *iniFileExt = strrchr(cfig.homes[0].target, fileSep);
        *(++iniFileExt) = 0;
    }

    cfig.fileRead = true;

    if ((f = openSection("TFTP-OPTIONS", 1, iniFile)))
    {
        while (readSection(raw, f))
        {
            mySplit(name, value, raw, '=');

            if (strlen(value))
            {
                if (!strcasecmp(name, "blksize"))
                {
                    MYDWORD tblksize = atol(value);

                    if (tblksize < 512)
                        blksize = 512;
                    else if (tblksize > USHRT_MAX - 32)
                        blksize = USHRT_MAX - 32;
                    else
                        blksize = tblksize;
                }
                else if (!strcasecmp(name, "threadpoolsize"))
                {
                    minThreads = atol(value);
                    if (minThreads < 1)
                        minThreads = 0;
                    else if (minThreads > 100)
                        minThreads = 100;
                }
                else if (!strcasecmp(name, "timeout"))
                {
                    timeout = atol(value);
                    if (timeout < 1)
                        timeout = 1;
                    else if (timeout > UCHAR_MAX)
                        timeout = UCHAR_MAX;
                }
                else if (!strcasecmp(name, "Read"))
                {
                    if (strchr("Yy", *value))
                        cfig.fileRead = true;
                    else
                        cfig.fileRead = false;
                }
                else if (!strcasecmp(name, "Write"))
                {
                    if (strchr("Yy", *value))
                        cfig.fileWrite = true;
                    else
                        cfig.fileWrite = false;
                }
                else if (!strcasecmp(name, "Overwrite"))
                {
                    if (strchr("Yy", *value))
                        cfig.fileOverwrite = true;
                    else
                        cfig.fileOverwrite = false;
                }
                else if (!strcasecmp(name, "port-range"))
                {
                    char *ptr = strchr(value, '-');
                    if (ptr)
                    {
                        *ptr = 0;
                        cfig.minport = atol(value);
                        cfig.maxport = atol(++ptr);

                        if (cfig.minport < 1024 || cfig.minport >= USHRT_MAX || cfig.maxport < 1024 || cfig.maxport >= USHRT_MAX || cfig.minport > cfig.maxport)
                        {
                            cfig.minport = 0;
                            cfig.maxport = 0;

                            sprintf(logBuff, "Invalid port range %s", value);
                            logMess(logBuff, 1);
                        }
                    }
                    else
                    {
                        sprintf(logBuff, "Invalid port range %s", value);
                        logMess(logBuff, 1);
                    }
                }
                else
                {
                    sprintf(logBuff, "Warning: unknown option %s, ignored", name);
                    logMess(logBuff, 1);
                }
            }
        }
    }

    if ((f = openSection("ALLOWED-CLIENTS", 1, iniFile)))
    {
#ifdef __REACTOS__
        MYWORD i = 0;
#else
        int i = 0;
#endif

        while (readSection(raw, f))
        {
#ifdef __REACTOS__
            if (i < _countof(cfig.hostRanges))
#else
            if (i < 32)
#endif
            {
                MYDWORD rs = 0;
                MYDWORD re = 0;
                mySplit(name, value, raw, '-');
                rs = htonl(my_inet_addr(name));

                if (strlen(value))
                    re = htonl(my_inet_addr(value));
                else
                    re = rs;

                if (rs && rs != INADDR_NONE && re && re != INADDR_NONE && rs <= re)
                {
                    cfig.hostRanges[i].rangeStart = rs;
                    cfig.hostRanges[i].rangeEnd = re;
                    i++;
                }
                else
                {
                    sprintf(logBuff, "Section [ALLOWED-CLIENTS] Invalid entry %s in ini file, ignored", raw);
                    logMess(logBuff, 1);
                }
            }
        }
    }

    if (verbatim)
    {
        printf("starting TFTP...\n");
    }
    else
    {
        sprintf(logBuff, "starting TFTP service");
        logMess(logBuff, 1);
    }

    for (int i = 0; i < MAX_SERVERS; i++)
        if (cfig.homes[i].target[0])
        {
            sprintf(logBuff, "alias /%s is mapped to %s", cfig.homes[i].alias, cfig.homes[i].target);
            logMess(logBuff, 1);
        }

    if (cfig.hostRanges[0].rangeStart)
    {
        char temp[128];

#ifdef __REACTOS__
        for (MYWORD i = 0; i < _countof(cfig.hostRanges) && cfig.hostRanges[i].rangeStart; i++)
#else
        for (MYWORD i = 0; i <= sizeof(cfig.hostRanges) && cfig.hostRanges[i].rangeStart; i++)
#endif
        {
            sprintf(logBuff, "%s", "permitted clients: ");
            sprintf(temp, "%s-", IP2String(tempbuff, htonl(cfig.hostRanges[i].rangeStart)));
            strcat(logBuff, temp);
            sprintf(temp, "%s", IP2String(tempbuff, htonl(cfig.hostRanges[i].rangeEnd)));
            strcat(logBuff, temp);
            logMess(logBuff, 1);
        }
    }
    else
    {
        sprintf(logBuff, "%s", "permitted clients: all");
        logMess(logBuff, 1);
    }

    if (cfig.minport)
    {
        sprintf(logBuff, "server port range: %u-%u", cfig.minport, cfig.maxport);
        logMess(logBuff, 1);
    }
    else
    {
        sprintf(logBuff, "server port range: all");
        logMess(logBuff, 1);
    }

    sprintf(logBuff, "max blksize: %u", blksize);
    logMess(logBuff, 1);
    sprintf(logBuff, "default blksize: %u", 512);
    logMess(logBuff, 1);
    sprintf(logBuff, "default timeout: %u", timeout);
    logMess(logBuff, 1);
    sprintf(logBuff, "file read allowed: %s", cfig.fileRead ? "Yes" : "No");
    logMess(logBuff, 1);
    sprintf(logBuff, "file create allowed: %s", cfig.fileWrite ? "Yes" : "No");
    logMess(logBuff, 1);
    sprintf(logBuff, "file overwrite allowed: %s", cfig.fileOverwrite ? "Yes" : "No");
    logMess(logBuff, 1);

    if (!verbatim)
    {
        sprintf(logBuff, "logging: %s", cfig.logLevel > 1 ? "all" : "errors");
        logMess(logBuff, 1);
    }

    lEvent = CreateEvent(
        NULL,                  // default security descriptor
        FALSE,                 // ManualReset
        TRUE,                  // Signalled
        TEXT("AchalTFTServerLogEvent"));  // object name

    if (lEvent == NULL)
    {
        printf("CreateEvent error: %lu\n", GetLastError());
        exit(-1);
    }
    else if ( GetLastError() == ERROR_ALREADY_EXISTS )
    {
        sprintf(logBuff, "CreateEvent opened an existing Event\nServer May already be Running");
        logMess(logBuff, 0);
        exit(-1);
    }

    tEvent = CreateEvent(
        NULL,                  // default security descriptor
        FALSE,                 // ManualReset
        FALSE,                 // Signalled
        TEXT("AchalTFTServerThreadEvent"));  // object name

    if (tEvent == NULL)
    {
        printf("CreateEvent error: %lu\n", GetLastError());
        exit(-1);
    }
    else if ( GetLastError() == ERROR_ALREADY_EXISTS )
    {
        sprintf(logBuff, "CreateEvent opened an existing Event\nServer May already be Running");
        logMess(logBuff, 0);
        exit(-1);
    }

    sEvent = CreateEvent(
        NULL,                  // default security descriptor
        FALSE,                 // ManualReset
        TRUE,                  // Signalled
        TEXT("AchalTFTServerSocketEvent"));  // object name

    if (sEvent == NULL)
    {
        printf("CreateEvent error: %lu\n", GetLastError());
        exit(-1);
    }
    else if ( GetLastError() == ERROR_ALREADY_EXISTS )
    {
        sprintf(logBuff, "CreateEvent opened an existing Event\nServer May already be Running");
        logMess(logBuff, 0);
        exit(-1);
    }

    cEvent = CreateEvent(
        NULL,                  // default security descriptor
        FALSE,                 // ManualReset
        TRUE,                  // Signalled
        TEXT("AchalTFTServerCountEvent"));  // object name

    if (cEvent == NULL)
    {
        printf("CreateEvent error: %lu\n", GetLastError());
        exit(-1);
    }
    else if ( GetLastError() == ERROR_ALREADY_EXISTS )
    {
        sprintf(logBuff, "CreateEvent opened an existing Event\nServer May already be Running");
        logMess(logBuff, 0);
        exit(-1);
    }

    if (minThreads)
    {
        for (int i = 0; i < minThreads; i++)
        {
            _beginthread(
                  processRequest,                 // thread function
                  0,                            // default security attributes
                  NULL);                          // argument to thread function
        }

        sprintf(logBuff, "thread pool size: %u", minThreads);
        logMess(logBuff, 1);
    }

    for (int i = 0; i < MAX_SERVERS && network.tftpConn[i].port; i++)
    {
        sprintf(logBuff, "listening on: %s:%i", IP2String(tempbuff, network.tftpConn[i].server), network.tftpConn[i].port);
        logMess(logBuff, 1);
    }

    do
    {
        memset(&newNetwork, 0, sizeof(data1));

        bool bindfailed = false;

        if ((f = openSection("LISTEN-ON", 1, iniFile)))
        {
            MYBYTE i = 0;

            while (readSection(raw, f))
            {
                MYWORD port = 69;

                cfig.ifspecified = true;
                mySplit(name, value, raw, ':');

                if (value[0])
                    port = atoi(value);

                if(i < MAX_SERVERS)
                {
                    if (isIP(name))
                    {
                        MYDWORD addr = my_inet_addr(name);

                        if (!addr)
                        {
                            newNetwork.listenServers[0] = 0;
                            newNetwork.listenPorts[0] = port;
                            fclose(f);
                            break;
                        }
                        else if (!findServer(newNetwork.listenServers, addr))
                        {
                            newNetwork.listenServers[i] = addr;
                            newNetwork.listenPorts[i] = port;
                            i++;
                        }
                    }
                    else
                    {
                        sprintf(logBuff, "Warning: Section [LISTEN-ON], Invalid Interface Address %s, ignored", raw);
                        logMess(logBuff, 1);
                    }
                }
            }
        }

        if (!cfig.ifspecified)
        {
            sprintf(logBuff, "detecting Interfaces..");
            logMess(logBuff, 1);
            getInterfaces(&newNetwork);

            for (MYBYTE n = 0; n < MAX_SERVERS && newNetwork.staticServers[n]; n++)
            {
                newNetwork.listenServers[n] = newNetwork.staticServers[n];
                newNetwork.listenPorts[n] = 69;
            }
        }

        MYBYTE i = 0;

        for (int j = 0; j < MAX_SERVERS && newNetwork.listenPorts[j]; j++)
        {
            int k = 0;

            for (; k < MAX_SERVERS && network.tftpConn[k].loaded; k++)
            {
                if (network.tftpConn[k].ready && network.tftpConn[k].server == newNetwork.listenServers[j] && network.tftpConn[k].port == newNetwork.listenPorts[j])
                    break;
            }

            if (network.tftpConn[k].ready && network.tftpConn[k].server == newNetwork.listenServers[j] && network.tftpConn[k].port == newNetwork.listenPorts[j])
            {
                memcpy(&(newNetwork.tftpConn[i]), &(network.tftpConn[k]), sizeof(tftpConnType));

                if (newNetwork.maxFD < newNetwork.tftpConn[i].sock)
                    newNetwork.maxFD = newNetwork.tftpConn[i].sock;

                network.tftpConn[k].ready = false;
                //printf("%d, %s found\n", i, IP2String(tempbuff, newNetwork.tftpConn[i].server));
                i++;
                continue;
            }
            else
            {
                newNetwork.tftpConn[i].sock = socket(PF_INET, SOCK_DGRAM, IPPROTO_UDP);

                if (newNetwork.tftpConn[i].sock == INVALID_SOCKET)
                {
                    bindfailed = true;
                    sprintf(logBuff, "Failed to Create Socket");
                    logMess(logBuff, 1);
                    continue;
                }

                //printf("Socket %u\n", newNetwork.tftpConn[i].sock);

                errno = 0;
                newNetwork.tftpConn[i].addr.sin_family = AF_INET;
                newNetwork.tftpConn[i].addr.sin_addr.s_addr = newNetwork.listenServers[j];
                newNetwork.tftpConn[i].addr.sin_port = htons(newNetwork.listenPorts[j]);
                int nRet = bind(newNetwork.tftpConn[i].sock, (sockaddr*)&newNetwork.tftpConn[i].addr, sizeof(struct sockaddr_in));

                if (nRet == SOCKET_ERROR || errno)
                {
                    bindfailed = true;
                    closesocket(newNetwork.tftpConn[i].sock);
                    sprintf(logBuff, "%s Port %i bind failed", IP2String(tempbuff, newNetwork.listenServers[j]), newNetwork.listenPorts[j]);
                    logMess(logBuff, 1);
                    continue;
                }

                newNetwork.tftpConn[i].loaded = true;
                newNetwork.tftpConn[i].ready = true;
                newNetwork.tftpConn[i].server = newNetwork.listenServers[j];
                newNetwork.tftpConn[i].port = newNetwork.listenPorts[j];

                //printf("%d, %s created\n", i, IP2String(tempbuff, newNetwork.tftpConn[i].server));

                if (newNetwork.maxFD < newNetwork.tftpConn[i].sock)
                    newNetwork.maxFD = newNetwork.tftpConn[i].sock;

                if (!newNetwork.listenServers[j])
                    break;

                i++;
            }
        }

        if (bindfailed)
            cfig.failureCount++;
        else
            cfig.failureCount = 0;

        closeConn();
        memcpy(&network, &newNetwork, sizeof(data1));

        //printf("%i %i %i\n", network.tftpConn[0].ready, network.dnsUdpConn[0].ready, network.dnsTcpConn[0].ready);

        if (!network.tftpConn[0].ready)
        {
            sprintf(logBuff, "No Static Interface ready, Waiting...");
            logMess(logBuff, 1);
            continue;
        }

        for (int i = 0; i < MAX_SERVERS && network.tftpConn[i].loaded; i++)
        {
            sprintf(logBuff, "Listening On: %s:%d", IP2String(tempbuff, network.tftpConn[i].server), network.tftpConn[i].port);
            logMess(logBuff, 1);
        }

        network.ready = true;

    } while (detectChange());

    //printf("Exiting Init\n");

    _endthread();
    return;
}

Referenced by runProg(), and ServiceMain().

IP2String()

char* IP2String	(	char *	,
		MYDWORD
	)

Definition at line 1759 of file tftpd.cpp.

{
    data15 inaddr;
    inaddr.ip = ip;
    sprintf(target, "%u.%u.%u.%u", inaddr.octate[0], inaddr.octate[1], inaddr.octate[2], inaddr.octate[3]);
    return target;
}

Referenced by init(), and logMess().

isIP()

bool isIP

(

char *

s

)

Definition at line 1767 of file tftpd.cpp.

{
    int j = 0;

    for (; *string; string++)
    {
        if (*string == '.' && *(string + 1) != '.')
            j++;
        else if (*string < '0' || *string > '9')
            return 0;
    }

    if (j == 3)
        return 1;
    else
        return 0;
}

Referenced by init().

logMess() [1/2]

void logMess	(	request *	,
		MYBYTE
	)

Definition at line 2622 of file tftpd.cpp.

{
    WaitForSingleObject(lEvent, INFINITE);

    char tempbuff[256];

    if (verbatim)
    {
        if (!req->serverError.errormessage[0])
            sprintf(req->serverError.errormessage, strerror(errno));

        if (req->path[0])
            printf("Client %s:%u %s, %s\n", IP2String(tempbuff, req->client.sin_addr.s_addr), ntohs(req->client.sin_port), req->path, req->serverError.errormessage);
        else
            printf("Client %s:%u, %s\n", IP2String(tempbuff, req->client.sin_addr.s_addr), ntohs(req->client.sin_port), req->serverError.errormessage);
    }
    else if (cfig.logfile && logLevel <= cfig.logLevel)
    {
        time_t t = time(NULL);
        tm *ttm = localtime(&t);

        if (ttm->tm_yday != loggingDay)
        {
            loggingDay = ttm->tm_yday;
            strftime(extbuff, sizeof(extbuff), logFile, ttm);
            fprintf(cfig.logfile, "Logging Continued on file %s\n", extbuff);
            fclose(cfig.logfile);
            cfig.logfile = fopen(extbuff, "at");

            if (cfig.logfile)
            {
                fprintf(cfig.logfile, "%s\n\n", sVersion);
                WritePrivateProfileString("InternetShortcut","URL", extbuff, lnkFile);
                WritePrivateProfileString("InternetShortcut","IconIndex", "0", lnkFile);
                WritePrivateProfileString("InternetShortcut","IconFile", extbuff, lnkFile);
            }
            else
                return;
        }

        strftime(extbuff, sizeof(extbuff), "%d-%b-%y %X", ttm);

        if (req->path[0])
            fprintf(cfig.logfile, "[%s] Client %s:%u %s, %s\n", extbuff, IP2String(tempbuff, req->client.sin_addr.s_addr), ntohs(req->client.sin_port), req->path, req->serverError.errormessage);
        else
            fprintf(cfig.logfile, "[%s] Client %s:%u, %s\n", extbuff, IP2String(tempbuff, req->client.sin_addr.s_addr), ntohs(req->client.sin_port), req->serverError.errormessage);

        fflush(cfig.logfile);
    }
    SetEvent(lEvent);
}

logMess() [2/2]

void logMess	(	char *	,
		MYBYTE
	)

Definition at line 2585 of file tftpd.cpp.

{
    WaitForSingleObject(lEvent, INFINITE);

    if (verbatim)
        printf("%s\n", logBuff);
    else if (cfig.logfile && logLevel <= cfig.logLevel)
    {
        time_t t = time(NULL);
        tm *ttm = localtime(&t);

        if (ttm->tm_yday != loggingDay)
        {
            loggingDay = ttm->tm_yday;
            strftime(extbuff, sizeof(extbuff), logFile, ttm);
            fprintf(cfig.logfile, "Logging Continued on file %s\n", extbuff);
            fclose(cfig.logfile);
            cfig.logfile = fopen(extbuff, "at");

            if (cfig.logfile)
            {
                fprintf(cfig.logfile, "%s\n\n", sVersion);
                WritePrivateProfileString("InternetShortcut","URL", extbuff, lnkFile);
                WritePrivateProfileString("InternetShortcut","IconIndex", "0", lnkFile);
                WritePrivateProfileString("InternetShortcut","IconFile", extbuff, lnkFile);
            }
            else
                return;
        }

        strftime(extbuff, sizeof(extbuff), "%d-%b-%y %X", ttm);
        fprintf(cfig.logfile, "[%s] %s\n", extbuff, logBuff);
        fflush(cfig.logfile);
    }
    SetEvent(lEvent);
}

Referenced by detectChange(), init(), processRequest(), runProg(), and ServiceMain().

myGetToken()

char* myGetToken	(	char *	,
		MYBYTE
	)

Definition at line 1628 of file tftpd.cpp.

{
    while (*buff)
    {
        if (index)
            index--;
        else
            break;

        buff += strlen(buff) + 1;
    }

    return buff;
}

myLower()

char* myLower

(

char *

)

Definition at line 1795 of file tftpd.cpp.

{
    char diff = 'a' - 'A';
    MYWORD len = strlen(string);
    for (int i = 0; i < len; i++)
        if (string[i] >= 'A' && string[i] <= 'Z')
            string[i] += diff;
    return string;
}

myTrim()

char* myTrim	(	char *	,
		char *
	)

Definition at line 1689 of file tftpd.cpp.

{
    while ((*source) && (*source) <= 32)
        source++;

    int i = 0;

    for (; i < 511 && source[i]; i++)
        target[i] = source[i];

    target[i] = source[i];
    i--;

    for (; i >= 0 && target[i] <= 32; i--)
        target[i] = 0;

    return target;
}

Referenced by getSection(), mySplit(), openSection(), and readSection().

myUpper()

char* myUpper

(

char *

)

Definition at line 1785 of file tftpd.cpp.

{
    char diff = 'a' - 'A';
    MYWORD len = strlen(string);
    for (int i = 0; i < len; i++)
        if (string[i] >= 'a' && string[i] <= 'z')
            string[i] -= diff;
    return string;
}

Referenced by getSection(), and openSection().

openSection()

FILE* openSection	(	const char *	,
		MYBYTE	,
		char *
	)

Definition at line 1581 of file tftpd.cpp.

{
    //printf("%s=%s\n",fileName,sectionName);
    char section[128];
    sprintf(section, "[%s]", sectionName);
    myUpper(section);
    FILE *f = fopen(fileName, "rt");
    char buff[512];
    MYBYTE found = 0;

    if (f)
    {
        while (fgets(buff, 511, f))
        {
            myUpper(buff);
            myTrim(buff, buff);

            if (strstr(buff, section) == buff)
            {
                found++;

                if (found == serial)
                    return f;
            }
        }
        fclose(f);
    }
    return NULL;
}

Referenced by init().

printWindowsError()

void printWindowsError

(

)

Definition at line 323 of file tftpd.cpp.

{
    MYDWORD dw = GetLastError();

    if (dw)
    {
        LPVOID lpMsgBuf;

        FormatMessage(
            FORMAT_MESSAGE_ALLOCATE_BUFFER |
            FORMAT_MESSAGE_FROM_SYSTEM |
            FORMAT_MESSAGE_IGNORE_INSERTS,
            NULL,
            dw,
            MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
            (LPTSTR) &lpMsgBuf,
            0, NULL );

        printf("Error: %s\nPress Enter..\n", (LPTSTR)lpMsgBuf);
        getchar();
    }
}

Referenced by installService(), main(), and uninstallService().

processRequest()

void processRequest

(

LPVOID

lpParam

)

readSection()

char* readSection	(	char *	,
		FILE *
	)

Definition at line 1611 of file tftpd.cpp.

{
    while (fgets(buff, 511, f))
    {
        myTrim(buff, buff);

        if (*buff == '[')
            break;

        if (((*buff) >= '0' && (*buff) <= '9') || ((*buff) >= 'A' && (*buff) <= 'Z') || ((*buff) >= 'a' && (*buff) <= 'z') || ((*buff) && strchr("/\\?*", (*buff))))
            return buff;
    }

    fclose(f);
    return NULL;
}

Referenced by init().

runProg()

void runProg

(

)

Definition at line 393 of file tftpd.cpp.

{
    verbatim = true;

    if (_beginthread(init, 0, 0) == 0)
    {
        if (cfig.logLevel)
        {
            sprintf(logBuff, "Thread Creation Failed");
            logMess(logBuff, 1);
        }
        exit(-1);
    }

    fd_set readfds;
    timeval tv;
    int fdsReady = 0;
    tv.tv_sec = 20;
    tv.tv_usec = 0;

    printf("\naccepting requests..\n");

    do
    {
        network.busy = false;

        //printf("Active=%u Total=%u\n",activeThreads, totalThreads);

        if (!network.tftpConn[0].ready || !network.ready)
        {
            Sleep(1000);
            continue;
        }

        FD_ZERO(&readfds);

        for (int i = 0; i < MAX_SERVERS && network.tftpConn[i].ready; i++)
            FD_SET(network.tftpConn[i].sock, &readfds);

        fdsReady = select(network.maxFD, &readfds, NULL, NULL, &tv);

        if (!network.ready)
            continue;

        //errno = WSAGetLastError();

        //if (errno)
        //    printf("%d\n", errno);

        for (int i = 0; fdsReady > 0 && i < MAX_SERVERS && network.tftpConn[i].ready; i++)
        {
            if (network.ready)
            {
                network.busy = true;

                if (FD_ISSET(network.tftpConn[i].sock, &readfds))
                {
                    //printf("%d Requests Waiting\n", fdsReady);

                    WaitForSingleObject(sEvent, INFINITE);

                    currentServer = i;

                    if (!totalThreads || activeThreads >= totalThreads)
                    {
                        _beginthread(
                              processRequest,                 // thread function
                              0,                            // default security attributes
                              NULL);                          // argument to thread function
                    }
                    SetEvent(tEvent);

                    //printf("thread signalled=%u\n",SetEvent(tEvent));

                    WaitForSingleObject(sEvent, INFINITE);
                    fdsReady--;
                    SetEvent(sEvent);
                }
            }
        }
    }
    while (true);

    closeConn();

    WSACleanup();
}

Referenced by main().