/// <summary>
/// 通过用户名、密码、工具类型、工程id进行认证
/// </summary>
/// <param name="name">用户名</param>
/// <param name="passwd">密码</param>
/// <param name="tooltype">工具类型</param>
/// <param name="projectid">工程id</param>
/// <returns></returns>
public string CheckAuth(string name, string passwd, string tooltype, string projectid)
{
Message out_message = new Message();
Message in_message = new Message();
out_message.Command = Message.CommandHeader.LoginAuth;
//消息体携带用户名、密码信息
out_message.MessageBody = Encoding.Unicode.GetBytes(name + ":" + passwd + ":" + tooltype + ":" + projectid);
//获取服务器反馈 -->//可以加个定时器设置认证超时
in_message = StreamDataOutAndInHelper(out_message);
//假定认证成功服务器消息体返回allow,失败返回deny
if (in_message.Command == Message.CommandHeader.LoginAck
&& Encoding.Unicode.GetString(in_message.MessageBody).CompareTo("allow") == 0)
{
Console.WriteLine((in_message.Command == Message.CommandHeader.LoginAck
&& Encoding.Unicode.GetString(in_message.MessageBody).CompareTo("allow") == 0));
// dataStream.Write(Encoding.Unicode.GetBytes("ready"),0,"ready".Length);
out_message.MessageBody = Encoding.Unicode.GetBytes("ready");
dataStream.Write(out_message.ToBytes(), 0, out_message.MessageLength);
return "allow";
}
else if (in_message.Command == Message.CommandHeader.LoginAck
&& Encoding.Unicode.GetString(in_message.MessageBody).CompareTo("deny") == 0)
{
return "deny";
}
else if (in_message.Command == Message.CommandHeader.LoginAck
&& Encoding.Unicode.GetString(in_message.MessageBody).CompareTo("deny_pw") == 0)
{
return "wrongidpw";
}
else
return "fail";
}
/// <summary>
/// 传入要发送的消息，发送给服务器，返回解析后服务器反馈消息
/// </summary>
/// <param name="out_message"></param>
/// <returns></returns>
private Message StreamDataOutAndInHelper(Message out_message)
{
Message temp;
//获得输出流
dataStream = client.GetStream();
while (!dataStream.CanWrite)
{
System.Threading.Thread.Sleep(1000);//sleep 1s
}
//将打包成message的信息写入输出流
dataStream.Write(out_message.ToBytes(), 0, out_message.MessageLength);
//读取服务器反馈
temp = Message.Parse(dataStream);
return temp;
}
/// <summary>
/// 客户端发送从客户端传指定工程项目xml描述文件projectname到服务器的请求，得到服务器的反馈
/// 客户端传输文件给服务器需要服务器的权限确认后，客户端才能发送文件给服务器
/// </summary>
/// <param name="solutionName">项目名</param>
/// <param name="projectName">工程名</param>
/// <returns></returns>
public Message SendXmlRequest(string solutionName,string projectName)
{
Message out_message = new Message();
out_message.Command = Message.CommandHeader.SendXmlRequest;
out_message.MessageBody = Encoding.Unicode.GetBytes(solutionName+":"+projectName);
//获取服务器反馈
return StreamDataOutAndInHelper(out_message);
}
/// <summary>
/// 发送客户端项目工程xml描述文件projectNameXml给服务器，整个文件的发送均在此方法内完成
/// </summary>
/// <param name="projectNameXml"></param>
/// <returns></returns>
public Boolean SendXml(string solutionName,string projectName,string xmlFilePath)
{
//打开本地文件，读取内容，分行写到message中,连续发送，不需客户端的确认
//第一个数据包消息体：文件名
//最后一个数据包消息体：#
try
{
Message out_message;
string solutionProjectDirectory = solutionName+"\\"+projectName;
//相对客户端运行目录的路径
string projectNameWithRelativePath = string.Empty;
//若传入的绝对路径参数不为空
if (xmlFilePath != null && File.Exists(xmlFilePath)) {
projectNameWithRelativePath = xmlFilePath;
}
else if (Directory.GetFiles(solutionProjectDirectory).Count() == 0)
{
//若传入的绝对路径参数为空，判断solutionProjectName目录下由xml描述文件存在
Console.WriteLine("there is no project file existing in the " + solutionProjectDirectory);
}
else
{
//找到一个最近时间戳的xml描述文件
//获取目录下所有文件的信息
FileInfo[] files = (new DirectoryInfo(solutionProjectDirectory)).GetFiles();
FileInfo theLastestFile = files.First();
foreach (var file in files) {
if (file.LastWriteTime > theLastestFile.LastWriteTime) {
theLastestFile = file;
}
}
projectNameWithRelativePath = solutionProjectDirectory + "\\" + theLastestFile;
}
using (StreamReader sr = new StreamReader(projectNameWithRelativePath, Encoding.Default))//当前目录下的projectNameXml
{
//传送文件开始的第一个数据包
out_message = new Message();
//获得文件名
string filename = projectNameWithRelativePath.Split('\\')[projectNameWithRelativePath.Split('\\').Length - 1];
//文件名包含项目及工程solutonName:projectName
out_message.MessageBody = Encoding.Unicode.GetBytes(solutionName+":"+projectName);
Console.WriteLine("The file name: " + filename);
out_message.Command = Message.CommandHeader.SendXml;
out_message.MessageFlag = Message.MessageFlagHeader.FileBegin;
dataStream.Write(out_message.ToBytes(), 0, out_message.MessageLength);
//下面传送文件内容
string line = string.Empty;
int packetNum = 1;
while ((line = sr.ReadLine()) != null)
{
out_message = new Message();
out_message.MessageBody = Encoding.Unicode.GetBytes(line);
out_message.Command = Message.CommandHeader.SendXml;
out_message.FilePacketNumber = packetNum++;//数据包添加序号
out_message.MessageFlag = Message.MessageFlagHeader.FileMiddle;
dataStream.Write(out_message.ToBytes(), 0, out_message.MessageLength);
}
//文件读取结束
out_message = new Message();
out_message.MessageBody = Encoding.Unicode.GetBytes("#");//该消息体最终被舍弃
out_message.Command = Message.CommandHeader.SendXml;
out_message.MessageFlag = Message.MessageFlagHeader.FileEnd;
dataStream.Write(out_message.ToBytes(), 0, out_message.MessageLength);
}
//读取服务器反馈，以判断服务器是否正确接收文件
Message in_message = Message.Parse(dataStream);
if (in_message.Command == Message.CommandHeader.ReceivedXml
&& Encoding.Unicode.GetString(in_message.MessageBody).CompareTo("yes") == 0)
{
return true;
}
}
catch (Exception ex)
{
Console.WriteLine("SendXml: "+ex.Message);
}
return false;
}
/// <summary>
/// 发送心跳包
/// </summary>
/// <param name="source"></param>
/// <param name="e"></param>
public void SendHelloMessage(object source,ElapsedEventArgs e)
{
Message hello_message = new Message();
hello_message.Command = Message.CommandHeader.Hello;
hello_message.MessageBody = Encoding.Unicode.GetBytes(
this.client.Client.LocalEndPoint.ToString() + " is Alive");
//判断连接、即使是断开了，也可能返回true，所以改判断无效
//if (this.client.Connected)
//{
// Console.WriteLine(this.client.Client.RemoteEndPoint.ToString() + " still connected.");//测试
//}
StreamDataOutAndInHelper(hello_message);
}
/// <summary>
/// 客户端发送任意格式文档给服务器，整个文件的发送均在此方法内完成
/// </summary>
/// <param name="solutionName"></param>
/// <param name="projectName"></param>
/// <param name="documentName"></param>
/// <returns></returns>
public Boolean SendDocument(string solutionName, string projectName, string documentName,string documentFilePath)
{
try
{
Message out_message;
string solutionProjectDirectory = solutionName + "\\" + projectName;
//文档的绝对路径
string projectDocumentFilePath = string.Empty;
//相对客户端运行目录的路径
//string projectNameWithRelativePath = solutionProjectDirectory+"\\"+documentName;
if (documentFilePath != null && File.Exists(documentFilePath)) {
projectDocumentFilePath = documentFilePath;
}
else if(Directory.GetFiles(solutionProjectDirectory).Count() == 0){
Console.WriteLine(documentName + " not exist.");
return false;
}
else if(File.Exists(solutionProjectDirectory + "\\" + documentName)){
projectDocumentFilePath = solutionProjectDirectory + "\\" + documentName;
}
else
{
//找到一个最近时间戳的文档
//获取目录下所有文件的信息
FileInfo[] files = (new DirectoryInfo(solutionProjectDirectory)).GetFiles();
FileInfo theLastestFile = files.First();
foreach (var file in files)
{
if (file.LastWriteTime > theLastestFile.LastWriteTime)
{
theLastestFile = file;
}
}
projectDocumentFilePath = solutionProjectDirectory + "\\" + theLastestFile;
}
//传送文件开始的第一个数据包
out_message = new Message();
out_message.MessageBody = Encoding.Unicode.GetBytes(solutionName + ":" + projectName + ":" + documentName);
out_message.Command = Message.CommandHeader.SendDocument;
out_message.MessageFlag = Message.MessageFlagHeader.FileBegin;
dataStream.Write(out_message.ToBytes(), 0, out_message.MessageLength);
//传送文件内容
using (FileStream fs = new FileStream(projectDocumentFilePath, FileMode.Open, FileAccess.Read))
{
BinaryReader br = new BinaryReader(fs);
byte[] buffer = new byte[8192];//2^13=8kb
try
{
int checksize;
do
{
checksize = br.Read(buffer, 0, 8192);
if (checksize > 0)
{
out_message = new Message();
out_message.Command = Message.CommandHeader.SendDocument;
out_message.MessageFlag = Message.MessageFlagHeader.FileMiddle;
out_message.MessageBody = new byte[checksize];
if (checksize < 8192)//最后一个数据包字节数不够
{
for (int i = 0; i < checksize; i++)
out_message.MessageBody[i] = buffer[i];
}
else
{
buffer.CopyTo(out_message.MessageBody, 0);
}
dataStream.Write(out_message.ToBytes(), 0, out_message.MessageLength);
}
} while (checksize > 0);
//最后一个数据包
out_message = new Message();
out_message.Command = Message.CommandHeader.SendDocument;
out_message.MessageFlag = Message.MessageFlagHeader.FileEnd;
out_message.MessageBody = Encoding.Unicode.GetBytes("#");//该消息体最终被舍弃
dataStream.Write(out_message.ToBytes(), 0, out_message.MessageLength);
//读取服务器反馈，以判断服务器是否正确接收文件
Message in_message = Message.Parse(dataStream);
if (in_message.Command == Message.CommandHeader.ReceivedDocument
&& Encoding.Unicode.GetString(in_message.MessageBody).CompareTo("yes") == 0)
{
return true;
}
}
catch (EndOfStreamException ex)
{
Console.WriteLine("SendDocument: "+ex.Message);
}
}
}
catch (Exception ex)
{
Console.WriteLine("SendDocument: " + ex.Message);
}
return false;
}
/// <summary>
/// 客户端接受服务器发过来的指定项目工程xml描述文件
/// </summary>
/// <returns></returns>
public Boolean GetXml()
{
//整个文件接收工作需要在这全部完成，阻塞方式
FileStream fs;
Message in_message;
Message out_message;
string solutionName = string.Empty;
string projectName = string.Empty;
//格式@主目录\\子目录\\工程描述文件
string projectNameAddTimestampWithRelativePath = string.Empty;
do
{
in_message = Message.Parse(dataStream);
switch (in_message.MessageFlag)
{
case Message.MessageFlagHeader.FileBegin:
//得到项目名和文件名，文件名不带后缀.xml
solutionName = Encoding.Unicode.GetString(in_message.MessageBody).Split(':')[0];
projectName = Encoding.Unicode.GetString(in_message.MessageBody).Split(':')[1];
//判断当前目录下是否存在该项目及工程文件夹
if (!Directory.Exists(solutionName + "\\" + projectName))
{
//在当前路径下创建项目文件夹
Directory.CreateDirectory(solutionName + "\\" + projectName);
}
//生成时间戳 格式：__2011_12_2_19_48_37
string timestamp = "__" + DateTime.Now.ToString().Replace(':', '_').Replace(' ', '_').Replace('/', '_');
projectNameAddTimestampWithRelativePath = solutionName + "\\" + projectName + "\\" + projectName + timestamp;
if (!File.Exists(projectNameAddTimestampWithRelativePath))
{
fs = File.Create(projectNameAddTimestampWithRelativePath);
fs.Close();//需要关闭该文件，否则下面无法获取文件锁，进行文件读写
}
break;
case Message.MessageFlagHeader.FileMiddle://得到文件内容,采用追加方式写入文件
if (!File.Exists(projectNameAddTimestampWithRelativePath))
{
Console.WriteLine(projectNameAddTimestampWithRelativePath + " not exist");
//发送接收失败的确认数据包
out_message = new Message();
out_message.Command = Message.CommandHeader.ReceivedXml;
out_message.MessageBody = Encoding.Unicode.GetBytes("no");
dataStream.Write(out_message.ToBytes(), 0, out_message.MessageLength);
return false;
}
fs = File.Open(projectNameAddTimestampWithRelativePath, FileMode.Append);
StreamWriter sw = new StreamWriter(fs, Encoding.Unicode, in_message.MessageBody.Length);
sw.WriteLine(Encoding.Unicode.GetString(in_message.MessageBody));
sw.Flush();
sw.Close();
fs.Close();
break;
case Message.MessageFlagHeader.FileEnd://文件传送结束,需要判断第一个数据包直接跳到这项的
//发送成功接收的确认数据包
out_message = new Message();
out_message.Command = Message.CommandHeader.ReceivedXml;
out_message.MessageBody = Encoding.Unicode.GetBytes("yes");
dataStream.Write(out_message.ToBytes(), 0, out_message.MessageLength);
return true;
default:
//发送接收失败的确认数据包
out_message = new Message();
out_message.Command = Message.CommandHeader.ReceivedXml;
out_message.MessageBody = Encoding.Unicode.GetBytes("no");
dataStream.Write(out_message.ToBytes(), 0, out_message.MessageLength);
Console.WriteLine("The xml file trasfer failed...");
//若存在接受不全的文件，删除
if (File.Exists(projectNameAddTimestampWithRelativePath))
{
File.Delete(projectNameAddTimestampWithRelativePath);
}
break;
}
} while (in_message.MessageFlag == Message.MessageFlagHeader.FileBegin ||
in_message.MessageFlag == Message.MessageFlagHeader.FileMiddle);
return false;
}
/// <summary>
/// 客户端获取服务器任意格式的文件
/// </summary>
public Boolean GetDocument()
{
//整个文件接收工作需要在这全部完成，阻塞方式
try
{
FileStream fs;
Message in_message;
Message out_message;
string solutionName = string.Empty;
string projectName = string.Empty;
string documentName = string.Empty;
string documentNameAddTimestampWithRelativePath = string.Empty;
do
{
in_message = Message.Parse(dataStream);
switch (in_message.MessageFlag)
{
case Message.MessageFlagHeader.FileBegin:
//得到项目名
solutionName = Encoding.Unicode.GetString(in_message.MessageBody).Split(':')[0];
//得到工程名
projectName = Encoding.Unicode.GetString(in_message.MessageBody).Split(':')[1];
//得到文档名
documentName = Encoding.Unicode.GetString(in_message.MessageBody).Split(':')[2];
//判断当前目录下是否存在该项目及工程文件夹
if (!Directory.Exists(solutionName + "\\" + projectName))
{
//在当前路径下创建项目文件夹
Directory.CreateDirectory(solutionName + "\\" + projectName);
}
//生成时间戳 格式：__2011_12_2_19_48_37
string timestamp = "__"+DateTime.Now.ToString().Replace(':', '_').Replace(' ', '_').Replace('/', '_');
documentNameAddTimestampWithRelativePath = solutionName + "\\" + projectName + "\\" + documentName + timestamp;
if (!File.Exists(documentNameAddTimestampWithRelativePath))
{
fs = File.Create(documentNameAddTimestampWithRelativePath);
fs.Close();
}
break;
case Message.MessageFlagHeader.FileMiddle:
if (!File.Exists(documentNameAddTimestampWithRelativePath))
{
Console.WriteLine(documentNameAddTimestampWithRelativePath + " not exist.");
}
using (fs = File.Open(documentNameAddTimestampWithRelativePath, FileMode.Append))
{
BinaryWriter bw = new BinaryWriter(fs);
bw.Write(in_message.MessageBody,0,in_message.MessageBody.Length);
bw.Close();
fs.Close();
}
break;
case Message.MessageFlagHeader.FileEnd:
//发送成功接收的确认数据包
out_message = new Message();
out_message.Command = Message.CommandHeader.ReceivedDocument;
out_message.MessageBody = Encoding.Unicode.GetBytes("yes");
dataStream.Write(out_message.ToBytes(),0,out_message.MessageLength);
return true;
default:
//发送接收失败的确认数据包
out_message = new Message();
out_message.Command = Message.CommandHeader.ReceivedDocument;
out_message.MessageBody = Encoding.Unicode.GetBytes("no");
dataStream.Write(out_message.ToBytes(), 0, out_message.MessageLength);
Console.WriteLine("The document file trasfer failed...");
//若存在接受不全的文件，删除
if (File.Exists(documentNameAddTimestampWithRelativePath))
{
File.Delete(documentNameAddTimestampWithRelativePath);
}
break;
}
} while (in_message.MessageFlag == Message.MessageFlagHeader.FileBegin
|| in_message.MessageFlag == Message.MessageFlagHeader.FileMiddle);
}
catch (Exception ex)
{
Console.WriteLine("GetDocument: "+ex.Message);
}
return false;
}
/// <summary>
/// 断开服务器
/// </summary>
/// <param name="hostname"></param>
/// <param name="port"></param>
/// <returns></returns>
public Boolean DisConnectServer()
{
try
{
Console.WriteLine("disconnect with the server {0}...", this.client.Client.RemoteEndPoint);
//枚举当前程序的所有线程
//ProcessThreadCollection ptCollection = Process.GetCurrentProcess().Threads;
//foreach (ProcessThread pt in ptCollection)
//{
// Console.WriteLine("ID:{0},State:{1},Priority:{2}", pt.Id, pt.ThreadState, pt.PriorityLevel);
//}
//向服务器发送一个offline数据包
Message out_message = new Message();
out_message.Command = Message.CommandHeader.Offline;
out_message.MessageBody = Encoding.Unicode.GetBytes("offline");
//获得输出流
dataStream = client.GetStream();
while (!dataStream.CanWrite)
{
System.Threading.Thread.Sleep(1000);//sleep 1s
}
//将打包成message的信息写入输出流
dataStream.Write(out_message.ToBytes(), 0, out_message.MessageLength);
this.client.Close();
Console.WriteLine("the client is offline...");
}
catch (Exception ex)
{
Console.WriteLine("DisConnectServer: "+ex.Message);
return false;
}
return true;
}
public static Message Parse(NetworkStream connection)
{
Message message = new Message();
byte[] buffer = new byte[4];
//先读出前4个字节，即Message的长度
if (ReadMessagePartial(connection, buffer, 4))
{
message.MessageLength = BitConverter.ToInt32(buffer, 0);
}
if (message.MessageLength > 3)
{
buffer = new byte[message.MessageLength - 4];
//读出消息的其它字节
if (ReadMessagePartial(connection, buffer, message.MessageLength - 4))
{
//读出一个字节的文件信息标志
message.MessageFlag = (MessageFlagHeader)buffer[0];
//读出数据包序号,4个字节
message.FilePacketNumber = BitConverter.ToInt32(buffer, 1);
//读出命令
message.Command = (CommandHeader)buffer[5];
//读出消息体
int i = 0;
message.MessageBody = new byte[message.MessageLength - 10];
while (i < message.MessageLength - 10)
{
message.MessageBody[i] = buffer[6 + i];
i++;
}
}
}
return message;
}
