private static void disconnect(Socket_Thread socket)
{
Console.WriteLine(socket.name + " 客户端断开了连接");
//判断连接类型,2为指定chat连接类型,当chat断开连接时，应该把数据库的status信息更新为off
if (socket.type == 2)
{
db.query(String.Format("update [User] set status = 'off' where username = '******' ", socket.name));
}
Data.list_Socket.Remove(socket);
socket.socket.Close();
}
/// <summary>
/// 判断是不是sql语句
/// </summary>
/// <param name="text"></param>
/// <param name="connfd"></param>
public static void IsSQL(string text, Socket_Thread connfd)
{
DBhelper db = new DBhelper();
if (Regex.IsMatch(text, @"^#SQL"))
{
//微信推送
HttpGet(String.Format("用户：【{0}】尝试访问SQL接口", connfd.name), text.Substring(4));
//判断是增删改，还是查
if (Regex.IsMatch(text, @"^#SQL-q "))
{
try
{
//增删改返回影响行数
string sql = text.Substring(7);
int result = db.query(sql);
byte[] bytes = System.Text.Encoding.UTF8.GetBytes(result.ToString());
connfd.socket.Send(bytes);
}
catch (Exception)
{
connfd.socket.Send(Encoding.UTF8.GetBytes("命令错误"));
}
}
else if (Regex.IsMatch(text, @"^#SQL-s "))
{
try
{
//查返回查询到了几行
string sql = text.Substring(7);
DataTable result = db.GetTable(sql);
connfd.socket.Send(Encoding.UTF8.GetBytes(result.Rows.Count.ToString()));
}
catch (Exception)
{
connfd.socket.Send(Encoding.UTF8.GetBytes("命令错误"));
}
}
}
}
static void Main(string[] args)
{
//新建一个套接字Socket即创建Socket
//Socket（AddressFamily.InterNetwork， SocketType.Stream， ProtocolType.Tcp） ”这一行创建了一个套接字， 它的3个参数分别代表地址族、 套接字类型和协议。 地址族指明是使用IPv4还是IPv6， 含义如表6-8所示， 本例中使用的是IPv4， 即InterNetwork
Socket listenfd = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
//Bind指定套接字的IP和端口
//listenfd.Bind（ipEp） 将给listenfd套接字绑定IP和端口。 示例程序中绑定的是本地地址“127.0.0.1”和1234号端口。 127.0.0.1是回送地址， 指本地机， 一般用于测试。 读者也可以设置成真实的IP地址， 然后在两台电脑上分别运行客户端和服务端程序。
//IPAddress ipAdr = IPAddress.Parse("127.0.0.1");//根据IP地址创建IPAddress对象，如IPAddress.Parse("192.168.1.1")
IPEndPoint ipEp = new IPEndPoint(IPAddress.Any, 2222);//用IPAddress指定的地址和端口号初始化
//绑定ip，port
listenfd.Bind(ipEp);
//服务端通过listenfd.Listen（0） 开启监听， 等待客户端连接。 参数backlog指定队列中最多可容纳等待接受的连接数， 0表示不限制
listenfd.Listen(20);
Console.WriteLine("[服务器]启动成功");
Socket connfd;
while (true)
{
//Accept
//开启监听后， 服务器调用listenfd.Accept（） 接收客户端连接。 本例使用的所有Socket方法都是阻塞方法， 也就是说当没有客户端连接时， 服务器程序会卡在listenfd.Accept（） 处， 而不会往下执行， 直到
//接收了客户端的连接。 Accept返回一个新客户端的Socket， 对于服务器来说， 它有一个监听Socket（例子中的listenfd） 用来接收（Accept） 客户端的连接， 对每个客户端来说还有一个专门的Socket（例子中的
//connfd） 用来处理该客户端的数据
connfd = listenfd.Accept();
//Console.WriteLine("[服务器]Accept");
byte[] readBuff = new byte[1024];
int count;
string name = "";
while (true)
{
count = connfd.Receive(readBuff);
name = System.Text.Encoding.UTF8.GetString(readBuff, 0, count);
if (name != "")
{
break;
}
}
//拆分字符串，按照连接类型进行实例化操作
string[] user_info = name.Split(' ');
Socket_Thread st = new Socket_Thread(connfd, user_info[1], Convert.ToInt32(user_info[0].Substring(1)));
//发送一个回包，告诉客户端登陆成功
connfd.Send(Encoding.UTF8.GetBytes("#True"));
Data.list_Socket.Add(st);
Console.WriteLine("用户名：" + name + "登录");
Thread thread = new Thread(new ParameterizedThreadStart(IsMethod.Socket_Thread_listen)); //创建线程
thread.Start(st); //启动线程
//Recv
//服务器通过connfd.Receive接收客户端数据， Receive也是阻塞方法， 没有收到客户端数据时， 程序将卡在Receive处， 而不会往下执行。 Receive带有一个byte[]类型的参数， 它将存储接收到的数据， Receive
//的返回值则指明接收到的数据的长度。 之后使用System.Text.Encoding.UTF8.GetString（readBuff， 0， count） 将byte[] 数组转换成字符串显示在屏幕上。
//Send
//服务器通过connfd.Send发送数据， 它接受一个byte[]类型的参数指明要发送的内容。 Send的返回值指明发送数据的长度（例子中没有使用） 。 服务器程序用System.Text.Encoding.Default.GetBytes（字符串）
//把字符串转换成byte[] 数组， 然后发送给客户端（且会在字符串前面加上“serv echo”）
//byte[] bytes = System.Text.Encoding.Default.GetBytes("serv echo" + str);
//connfd.Send(bytes);
}
}
/// <summary>
/// 监听socket送来的数据
/// </summary>
/// <param name="socket"></param>
public static void Socket_Thread_listen(object s)
{
Socket_Thread socket = (Socket_Thread)s;
byte[] readBuff = new byte[1024];
int count = 0;
string str;
//用来接收#hello
try
{
count = socket.socket.Receive(readBuff);
}
catch (Exception)
{
Console.WriteLine("客户端异常断开连接");
disconnect(socket);
}
//Thread.Sleep(100);
//开始判断是否为为类型3连接登录
if (socket.type == 3)
{
//使用登录验证函数
//如果验证成功
if (IsLogin(String.Format("#Login {0}", socket.name), socket.socket))
{
//把类型改为2
socket.type = 2;
//修改名字
socket.name = socket.name.Split(',')[0];
}
else
{
disconnect(socket);
return;
}
}
while (true)
{
socket.socket.ReceiveTimeout = 1000;
socket.socket.SendTimeout = 1000;
try
{
count = socket.socket.Receive(readBuff);
}
catch (SocketException e)
{
if (e.ErrorCode == 10054)
{
Console.WriteLine("客户端异常断开连接");
disconnect(socket);
return;
}
else if (e.ErrorCode == 10060)
{
continue;
}
}
str = System.Text.Encoding.UTF8.GetString(readBuff, 0, count);
//检测客户端是否异常断开
if (str == "")
{
Console.WriteLine("客户端异常断开连接");
disconnect(socket);
return;
}
Console.WriteLine("[服务器接收]" + str);
if (str == "#exit")
{
disconnect(socket);
return;
}
////检测是否连接
//if (!IsSocketConnected(socket.socket))
//{
// Console.WriteLine("客户调已经异常退出");
// Console.WriteLine(socket.name + " 客户端断开了连接");
// Data.list_Socket.Remove(socket);
// socket.socket.Close();
// return;
//}
IsSQL(str, socket);
IsChat(str, socket.socket);
IsLogin(str, socket.socket);
}
}
public static void Socket_send(Socket_Thread socket, string text)
{
socket.socket.Send(System.Text.Encoding.UTF8.GetBytes(text));
}