Socket操作类

using System; using System.Collections.Generic; using System.Linq; using System.Text; using System.Net.Sockets; using System.Collections; using System.Net; using System.Runtime.Serialization; using System.Runtime.Serialization.Formatters.Binary; using System.IO; using System.Collections.Specialized; using System.Threading; public class DuxSocketClient { #region 私有字段 /// <summary> /// 设置数据缓冲区大小 默认1024 /// </summary> private static int m_maxpacket = 1024 * 4; public delegate void SendFileProgress(int progress); public delegate void ReceiveFileProgress(int progress); #endregion #region 服务器侦听 /// <summary> /// 服务器侦听方法 返回null则说明没有链接上 /// </summary> /// <returns>返回一个套接字(Socket)</returns> public static Socket ListenerSocket(TcpListener listener) { try { Socket socket = listener.AcceptSocket(); return socket; } catch { return null; } } /// <summary> /// 服务器侦听方法 返回null则说明没有链接上 /// </summary> /// <param name="listener"></param> /// <returns>返回一个网络流</returns> public static NetworkStream ListenerStream(TcpListener listener) { try { TcpClient client = listener.AcceptTcpClient(); return client.GetStream(); } catch { return null; } } #endregion #region 客户端连接 public static Socket ConnectSocket(TcpClient tcpclient, IPEndPoint ipendpoint) { try { tcpclient.Connect(ipendpoint); return tcpclient.Client; } catch { return null; } } public static Socket ConnectSocket(TcpClient tcpclient, IPAddress ipadd, int port) { try { tcpclient.Connect(ipadd, port); return tcpclient.Client; } catch { return null; } } public static NetworkStream ConnectStream(TcpClient tcpclient, IPEndPoint ipendpoint) { try { tcpclient.Connect(ipendpoint); return tcpclient.GetStream(); } catch { return null; } } public static NetworkStream ConnectStream(TcpClient tcpclient, IPAddress ipadd, int port) { try { tcpclient.Connect(ipadd, port); return tcpclient.GetStream(); } catch { return null; } } #endregion #region Socket接收数据 /// <summary> /// 接受固定长度字符串 /// </summary> /// <param name="socket"></param> /// <param name="size"></param> /// <returns></returns> public static byte[] ReceiveFixData(Socket socket, int size) { int offset = 0; int recv = 0; int dataleft = size; byte[] msg = new byte[size]; while (dataleft > 0) { recv = socket.Receive(msg, offset, dataleft, 0); if (recv == 0) { break; } offset += recv; dataleft -= recv; } return msg; } /// <summary> /// 接收变长字符串 /// 为了处理粘包问题 ,每次发送数据时 包头(数据字节长度) + 正文 /// 这个发送小数据 /// 设置包头的字节为8,不能超过8位数的字节数组 /// </summary> /// <param name="socket"></param> /// <returns>byte[]数组</returns> public static byte[] ReceiveVarData(Socket socket) { //每次接受数据时,接收固定长度的包头,包头长度为8 byte[] lengthbyte = ReceiveFixData(socket, 8); //length得到字符长度 然后加工处理得到数字 int length = GetPacketLength(lengthbyte); //得到正文 return ReceiveFixData(socket, length); } /// <summary> /// 接收T类对象,反序列化 /// </summary> /// <typeparam name="T">接收T类对象,T类必须是一个可序列化类</typeparam> /// <param name="socket"></param> /// <returns></returns> public static T ReceiveVarData<T>(Socket socket) { //先接收包头长度 固定8个字节 byte[] lengthbyte = ReceiveFixData(socket, 8); //得到字节长度 int length = GetPacketLength(lengthbyte); byte[] bytecoll = new byte[m_maxpacket]; IFormatter format = new BinaryFormatter(); MemoryStream stream = new MemoryStream(); int offset = 0; //接收字节个数 int lastdata = length; //还剩下多少没有接收,初始大小等于实际大小 int receivedata = m_maxpacket; //每次接收大小 //循环接收 int mark = 0; //标记几次接收到的数据为0长度 while (true) { //剩下的字节数是否小于缓存大小 if (lastdata < m_maxpacket) { receivedata = lastdata; //就只接收剩下的字节数 } int count = socket.Receive(bytecoll,0,receivedata,0); if (count > 0) { stream.Write(bytecoll, 0, count); offset += count; lastdata -= count; mark = 0; } else { mark++; if (mark == 10) { break; } } if (offset == length) { break; } } stream.Seek(0, SeekOrigin.Begin); //必须要这个 或者stream.Position = 0; T t = (T)format.Deserialize(stream); stream.Close(); return t; } /// <summary> /// 在预先得到文件的文件名和大小 /// 调用此方法接收文件 /// </summary> /// <param name="socket"></param> /// <param name="path">路径必须存在</param> public static bool ReceiveFile(Socket socket, string path, string filename, long size,ReceiveFileProgress progress) { bool ret = false; if (Directory.Exists(path)) { //主要是防止有重名文件 string savepath = GetPath(path, filename); //得到文件路径 //缓冲区 byte[] file = new byte[m_maxpacket]; int count = 0; //每次接收的实际长度 int receivedata = m_maxpacket; //每次要接收的长度 long offset = 0; //循环接收的总长度 long lastdata = size; //剩余多少还没接收 int mark = 0; using (FileStream fs = new FileStream(savepath, FileMode.OpenOrCreate, FileAccess.Write)) { if (size > 0) { while (true) { if (lastdata < receivedata) { receivedata = Convert.ToInt32(lastdata); } count = socket.Receive(file, 0, receivedata, SocketFlags.None); if (count > 0) { fs.Write(file, 0, count); offset += count; lastdata -= count; mark = 0; } else { mark++; //连续5次接收为0字节 则跳出循环 if (mark ==10) { break; } } //接收进度 if (progress != null) { progress(Convert.ToInt32(((Convert.ToDouble(offset) / Convert.ToDouble(size)) * 100))); } //接收完毕 if (offset == size) { ret = true; break; } } } fs.Close(); } } return ret; } public static bool ReceiveFile(Socket socket, string path, string filename, long size) { return ReceiveFile(socket, path, filename, size,null); } /// <summary> /// 预先不知道文件名和文件大小 用此方法接收 /// 此方法对于的发送方法是SendFile() /// </summary> /// <param name="socket"></param> /// <param name="path">要保存的目录</param> public static void ReceiveFile(Socket socket, string path) { //得到包头信息字节数组 (文件名 + 文件大小 的字符串长度) //取前8位 byte[] info_bt = ReceiveFixData(socket, 8); //得到包头信息字符长度 int info_length = GetPacketLength(info_bt); //提取包头信息,(文件名 + 文件大小 的字符串长度) byte[] info = ReceiveFixData(socket, info_length); //得到文件信息字符串 (文件名 + 文件大小) string info_str = System.Text.Encoding.UTF8.GetString(info); string[] strs = info_str.Split('|'); string filename = strs[0]; //文件名 long length = Convert.ToInt64(strs[1]); //文件大小 //开始接收文件 ReceiveFile(socket, path, filename, length); } private static int GetPacketLength(byte[] length) { string str = System.Text.Encoding.UTF8.GetString(length); str = str.TrimEnd('*'); ;//("*", ""); int _length = 0; if (int.TryParse(str, out _length)) { return _length; } else { return 0; } } /// <summary> /// 得到文件路径(防止有文件名重复) /// 如:aaa.txt已经在directory目录下存在,则会得到文件aaa(1).txt /// </summary> /// <param name="directory">目录名</param> /// <param name="file">文件名</param> /// <returns>文件路径</returns> static int i = 0; static string markPath = String.Empty; public static string GetPath(string directory, string file) { if (markPath == String.Empty) { markPath = Path.Combine(directory, file); } string path = Path.Combine(directory, file); if (File.Exists(path)) { i++; string filename = Path.GetFileNameWithoutExtension(markPath) + "(" + i.ToString() + ")"; string extension = Path.GetExtension(markPath); return GetPath(directory, filename + extension); } else { i = 0; markPath = String.Empty; return path; } } #endregion #region Socket发送数据 /// <summary> /// 发送固定长度消息 /// 发送字节数不能大于int型最大值 /// </summary> /// <param name="socket"></param> /// <param name="msg"></param> /// <returns>返回发送字节个数</returns> public static int SendFixData(Socket socket, byte[] msg) { int size = msg.Length; //要发送字节长度 int offset = 0; //已经发送长度 int dataleft = size; //剩下字符 int senddata = m_maxpacket; //每次发送大小 while (true) { //如过剩下的字节数 小于 每次发送字节数 if (dataleft < senddata) { senddata = dataleft; } int count = socket.Send(msg, offset, senddata, SocketFlags.None); offset += count; dataleft -= count; if (offset == size) { break; } } return offset; } /// <summary> /// 发送变长信息 格式 包头(包头占8位) + 正文 /// </summary> /// <param name="socket"></param> /// <param name="contact">发送文本</param> /// <returns></returns> public static int SendVarData(Socket socket, string contact) { //得到字符长度 int size = System.Text.Encoding.UTF8.GetBytes(contact).Length; //包头字符 string length = GetSendPacketLengthStr(size); //包头 + 正文 byte[] sendbyte = System.Text.Encoding.UTF8.GetBytes(length + contact); //发送 return SendFixData(socket, sendbyte); } /// <summary> /// 发送变成信息 /// </summary> /// <param name="socket"></param> /// <param name="bytes"></param> /// <returns></returns> public static int SendVarData(Socket socket, byte[] bytes) { //得到包头字节 int size = bytes.Length; string length = GetSendPacketLengthStr(size); byte[] lengthbyte = System.Text.Encoding.UTF8.GetBytes(length); //发送包头 SendFixData(socket, lengthbyte); //因为不知道正文是什么编码所以没有合并 //发送正文 return SendFixData(socket, bytes); } /// <summary> /// 发送T类型对象,序列化 /// </summary> /// <typeparam name="T">T类型</typeparam> /// <param name="socket"></param> /// <param name="obj">T类型对象,必须是可序列化的</param> /// <returns></returns> public static int SendSerializeObject<T>(Socket socket, T obj) { byte[] bytes = SerializeObject(obj); return SendVarData(socket, bytes); } /// <summary> /// 发送文件 /// </summary> /// <param name="socket">socket对象</param> /// <param name="path">文件路径</param> /// <param name="issend">是否发送文件(头)信息,如果当前知道文件[大小,名称]则为false</param> /// <param name="progress"></param> /// <returns></returns> public static bool SendFile(Socket socket, string path,bool issend,SendFileProgress progress) { bool ret = false; if (File.Exists(path)) { FileInfo fileinfo = new FileInfo(path); string filename = fileinfo.Name; long length = fileinfo.Length; //发送文件信息 if (issend) { SendVarData(socket, filename + "|" + length); } //发送文件 long offset = 0; byte[] b = new byte[m_maxpacket]; int mark = 0; using (FileStream fs = new FileStream(path, FileMode.Open, FileAccess.Read)) { int senddata = b.Length; long i = length; //循环读取发送 while (true) { int count = fs.Read(b, 0, senddata); if (count > 0) { socket.Send(b, 0, count, SocketFlags.None); offset += count; mark = 0; } else { mark++; if (mark == 10) { break; } } if (progress != null) { progress(Convert.ToInt32(((Convert.ToDouble(offset) / Convert.ToDouble(length)) * 100))); } if (offset == length) { break; } Thread.Sleep(50); //设置等待时间,以免粘包 } } } return ret; } /// <summary> /// 发送文件,不需要进度信息 /// </summary> /// <param name="socket">socket对象</param> /// <param name="path">文件路径</param> /// <param name="issend">是否发生(头)信息</param> /// <returns></returns> public static bool SendFile(Socket socket, string path,bool issend) { return SendFile(socket, path, issend, null); } /// <summary> /// 发送文件,不需要进度信息和(头)信息 /// </summary> /// <param name="socket">socket对象</param> /// <param name="path">文件路径</param> /// <returns></returns> public static bool SendFile(Socket socket, string path) { return SendFile(socket, path, false, null); } private static byte[] SerializeObject(object obj) { IFormatter format = new BinaryFormatter(); MemoryStream stream = new MemoryStream(); format.Serialize(stream, obj); byte[] ret = stream.ToArray(); stream.Close(); return ret; } private static string GetSendPacketLengthStr(int size) { string length = size.ToString() + "********"; //得到size的长度 return length.Substring(0, 8); //截取前前8位 } #endregion #region NetworkStream接收数据 //没写 #endregion #region NetworkStream发送数据 //没写 #endregion }

 

说明:每个接收的方法都对应着有发送方法

 

如:

发送:SendFixData(socket,"01");

 

接收:ReceiveFixData(socket,2);  //size 就为2

 

不知道发送文本长度:

 

string txt = ????   //不知道有多少字符

发送:SendVarData(socket,txt); //有重载版

 

接收:ReceiveVarData(socket);