313 lines
8.2 KiB
C#
313 lines
8.2 KiB
C#
/********************************************************************************
|
||
* 文件名：SocketOutputStream.cs
|
||
* 全路径： \NetWork\SocketAPI\SocketOutputStream.cs
|
||
* 创建人： 王华
|
||
* 创建时间：2013-11-29
|
||
*
|
||
* 功能说明： Socket的应用层输出缓冲区
|
||
*
|
||
* 修改记录：
|
||
*********************************************************************************/
|
||
|
||
using System;
|
||
using System.Collections.Generic;
|
||
using System.Net;
|
||
#if UNITY_WP8
|
||
using UnityPortSocket;
|
||
#else
|
||
using System.Net.Sockets;
|
||
#endif
|
||
using System.Text;
|
||
|
||
#pragma warning disable
|
||
|
||
namespace SPacket.SocketInstance
|
||
{
|
||
public class SocketOutputStream
|
||
{
|
||
public const bool FALSE = false;
|
||
public const bool TRUE = true;
|
||
public const uint SOCKET_ERROR = 0xFFFFFFFF;
|
||
public const uint DEFAULT_SOCKET_OUTPUT_BUFFER_SIZE = 1024 * 8;
|
||
public const uint DISCONNECT_SOCKET_OUTPUT_BUFFER_SIZE = 1024 * 20;
|
||
|
||
public SocketOutputStream( SocketInstance sock,
|
||
uint BufferLen = DEFAULT_SOCKET_OUTPUT_BUFFER_SIZE,
|
||
uint MaxBufferLen = DISCONNECT_SOCKET_OUTPUT_BUFFER_SIZE)
|
||
{
|
||
m_pSocket = sock;
|
||
m_BufferLen = BufferLen;
|
||
m_MaxBufferLen = MaxBufferLen;
|
||
m_Head = 0;
|
||
m_Tail = 0;
|
||
m_Buffer = new Byte[m_BufferLen];
|
||
m_BufferTemp = new Byte[m_BufferLen];
|
||
}
|
||
|
||
void ClearBufferTemp()
|
||
{
|
||
if (m_BufferTemp != null)
|
||
{
|
||
int nCount = m_BufferTemp.Length;
|
||
Array.Clear(m_BufferTemp, 0, nCount);
|
||
}
|
||
}
|
||
|
||
public uint Write (Byte[] buf, uint len)
|
||
{
|
||
uint nFree = ((m_Head <= m_Tail) ? (m_BufferLen - m_Tail + m_Head - 1) : (m_Head - m_Tail - 1));
|
||
|
||
if (len >= nFree)
|
||
{
|
||
if (!Resize(len - nFree + 1))
|
||
return 0;
|
||
}
|
||
|
||
if (m_Head <= m_Tail)
|
||
{
|
||
if (m_Head == 0)
|
||
{
|
||
nFree = m_BufferLen - m_Tail - 1;
|
||
// memcpy(&m_Buffer[m_Tail], buf, len);
|
||
for (int i = 0; i < len;++i )
|
||
{
|
||
m_Buffer[m_Tail + i] = buf[i];
|
||
}
|
||
}
|
||
else
|
||
{
|
||
nFree = m_BufferLen - m_Tail;
|
||
if (len <= nFree)
|
||
{
|
||
//memcpy(&m_Buffer[m_Tail], buf, len);
|
||
for (int i = 0; i < len; ++i)
|
||
{
|
||
m_Buffer[m_Tail + i] = buf[i];
|
||
}
|
||
}
|
||
else
|
||
{
|
||
//memcpy(&m_Buffer[m_Tail], buf, nFree);
|
||
for (int i = 0; i < nFree; ++i)
|
||
{
|
||
m_Buffer[m_Tail + i] = buf[i];
|
||
}
|
||
for (int i = 0; i < len - nFree; ++i)
|
||
{
|
||
m_Buffer[i] = buf[nFree+i];
|
||
}
|
||
// memcpy(m_Buffer, &buf[nFree], len - nFree);
|
||
}
|
||
}
|
||
}
|
||
else
|
||
{
|
||
// memcpy(&m_Buffer[m_Tail], buf, len);
|
||
for (int i = 0; i < len; ++i)
|
||
{
|
||
m_Buffer[m_Tail + i] = buf[i];
|
||
}
|
||
}
|
||
|
||
m_Tail = (m_Tail + len) % m_BufferLen;
|
||
|
||
return len;
|
||
}
|
||
|
||
public uint Flush ()
|
||
{
|
||
|
||
if (m_pSocket == null) return 0;
|
||
|
||
uint nFlushed = 0;
|
||
uint nSent = 0;
|
||
uint nLeft;
|
||
SocketFlags flag = SocketFlags.None;
|
||
|
||
if (m_BufferLen > m_MaxBufferLen)
|
||
{
|
||
Initsize();
|
||
return SOCKET_ERROR;
|
||
}
|
||
try
|
||
{
|
||
if ( m_Head < m_Tail )
|
||
{
|
||
nLeft = m_Tail - m_Head;
|
||
|
||
while ( nLeft > 0 )
|
||
{
|
||
ClearBufferTemp();
|
||
for (int i=0;i<nLeft;++i)
|
||
{
|
||
m_BufferTemp[i] = m_Buffer[m_Head+i];
|
||
}
|
||
nSent = m_pSocket.Send(m_BufferTemp, (int)nLeft, flag);
|
||
//LogModule.DebugLog(nSent.ToString());
|
||
//if (nSent==SOCKET_ERROR_WOULDBLOCK) return 0 ;
|
||
if (nSent==SOCKET_ERROR) return SOCKET_ERROR ;
|
||
if (nSent==0) return 0;
|
||
|
||
nFlushed += nSent;
|
||
nLeft -= nSent;
|
||
m_Head += nSent;
|
||
}
|
||
|
||
|
||
|
||
}
|
||
else if( m_Head > m_Tail )
|
||
{
|
||
nLeft = m_BufferLen - m_Head;
|
||
|
||
while ( nLeft > 0 )
|
||
{
|
||
ClearBufferTemp();
|
||
for (int i=0;i<nLeft;++i)
|
||
{
|
||
m_BufferTemp[i] = m_Buffer[m_Head+i];
|
||
}
|
||
|
||
nSent = m_pSocket.Send(m_BufferTemp , (int)nLeft , flag );
|
||
//if (nSent==SOCKET_ERROR_WOULDBLOCK) return 0 ;
|
||
if (nSent==SOCKET_ERROR) return SOCKET_ERROR-3 ;
|
||
if (nSent==0) return 0;
|
||
|
||
nFlushed += nSent;
|
||
nLeft -= nSent;
|
||
m_Head += nSent;
|
||
}
|
||
|
||
// Assert( m_Head == m_BufferLen );
|
||
|
||
m_Head = 0;
|
||
|
||
nLeft = m_Tail;
|
||
|
||
while ( nLeft > 0 )
|
||
{
|
||
ClearBufferTemp();
|
||
for (int i=0;i<nLeft;++i)
|
||
{
|
||
m_BufferTemp[i] = m_Buffer[m_Head + i];
|
||
}
|
||
|
||
nSent = m_pSocket.Send(m_BufferTemp, (int)nLeft, flag);
|
||
//if (nSent==SOCKET_ERROR_WOULDBLOCK) return 0 ;
|
||
if (nSent==SOCKET_ERROR) return SOCKET_ERROR ;
|
||
if (nSent==0) return 0;
|
||
|
||
nFlushed += nSent;
|
||
nLeft -= nSent;
|
||
m_Head += nSent;
|
||
}
|
||
|
||
|
||
}
|
||
|
||
}
|
||
catch
|
||
{
|
||
if( nSent>0 )
|
||
{
|
||
m_Head += nSent;
|
||
}
|
||
}
|
||
|
||
m_Head = m_Tail = 0 ;
|
||
|
||
return nFlushed;
|
||
|
||
}
|
||
public void Initsize (uint BufferLen = DEFAULT_SOCKET_OUTPUT_BUFFER_SIZE)
|
||
{
|
||
m_Head = 0;
|
||
m_Tail = 0;
|
||
if (BufferLen > m_MaxBufferLen)
|
||
{
|
||
BufferLen = m_MaxBufferLen;
|
||
}
|
||
|
||
m_Buffer = new Byte[BufferLen];
|
||
m_BufferLen = BufferLen;
|
||
}
|
||
public bool Resize( uint size )
|
||
{
|
||
|
||
uint orgSize = size;
|
||
|
||
size = Math.Max(size, (uint)(m_BufferLen >> 1));
|
||
uint newBufferLen = m_BufferLen + size;
|
||
uint len = Length();
|
||
|
||
if (size < 0)
|
||
{
|
||
if (newBufferLen < 0 || newBufferLen < len)
|
||
return false;
|
||
}
|
||
|
||
Byte[] newBuffer = new Byte[newBufferLen];
|
||
if (newBuffer == null)
|
||
return false;
|
||
|
||
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];
|
||
}
|
||
}
|
||
m_Buffer = newBuffer;
|
||
m_BufferLen = newBufferLen;
|
||
m_Head = 0;
|
||
m_Tail = len;
|
||
|
||
return true;
|
||
}
|
||
|
||
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 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;
|
||
}
|
||
}
|