Tools/WinSN5.0Ver.iso/NETSCAP.50/WIN1998.ZIP/ns/cmd/winfe/feselect.cpp

C/C++ Source or Header | 1998-04-08 | 4.9 KB | 190 lines

/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
 *
 * The contents of this file are subject to the Netscape Public License
 * Version 1.0 (the "NPL"); you may not use this file except in
 * compliance with the NPL.  You may obtain a copy of the NPL at
 * http://www.mozilla.org/NPL/
 *
 * Software distributed under the NPL is distributed on an "AS IS" basis,
 * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the NPL
 * for the specific language governing rights and limitations under the
 * NPL.
 *
 * The Initial Developer of this code under the NPL is Netscape
 * Communications Corporation.  Portions created by Netscape are
 * Copyright (C) 1998 Netscape Communications Corporation.  All Rights
 * Reserved.
 */

#include "stdafx.h"

#include "feselect.h"
#include "timer.h"
#include "ssl.h"
#include "hiddenfr.h"

//  We have one and only one select tracker.
CSelectTracker selecttracker;


#define MAX_SIMULTANIOUS_SOCKETS 100
/* should never have more than MAX sockets */
PRPollDesc poll_desc_array[MAX_SIMULTANIOUS_SOCKETS];
unsigned int fd_set_size = 0;

//  Have any entries in select list?
BOOL CSelectTracker::HasSelect(SelectType stType)
{
    return fd_set_size;
}

//  Number of select entries in list.
int CSelectTracker::CountSelect(SelectType stType)
{
    return(fd_set_size);
}

//  Add a select entry, no duplicates.
void CSelectTracker::AddSelect(SelectType stType, PRFileDesc *prFD)
{
    unsigned int index = fd_set_size;
    unsigned int count;

#define CONNECT_FLAGS PR_POLL_READ | PR_POLL_EXCEPT | PR_POLL_WRITE
#define READ_FLAGS    PR_POLL_READ | PR_POLL_EXCEPT

    //  Go through the list, make sure that there is not already an entry for fd.
    for(count=0; count < fd_set_size; count++)
    {
        if(poll_desc_array[count].fd == prFD)
        {
            // found it
            // verify that it has the same flags that we wan't,
            // otherwise add it again with different flags
            if((stType == ConnectSelect && poll_desc_array[count].in_flags == CONNECT_FLAGS)
               || (stType == SocketSelect && poll_desc_array[count].in_flags == READ_FLAGS))
            {
                index = count;
                break;
            }
        }
    }

    // need to add a new one if we didnt' find the index
    if(index == fd_set_size)
    {
        poll_desc_array[fd_set_size].fd = prFD;
        fd_set_size++;
    }

    if(stType == ConnectSelect)
        poll_desc_array[index].in_flags = CONNECT_FLAGS;
    else if(stType == SocketSelect)
        poll_desc_array[index].in_flags = READ_FLAGS;
    else
        XP_ASSERT(0);
}

//  Remove a select if it exists.
void CSelectTracker::RemoveSelect(SelectType stType, PRFileDesc *prFD)
{
    unsigned int count;

    //  Go through the list
    for(count=0; count < fd_set_size; count++)
    {
        if(poll_desc_array[count].fd == prFD)
        {

            if((stType == ConnectSelect && poll_desc_array[count].in_flags == CONNECT_FLAGS)
               || (stType == SocketSelect && poll_desc_array[count].in_flags == READ_FLAGS))
            {
                // found it collapse the list

                fd_set_size--;
                if(count < fd_set_size)
                    XP_MEMCPY(&poll_desc_array[count], &poll_desc_array[count+1], (fd_set_size - count) * sizeof(PRPollDesc));

                return;
            }
        }
    }

    // didn't find it.  opps
}

BOOL CSelectTracker::HasSelect(SelectType stType, PRFileDesc *prFD)
{
    unsigned int count;

    //  Go through the list
    for(count=0; count < fd_set_size; count++)
    {
        if(poll_desc_array[count].fd == prFD)
        {

            if((stType == ConnectSelect && poll_desc_array[count].in_flags == CONNECT_FLAGS)
               || (stType == SocketSelect && poll_desc_array[count].in_flags == READ_FLAGS))
               return TRUE;
        }
    }
    return FALSE;
}

extern "C"
void FE_AbortDNSLookup(PRFileDesc *fd)
{

}

extern "C"
void FE_SetReadPoll(PRFileDesc *fd)
{
    selecttracker.AddSelect(SocketSelect, fd);
}

extern "C"
void FE_ClearReadPoll(PRFileDesc *fd)
{
    selecttracker.RemoveSelect(SocketSelect, fd);
}

extern "C"
void FE_SetConnectPoll(PRFileDesc *fd)
{
    selecttracker.AddSelect(ConnectSelect, fd);
}

extern "C"
void FE_ClearConnectPoll(PRFileDesc *fd)
{
    selecttracker.RemoveSelect(ConnectSelect, fd);
}

/* call PR_Poll and call Netlib if necessary
 *
 * return FALSE if nothing to do.
 */
Bool CSelectTracker::PollNetlib(void)
{
    static PRIntervalTime interval = PR_MillisecondsToInterval(1);

    if(fd_set_size < 1)
        return FALSE;

    register int status = PR_Poll(poll_desc_array, fd_set_size, interval);

    if(status < 1)
        return TRUE; // potential for doing stuff in the future.

    // for now call on all active sockets.
    // if this is too much call only one, but reorder the list each time.
    for(register unsigned int count=0; count < fd_set_size; count++)
    {
        if(poll_desc_array[count].out_flags & (PR_POLL_READ | PR_POLL_WRITE | PR_POLL_EXCEPT))
            NET_ProcessNet(poll_desc_array[count].fd, NET_SOCKET_FD);
    }

    return TRUE;
}