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 *	array,
		MYDWORD	ip
	)

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 *

req

)

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 *	array,
		MYDWORD	ip
	)

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 *

network

)

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 *	sectionName,
		char *	buffer,
		MYBYTE	serial,
		char *	fileName
	)

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 *

lpParam

)

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;
}

IP2String()

char * IP2String	(	char *	target,
		MYDWORD	ip
	)

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	(	char *	logBuff,
		MYBYTE	logLevel
	)

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().

logMess() [2/2]

void logMess	(	request *	req,
		MYBYTE	logLevel
	)

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);
}

myGetToken()

char * myGetToken	(	char *	buff,
		MYBYTE	index
	)

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 *

string

)

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 *	target,
		char *	source
	)

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 *

string

)

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 *	sectionName,
		MYBYTE	serial,
		char *	fileName
	)

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 *	buff,
		FILE *	f
	)

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().