zoukankan      html  css  js  c++  java
  • 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);

  • 相关阅读:
    EXE中释放文件
    关闭GS选项,解决注入后崩溃
    HDU2516 取石子游戏
    HDU2188 选拔志愿者
    HDU2149 Public Sale
    HDU2147 kiki's game
    HDU1846 Brave Game
    LightOJ1214 Large Division
    POJ2480 Longge's problem
    HDU 5880 Family View
  • 原文地址:https://www.cnblogs.com/xuexian/p/3532128.html
Copyright © 2011-2022 走看看