hxopwaveudpsock.cpp

来自「symbian 下的helix player源代码」· C++ 代码 · 共 537 行 · 第 1/2 页

    if (SUCCEEDED(res))
    {
        res = pReadBuffer->SetSize(m_ulReadSize);
    
        UCHAR* pData = pReadBuffer->GetBuffer();
        int nRead = read(pData, m_ulReadSize);
 
        //OpDPRINTF("UDP::DoRead: this=%p, read=%d, askfor=%d\n", this, nRead, m_ulReadSize);
        if (nRead > 0)
        {
            pReadBuffer->SetSize(nRead);
            m_pReadBuffer = pReadBuffer;
            // Don't ADDREF, m_pReadBuffer will be released in the callback OnReadDone
        
            if (m_pResponse)
            {
                /// still to be found that address and port arguments
                /// should be put here (remote or local)
                m_pResponse->ReadDone(res, m_pReadBuffer,
                    m_ulRemoteAddr, m_nRemotePort);
            }
            
        }
    }
    HX_RELEASE(m_pReadBuffer);
    return res;
}

STDMETHODIMP 
HXOpwaveUDPSocket::Write(THIS_
                         IHXBuffer*	pBuffer)
{
    
    DPRINTF(D_UDPSOCKET, ("HXOpwaveUDPSocket::Write()\n"));

    HX_RESULT res = HXR_OK;
    if (!pBuffer)
    {
    	res = HXR_INVALID_PARAMETER;
    }
  
    //OpDPRINTF("UDP:Write: this=%p, lwsize=%d, bread=%d, newwritesize=%d, bwrite=%d\n", this, m_ulBytesLeftToWrite, m_bReadable, pBuffer->GetSize(), m_bWritable);
    if (SUCCEEDED(res) && (m_state == udpBound))
    {
        /// First add to our list
        pBuffer->AddRef();
        m_writeList.AddTail(pBuffer);

        /// decide if we need to do a write 
        /// because last time when it is ready to write, there is no data 
        /// to be written.
        if (m_bWritable)
        {
            if (!m_pWriteBuffer)
            {
                m_pWriteBuffer = (IHXBuffer*)m_writeList.RemoveHead();
                m_ulBytesLeftToWrite = m_pWriteBuffer->GetSize();  
            }
            res =DoWrite();
            if (SUCCEEDED(res))
            {
                m_bWritable = FALSE;
            }
        }
    }
    return res;
}


/// This is working method to do actual write from
/// the m_pWriteBuffer which is guranteed to have data to be written
/// and the socket is ready to accept data
HX_RESULT HXOpwaveUDPSocket::DoWrite()
{
    HX_RESULT res = HXR_OK;
 
    UCHAR* pBufData = m_pWriteBuffer->GetBuffer(); 
    size_t ulActualWritten = write(pBufData, m_ulBytesLeftToWrite);
   
    HX_ASSERT(m_ulBytesLeftToWrite >= ulActualWritten);
 
    //OpDPRINTF("UDP::DoWrite, this=%p, written%d, write=%d\n\n",this, ulActualWritten, m_ulBytesLeftToWrite);
    m_ulBytesLeftToWrite -= ulActualWritten;
    if (m_ulBytesLeftToWrite > 0)
    {
        // more left in this m_pWriteBuffer to be written out
        UCHAR* pLeftData = new UCHAR[m_ulBytesLeftToWrite];
        if (!pLeftData)
        {
            res = HXR_OUTOFMEMORY;
            OnWriteDone(res);
            return res;
        }
            
        memcpy(pLeftData, pBufData+ulActualWritten, m_ulBytesLeftToWrite);
        m_pWriteBuffer->Set(pLeftData, m_ulBytesLeftToWrite); 
        delete pLeftData;
    }
    return res;
}

STDMETHODIMP
HXOpwaveUDPSocket::GetLocalPort(THIS_
                                REF(UINT16) port)
{
    
    DPRINTF(D_UDPSOCKET, ("HXOpwaveUDPSocket::GetLocalPort()\n"));
    
    HX_RESULT res = HXR_OK;
    port = m_nLocalPort;
    return res;
}



STDMETHODIMP 
HXOpwaveUDPSocket::SetOption(THIS_ 
                             HX_SOCKET_OPTION option,
                             UINT32 ulValue)
{
    
    DPRINTF(D_UDPSOCKET, ("HXOpwaveUDPSocket::SetOption(%d, %lu)\n", option, ulValue));
    HX_RESULT res = HXR_FAILED;
    
    if (m_state == udpNotInitialized)
    {
        switch(option)
        {
        case  HX_SOCKOPT_REUSE_ADDR:
        case HX_SOCKOPT_REUSE_PORT:
            //// OpSocket doesn't support set socket options
            {
                res = HXR_NOTIMPL;
            }
            break;
        case HX_SOCKOPT_BROADCAST:
        case HX_SOCKOPT_SET_RECVBUF_SIZE:
        case HX_SOCKOPT_SET_SENDBUF_SIZE:
        case HX_SOCKOPT_MULTICAST_IF:
            res = HXR_UNEXPECTED;
            break;
        default:
            break;
        }
    }
    
    return res;
}


STDMETHODIMP
HXOpwaveUDPSocket::WriteTo(THIS_ ULONG32	ulAddr,	UINT16 nPort,
                           IHXBuffer*		pBuffer)
{
    HX_RESULT res = HXR_OK;
    return res;
}

STDMETHODIMP
HXOpwaveUDPSocket::JoinMulticastGroup(THIS_	ULONG32	    ulMulticastAddr,
                                      ULONG32	    ulInterfaceAddr)
{
    HX_RESULT res = HXR_OK;
    return res;
}

STDMETHODIMP
HXOpwaveUDPSocket::LeaveMulticastGroup	(THIS_	ULONG32	    ulMulticastAddr,
                                         ULONG32	    ulInterfaceAddr)
{
    HX_RESULT res = HXR_OK;
    return res;
}


void HXOpwaveUDPSocket::OnWriteDone(HX_RESULT status)
{
    
    DPRINTF(D_UDPSOCKET, ("HXOpwaveUDPSocket::OnWriteDone(%ld)\n", status));
    DECLARE_SMART_POINTER_UNKNOWN scopeRef((IHXUDPSocket*)this);
    
    if (status == HXR_OK)
    {
        HX_RELEASE(m_pWriteBuffer);
        if (m_writeList.GetCount() > 0)
        {
            m_pWriteBuffer = (IHXBuffer*)m_writeList.RemoveHead();
            m_ulBytesLeftToWrite = m_pWriteBuffer->GetSize();     
        }
      
    }
    else
    {
        CloseSocket(status);
    }
}

void HXOpwaveUDPSocket::CloseSocket(HX_RESULT status)
{
    
    DPRINTF(D_UDPSOCKET, ("HXOpwaveUDPSocket::CloseConnection(%ld)\n", status));
    
    if (m_state != udpNotInitialized)
    {
        close(true);
    }	
    
    // Clear the writer list
    
    while (m_writeList.GetCount() > 0)
    {
        // Release all the left unsent buffers in the list
        IHXBuffer* pBuffer = (IHXBuffer*)m_writeList.RemoveHead();
        HX_RELEASE(pBuffer);
    }
    m_state = udpNotInitialized;
    
}

void HXOpwaveUDPSocket::onReadable(OpSocketEvent *pSocketEvent)
{
    
    HX_RESULT res = HXR_OK;
    //OpDPRINTF("UDP:onReadable: this=%p, mrsize=%d, bread=%d, state=%d, bwrite=%d\n", this, m_ulReadSize, m_bReadable, m_state, m_bWritable);
  
    if (m_ulReadSize > 0)
    {
        /// avoid recursion that might occur because 
        /// OnReadDone call in DoRead invoke ::Read again
        m_bReadable = FALSE;
        DoRead();
    }
}

void HXOpwaveUDPSocket::onWritable(OpSocketEvent *pSocketEvent)
{
    HX_RESULT res = HXR_OK;

    m_bWritable = TRUE;
    //OpDPRINTF("UDP:onWritable: this=%p, mrsize=%d, bread=%d, lwsize=%d, bwrite=%d\n", this, m_ulReadSize, m_bReadable, m_ulBytesLeftToWrite, m_bWritable);
  
  
    /// Normal write process
    /// Call OnWriteDone for handling the notification of last write
    if (m_ulBytesLeftToWrite == 0)
    {
        /// Notify last writing is completely done
        OnWriteDone(res);
    }
    
    /// Do the next writing
    if (m_pWriteBuffer && m_ulBytesLeftToWrite > 0)
    {
        if (DoWrite() == HXR_OK)
        {
            m_bWritable = FALSE;
        }
    }
}

void HXOpwaveUDPSocket::onException(OpSocketEvent *pSocketEvent)
{
    static int num = 0;
    DPRINTF(D_UDPSOCKET, ("HXOpwaveUDPSocket::OnException(%ld)\n", ++num));
}