/********************************************************************************
* 文件名：SocketInputStream.cs
* 全路径： \NetWork\SocketAPI\SocketInputStream.cs
* 创建人： 王华
* 创建时间：2013-11-29
*
* 功能说明： Socket的应用层输入缓冲区
*
* 修改记录：
*********************************************************************************/
using System;
using System.Collections.Generic;
using System.Net;
using Module.Log;
#if UNITY_WP8
using UnityPortSocket;
#else
using System.Net.Sockets;
#endif
namespace SPacket.SocketInstance
{
public class SocketInputStream
{
public const uint SOCKET_ERROR = 0xFFFFFFFF;
public const uint DEFAULT_SOCKET_INPUT_BUFFER_SIZE = 1024 * 256;
public const uint MAX_SOCKET_INPUT_BUFFER_SIZE = 1024 * 1024 * 8;
public SocketInputStream (SocketInstance socket,
uint BufferLen = DEFAULT_SOCKET_INPUT_BUFFER_SIZE,
uint MaxBufferLen = MAX_SOCKET_INPUT_BUFFER_SIZE)
{
m_pSocket = socket;
m_BufferLen = BufferLen;
m_MaxBufferLen = MaxBufferLen;
m_Head = 0;
m_Tail = 0;
m_Buffer = new Byte[BufferLen];
m_BufferTemp = new Byte[BufferLen];
}
void ClearBufferTemp()
{
if (m_BufferTemp != null)
{
int nCount = m_BufferTemp.Length;
Array.Clear(m_BufferTemp, 0, nCount);
}
}
public uint Length( )
{
if( m_Head<m_Tail )
return m_Tail-m_Head;
else if( m_Head>m_Tail )
return m_BufferLen-m_Head+m_Tail ;
return 0 ;
}
public uint Read(Byte[] buf, uint len )
{
if ( len == 0 )
return 0 ;
if ( len > Length() )
return 0 ;
if ( m_Head < m_Tail )
{
for(uint i =0;i< len;++i)
{
buf[i] = m_Buffer[m_Head+i];
}
}
else
{
uint rightLen = m_BufferLen-m_Head ;
if( len<=rightLen )
{
for (uint i = 0; i < len; ++i)
{
buf[i] = m_Buffer[m_Head + i];
}
}
else
{
for (uint i = 0; i < rightLen; ++i)
{
buf[i] = m_Buffer[m_Head + i];
}
for (uint i = 0; i < len - rightLen; ++i)
{
buf[i + rightLen] = m_Buffer[i];
}
}
}
m_Head = (m_Head+len)%m_BufferLen ;
return len ;
}
public bool Peek( Byte[] buf,uint len )
{
if( len==0 )
return false ;
if( len>Length() )
return false ;
if( m_Head<m_Tail )
{
for (uint i = 0; i < len; ++i)
{
buf[i] = m_Buffer[m_Head + i];
}
}
else
{
uint rightLen = m_BufferLen-m_Head ;
if( len<=rightLen )
{
for (uint i = 0; i < len; ++i)
{
buf[i] = m_Buffer[m_Head + i];
}
}
else
{
for (uint i = 0; i < rightLen; ++i)
{
buf[i] = m_Buffer[m_Head + i];
}
for (uint i = 0; i < len - rightLen; ++i)
{
buf[rightLen+i] = m_Buffer[ i];
}
}
}
return true ;
}
public bool Skip (uint len)
{
if (len == 0)
return false;
if (len > Length())
return false;
m_Head = (m_Head + len) % m_BufferLen;
return true;
}
public uint Fill( )
{
uint nFilled = 0;
uint nReceived = 0;
uint nFree = 0;
if(m_pSocket==null) return 0;
if ( m_Head <= m_Tail )
{
if ( m_Head == 0 )
{
//
// H T LEN=10
// 0123456789
// abcd......
//
nReceived = 0 ;
nFree = m_BufferLen-m_Tail-1 ;
if( nFree != 0 )
{
ClearBufferTemp();
nReceived = m_pSocket.receive(m_BufferTemp, (int)nFree);
if (nReceived == SOCKET_ERROR) return SOCKET_ERROR;
for (int i = 0; i < nReceived; ++i)
{
m_Buffer[m_Tail + i] = m_BufferTemp[i];
}
// if (nReceived==SOCKET_ERROR_WOULDBLOCK) return 0 ;
// if (nReceived==SOCKET_ERROR) return SOCKET_ERROR-1 ;
m_Tail += nReceived;
nFilled += nReceived;
}
if( nReceived == nFree )
{
uint available = m_pSocket.available();
if ( available > 0 )
{
if( (m_BufferLen+available+1)>m_MaxBufferLen )
{
var value = m_BufferLen + available + 1;
Initsize( ) ;
return SOCKET_ERROR ;
}
if( !Resize( available+1 ) )
return 0 ;
ClearBufferTemp();
nReceived = m_pSocket.receive(m_BufferTemp, (int)available);
if (nReceived == SOCKET_ERROR) return SOCKET_ERROR;
for (int i = 0; i < nReceived; ++i)
{
m_Buffer[m_Tail + i] = m_BufferTemp[i];
}
// if (nReceived==SOCKET_ERROR_WOULDBLOCK) return 0 ;
// if (nReceived==SOCKET_ERROR) return SOCKET_ERROR-4 ;
//if (nReceived==0) return SOCKET_ERROR-5;
m_Tail += nReceived;
nFilled += nReceived;
}
}
}
else
{
//
// H T LEN=10
// 0123456789
// ...abcd...
//
nFree = m_BufferLen-m_Tail ;
ClearBufferTemp();
nReceived = m_pSocket.receive(m_BufferTemp, (int)nFree);
if (nReceived == SOCKET_ERROR) return SOCKET_ERROR;
for (int i = 0; i < nReceived; ++i)
{
m_Buffer[m_Tail + i] = m_BufferTemp[i];
}
//nReceived = m_pSocket->receive( &m_Buffer[m_Tail], nFree );
// if( nReceived==SOCKET_ERROR_WOULDBLOCK ) return 0 ;
// if( nReceived==SOCKET_ERROR ) return SOCKET_ERROR-6 ;
// if( nReceived==0 ) return SOCKET_ERROR-7 ;
m_Tail = (m_Tail+nReceived)%m_BufferLen ;
nFilled += nReceived ;
if( nReceived==nFree )
{
// Assert( m_Tail == 0 );
nReceived = 0 ;
nFree = m_Head-1 ;
if( nFree!=0 )
{
ClearBufferTemp();
nReceived = m_pSocket.receive(m_BufferTemp, (int)nFree);
if (nReceived == SOCKET_ERROR) return SOCKET_ERROR;
for (int i = 0; i < nReceived; ++i)
{
m_Buffer[i] = m_BufferTemp[i];
}
// nReceived = m_pSocket->receive( &m_Buffer[0] , nFree );
// if( nReceived==SOCKET_ERROR_WOULDBLOCK ) return 0 ;
// if( nReceived==SOCKET_ERROR ) return SOCKET_ERROR -8;
// if( nReceived==0 ) return SOCKET_ERROR-9 ;
m_Tail += nReceived;
nFilled += nReceived;
}
if( nReceived==nFree )
{
uint available = m_pSocket.available();
if ( available > 0 )
{
if( (m_BufferLen+available+1)>m_MaxBufferLen )
{
Initsize( ) ;
return SOCKET_ERROR ;
}
if( !Resize( available+1 ) )
return 0 ;
ClearBufferTemp();
nReceived = m_pSocket.receive(m_BufferTemp, (int)available);
if (nReceived == SOCKET_ERROR) return SOCKET_ERROR;
for (int i = 0; i < nReceived; ++i)
{
m_Buffer[m_Tail + i] = m_BufferTemp[i];
}
// nReceived = m_pSocket->receive( &m_Buffer[m_Tail] , available );
//if (nReceived==SOCKET_ERROR_WOULDBLOCK) return 0 ;
//if (nReceived==SOCKET_ERROR) return SOCKET_ERROR-11 ;
// if (nReceived==0) return SOCKET_ERROR-12;
m_Tail += nReceived;
nFilled += nReceived;
}
}
}
}
}
else
{
//
// T H LEN=10
// 0123456789
// abcd...efg
//
nReceived = 0 ;
nFree = m_Head-m_Tail-1 ;
if( nFree!=0 )
{
ClearBufferTemp();
nReceived = m_pSocket.receive(m_BufferTemp, (int)nFree);
if (nReceived == SOCKET_ERROR) return SOCKET_ERROR;
for (int i = 0; i < nReceived; ++i)
{
m_Buffer[m_Tail + i] = m_BufferTemp[i];
}
// nReceived = m_pSocket->receive( &m_Buffer[m_Tail], nFree ) ;
// if( nReceived==SOCKET_ERROR_WOULDBLOCK ) return 0 ;
//if( nReceived==SOCKET_ERROR ) return SOCKET_ERROR-13 ;
//if( nReceived==0 ) return SOCKET_ERROR-14 ;
m_Tail += nReceived ;
nFilled += nReceived ;
}
if( nReceived==nFree )
{
uint available = m_pSocket.available( ) ;
if ( available>0 )
{
if( (m_BufferLen+available+1)>m_MaxBufferLen )
{
Initsize( ) ;
return SOCKET_ERROR;
}
if( !Resize( available+1 ) )
return 0 ;
ClearBufferTemp();
nReceived = m_pSocket.receive(m_BufferTemp, (int)available);
if (nReceived == SOCKET_ERROR) return SOCKET_ERROR;
for (int i = 0; i < nReceived; ++i)
{
m_Buffer[m_Tail + i] = m_BufferTemp[i];
}
// nReceived = m_pSocket->receive( &m_Buffer[m_Tail], available ) ;
// if( nReceived==SOCKET_ERROR_WOULDBLOCK ) return 0 ;
// if( nReceived==SOCKET_ERROR ) return SOCKET_ERROR-16 ;
// if( nReceived==0 ) return SOCKET_ERROR-17 ;
m_Tail += nReceived ;
nFilled += nReceived ;
}
}
}
return nFilled ;
}
public void Initsize (uint BufferSize = DEFAULT_SOCKET_INPUT_BUFFER_SIZE)
{
m_Head = 0;
m_Tail = 0;
if (BufferSize > m_MaxBufferLen)
{
BufferSize = m_MaxBufferLen;
}
m_Buffer = new Byte[BufferSize];
m_BufferLen = BufferSize;
}
public bool Resize( uint size )
{
LogModule.DebugLog("Resize packet");
size = Math.Max(size, (uint)(m_BufferLen >> 1));
uint newBufferLen = (uint)(m_BufferLen + size);
uint len = Length();
if (m_BufferTemp.Length < newBufferLen)
{
m_BufferTemp = new Byte[newBufferLen];
ClearBufferTemp();
}
if (size < 0)
{
if (newBufferLen < 0 || newBufferLen < len)
return false;
}
Byte[] newBuffer = new Byte[newBufferLen];
if (m_Head < m_Tail)
{
//memcpy(newBuffer, &m_Buffer[m_Head], m_Tail - m_Head);
for (int i = 0; i < m_Tail - m_Head;++i )
{
newBuffer[i] = m_Buffer[m_Head + i];
}
}
else if (m_Head > m_Tail)
{
//memcpy(newBuffer, &m_Buffer[m_Head], m_BufferLen - m_Head);
for (int i = 0; i < m_BufferLen - m_Head; ++i)
{
newBuffer[i] = m_Buffer[m_Head + i];
}
//memcpy(&newBuffer[m_BufferLen - m_Head], m_Buffer, m_Tail);
for (int i = 0; i < m_Tail; ++i)
{
newBuffer[m_BufferLen - m_Head+i] = m_Buffer[i];
}
}
// delete[] m_Buffer;
m_Buffer = newBuffer;
m_BufferLen = newBufferLen;
m_Head = 0;
m_Tail = len;
return true;
}
public void CleanUp ()
{
m_Head = 0;
m_Tail = 0;
}
SocketInstance m_pSocket;
Byte[] m_Buffer =null;
Byte[] m_BufferTemp = null; //临时缓存，解决new的GC问题
uint m_BufferLen;
uint m_MaxBufferLen;
uint m_Head;
uint m_Tail;
}
}