权限管理系统系列之登录、升级模块

之前写了两篇关于权限管理系统的博客了，由于这段时间有事比较忙就暂停了，今天继续编写权限管理系统之登陆和升级模块，登陆和升级也是每个系统之必须的模块，对于一个Winform程序，在登陆之前必须要先进行程序的升级，所以先介绍升级模块。

升级模块

表结构如下：

插入数据表数据如下：

一般程序升级可能有好几种，比如说有根本版本号比较升级、根据DLL生成时间比较等等，而我介绍的是根据版本号进行升级。我们需要在服务端将DLL文件的版本写入到数据表中，这样供每个客户端去比较版本号，就是每次打开客户端之前进行检测版本号，如果版本号相等则不升级，如果不相等则进行升级操作。

服务端实现逻辑如下：

 1         /// <summary>
 2         /// 更新客户端程序集信息
 3         /// </summary>
 4         public static void UpdateAssembleInfo()
 5         {
 6             DbHelper dbhelper = new DbHelper(AppServer.dbName);
 7             string sql = "select * from t_fw_assemble_list t";
 8             DataTable dt = dbhelper.Query(sql);
 9             DirectoryInfo dirInfo = new DirectoryInfo(Application.StartupPath);
10             FileVersionInfo fvi = null;
11             string updateSql = "update t_fw_assemble_list set s_versionno= '{0}',t_timestamp=getdate()  where s_type='{1}' and s_filename='{2}'";
12             string fName = string.Empty;
13             GetAllFiles(dirInfo);
14             bool isAddLog = true;
15             StreamReader sr = null;
16             for (int i = 0; i < dt.Rows.Count; i++)
17             {
18                 fName = dt.Rows[i]["S_FILENAME"].ToString();
19                 if (fileNameDic.ContainsKey(fName))
20                 {
21                     if (dt.Rows[i]["L_UPDATE"].ToString() == "2")
22                     {
23                         sr = new StreamReader(Application.StartupPath + "\" + fName);
24                         string fileContext = sr.ReadToEnd();
25                         sr.Close();
26                         sr.Dispose();
27                         sql = string.Format(updateSql, Common.Util.MD5Encrypt.MD5EncryptDES(fileContext), dt.Rows[i]["S_TYPE"], fName);
28                         dbhelper.ExecuteSql(sql);
29                         isAddLog = false;
30                     }
31                     else
32                     {
33                         fvi = FileVersionInfo.GetVersionInfo(fileNameDic[fName]);
34                         if (!dt.Rows[i]["S_VERSIONNO"].ToString().Equals(fvi.FileVersion))
35                         {
36                             sql = string.Format(updateSql, fvi.FileVersion, dt.Rows[i]["S_TYPE"], fName);
37                             dbhelper.ExecuteSql(sql);
38                             isAddLog = false;
39                         }
40                     }
41                     if (!isAddLog)
42                     {
43                         isAddLog = AddLog();
44                     }
45                 }
46             }
47         }

以上代码主要是写入版本到数据库里，每次服务端启动首先执行这段代码。

服务端搞定我就来看看客户端了，客户端启动时调用以下方法实现：

 1                 bool isDownLoad = false;
 2                 if (args != null)
 3                 {
 4                     if (args.Length > 0)
 5                     {
 6                         for (int i = 0; i < args.Length; i++)
 7                         {
 8                             if (args[i] == "Update")
 9                             {
10                                 isDownLoad = true;
11                                 break;
12                             }
13                         }
14                     }
15                 }
16                 if (isDownLoad)
17                 {
18                     //启动主界面
19                     Application.Run(new LoginForm());
20                 }
21                 else
22                 {
23                     //更新客户端程序集
24                     if (UpdateAssembleData.UpdateAssembleInfo() > 0)
25                     {
26                         DownLoadForm dlf = new DownLoadForm();
27                         dlf.fileNameList = UpdateAssembleData.fileNameList;
28                         //启动下载程序界面
29                         Application.Run(dlf);
30                     }
31                     else
32                     {
33                         //启动主界面
34                         Application.Run(new LoginForm());
35                     }
36                 }

更新时会弹出升级窗体，上面会显示升级的DLL文件，以及文件升级的进度条，升级完成启动新的程序，升级过程的核心代码：

  1         /// <summary>
  2         /// 下载文件
  3         /// </summary>
  4         private void DownLoadFile()
  5         {
  6             if (fileNameList.Count > 0)
  7             {
  8                 //CallService service = new CallService("GetFile");
  9                 int countLen = fileNameList.Count;
 10                 this.pbarDownLoad.Position = this.pbarDownLoad.Properties.Minimum;
 11                 double step = this.pbarDownLoad.Properties.Maximum / countLen;
 12                 string fName = string.Empty;
 13                 string upPath = Application.StartupPath + "\Update\";
 14                 if (!Directory.Exists(upPath))
 15                 {
 16                     Directory.CreateDirectory(upPath);
 17                 }
 18 
 19                 string sql = string.Empty;
 20                 FileStream fs;
 21                 bool isStartUpdate = false;
 22                 List<string> list = new List<string>();
 23                 List<string> getList = new List<string>();
 24                 //int fLen = 0;
 25                 long pageNum = 0;
 26                 for (int i = 0; i < countLen; i++)
 27                 {
 28                     bool isFirstPBLen = true;
 29                     isStartUpdate = false;
 30                     fName = fileNameList[i];
 31                     IAsyncResult iart = this.lblDownLoad.BeginInvoke(new SetLabelText(AsyncLabel), new object[] { "正在下载文件 " + fName });
 32                     this.lblDownLoad.EndInvoke(iart);
 33                     pageNum = 1;
 34                     list.Clear();
 35                     list.Add(fName);
 36                     list.Add(pageNum.ToString ());
 37                     
 38                     //创建服务器下载的文件
 39                     string tmpPath = upPath + fName;
 40                     tmpPath = tmpPath.Substring(0, tmpPath.LastIndexOf('\'));
 41                     if (!Directory .Exists (tmpPath))
 42                     {
 43                         Directory.CreateDirectory(tmpPath);
 44                     }
 45                     fs = new FileStream(upPath + fName, FileMode.Create, FileAccess.Write);
 46 
 47                     while (true)
 48                     {
 49                         Result result = FileData.DoGetFile(list);
 50                         if (!result.success)//DoGetFile
 51                         {
 52                             Comm.WriteLogAndShowMessageBox.Error(result.msg, "Client.Win.DownLoadForm.DownLoadFile()出错:" + result.msg);
 53                             StartUpdateApp(isStartUpdate);
 54                             break;
 55                         }
 56                         else
 57                         {
 58                             getList = JSON.Json2Object<List<string>>(result.data); //service.GetResult<List<string>>();
 59                             byte[] buffer = Convert.FromBase64String(getList[2]);
 60 
 61                             if (isFirstPBLen)
 62                             {
 63                                 //初始化当前文件进度条
 64                                 iart = this.pbarCurrDownLoad.BeginInvoke(new UpdateProcessBar(AsyncIni), new object[] { Convert.ToInt32 (getList[0]), this.pbarCurrDownLoad });
 65                                 this.pbarCurrDownLoad.EndInvoke(iart);
 66                                 isFirstPBLen = false;
 67                                 Thread.Sleep(100);
 68                             }
 69                             
 70                             //接收服务器返回的二制数据
 71                             fs.Write(buffer, 0, buffer.Length);
 72                             pageNum ++;
 73                             list[1] = pageNum.ToString();
 74                             AsyncProcessBar(this.pbarCurrDownLoad, 1);
 75                             if (buffer.Length < Convert.ToInt32(getList[1]))
 76                             {
 77                                 break;
 78                             }
 79                             
 80                         }
 81                     }
 82                     fs.Flush();
 83                     fs.Close();
 84                     //插入日志记录到服务器
 85 
 86                     Tools.WriteOptLogToDb(Tools.OPType.Update, "", fName, "从服务器更新文件" + fName);
 87                     //刷新进度条
 88                     if (this.pbarDownLoad.Position < this.pbarDownLoad.Properties.Maximum)
 89                     {
 90                         if (i == countLen - 1)
 91                         {
 92                             AsyncProcessBar(this.pbarDownLoad, step + 1);
 93                         }
 94                         else
 95                         {
 96                             AsyncProcessBar(this.pbarDownLoad, step);
 97                         }
 98                     }
 99                     isStartUpdate = true;
100                 }
101                 StartUpdateApp(isStartUpdate);
102             }
103         }

启动本地程序：

 1         /// <summary>
 2         /// 启动更新程序并退出本程序
 3         /// </summary>
 4         private void StartUpdateApp(bool isStartUpdate)
 5         {
 6             if (isStartUpdate)
 7             {
 8                 string filePath = Application.StartupPath + "\Update\Update.exe";
 9                 if (File.Exists (filePath))
10                 {
11                     File.Copy(filePath, Application.StartupPath + "\Update.exe" , true);
12                     File.Delete(filePath);
13                 }
14 
15                 Process.Start(Application.StartupPath + "\Update.exe");
16             }
17             //Environment.Exit(0);
18             Application.Exit();
19             Process.GetCurrentProcess().Kill();
20         }

以上基本上可以实现对程序的升级了。O(∩_∩)O哈哈~

登陆模块

用户表结构：

升级完成后就该启动登陆模块。登陆界面如下：

相对而言登陆界面就超级简单了，用户名和密码两个文本框，两个按钮一个是登陆和一个取消按钮，最下面显示版权。

  1         #region 按钮
  2         //登录
  3         private void logButton_Click(object sender, EventArgs e)
  4         {
  5             if (locked)
  6             {
  7                 #region 解锁
  8                 try
  9                 {
 10                     if (pswTextBox.Text == context.Password)
 11                     {
 12                         this.Hide();
 13                         context.ParentForm.Enabled = true;
 14                         context.ParentForm.Show();
 15                         Form[] childrens = context.ParentForm.OwnedForms;
 16                         if (childrens != null)
 17                         {
 18                             foreach (Form item in childrens)
 19                             {
 20                                 if (item.IsDisposed)
 21                                     continue;
 22                                 if (!item.IsMdiChild && !"系统已锁定".Equals(item.Text))
 23                                 {
 24                                     item.Show();
 25                                 }
 26                             }
 27                         }
 28                         context.ParentForm.Activate();
 29                         this.DialogResult = DialogResult.OK;
 30                     }
 31                     else
 32                     {
 33                         Comm.MessageBox.Info("密码错误!");
 34                         pswTextBox.Text = ""; ;
 35                         pswTextBox.Focus();
 36                         return;
 37                     }
 38                 }
 39                 catch (Exception exMsg)
 40                 {
 41                     WriteLogAndShowMessageBox.Error("解锁出错:" + exMsg.Message, "解锁出错:" + exMsg.ToString());
 42                     pswTextBox.Text = ""; ;
 43                     pswTextBox.Focus();
 44                 }
 45                 #endregion 解锁
 46             }
 47             else
 48             {
 49                 #region 登录
 50                 try
 51                 {
 52                     if (string.IsNullOrWhiteSpace(nameTextBox.Text))
 53                     {
 54                         Comm.MessageBox.Info("请输入账号");
 55                         nameTextBox.Focus();
 56                         return;
 57                     }
 58                     if (string.IsNullOrWhiteSpace(pswTextBox.Text))
 59                     {
 60                         Comm.MessageBox.Info("请输入密码");
 61                         pswTextBox.Focus();
 62                         return;
 63                     }
 64                     string name = nameTextBox.Text;
 65                     string password = pswTextBox.Text;
 66                     string despsw = Encrypt.EncryptDES(password, Const.EncryptKey);
 67 
 68                     Result result = LoginData.Login(name, despsw);
 69                     if (!result.success)
 70                     {
 71                         Comm.MessageBox.Info(result.msg);
 72                         if (result.msg.Contains("账号不存在"))
 73                         {
 74                             nameTextBox.Text = "";
 75                             nameTextBox.Focus();
 76                         }
 77                         else if (result.msg.Contains("密码错误"))
 78                         {
 79                             pswTextBox.Text = "";
 80                             pswTextBox.Focus();
 81                         }
 82                         else
 83                         {
 84                             nameTextBox.Text = "";
 85                             pswTextBox.Text = "";
 86                             nameTextBox.Focus();
 87                         }
 88                         return;
 89                     }
 90                     else
 91                     {
 92                         t_fw_user user = new t_fw_user();
 93                         user.s_usercode = name;
 94                         user.s_password = password;
 95                         string resData = JSON.Json2Object<string>(result.data);
 96                         user.s_username = resData;
 97                         //user.Mode = int.Parse(resData[1]);
 98                         //user.UserCode2 = resData[2];
 99                         this.Hide();
100                         MainForm mainform = new MainForm(user);
101                         mainform.Show();
102                         mainform.Text = systemName;
103                         nameTextBox.Text = "";
104                         pswTextBox.Text = "";
105                         Log.Info(name + " 用户登录成功!");
106                         Tools.WriteOptLogToDb(Tools.OPType.Login, "", "用户:" + name + "登录", "用户登录");
107                     }
108                 }
109                 catch (Exception exMsg)
110                 {
111                     WriteLogAndShowMessageBox.Error("登录出错:" + exMsg.Message, "登录出错:" + exMsg.ToString());
112                     nameTextBox.Text = "";
113                     pswTextBox.Text = "";
114                     nameTextBox.Focus();
115                 }
116                 #endregion 登录
117             }
118         }

以上即可实现用户的登陆，密码采用DES加密，DES加密和解密方法如下：

 1     /// <summary>
 2     /// Description of Encrypt.
 3     /// </summary>
 4     public static class Encrypt
 5     {
 6         //默认密钥向量
 7         private static byte[] Keys = { 0x12, 0x34, 0x56, 0x78, 0x90, 0xAB, 0xCD, 0xEF };
 8         /// <summary>
 9         /// DES加密字符串
10         /// </summary>
11         /// <param name="encryptString">待加密的字符串</param>
12         /// <param name="encryptKey">加密密钥,要求为8位</param>
13         /// <returns>加密成功返回加密后的字符串，失败返回源串</returns>
14         public static string EncryptDES(string encryptString, string encryptKey)
15         {
16             if (!string.IsNullOrEmpty(encryptString))
17             {
18                 try
19                 {
20                     byte[] rgbKey = Encoding.UTF8.GetBytes(encryptKey.Substring(0, 8));
21                     byte[] rgbIV = Keys;
22                     byte[] inputByteArray = Encoding.UTF8.GetBytes(encryptString);
23                     DESCryptoServiceProvider dCSP = new DESCryptoServiceProvider();
24                     MemoryStream mStream = new MemoryStream();
25                     CryptoStream cStream = new CryptoStream(mStream, dCSP.CreateEncryptor(rgbKey, rgbIV), CryptoStreamMode.Write);
26                     cStream.Write(inputByteArray, 0, inputByteArray.Length);
27                     cStream.FlushFinalBlock();
28                     return Convert.ToBase64String(mStream.ToArray());
29                 }
30                 catch
31                 {
32                     return encryptString;
33                 }
34             }
35             else
36             {
37                 return string.Empty;
38             }
39         }
40         
41         /// <summary>
42         /// DES解密字符串
43         /// </summary>
44         /// <param name="decryptString">待解密的字符串</param>
45         /// <param name="decryptKey">解密密钥,要求为8位,和加密密钥相同</param>
46         /// <returns>解密成功返回解密后的字符串，失败返源串</returns>
47         public static string DecryptDES(string decryptString, string decryptKey)
48         {
49             try
50             {
51                 byte[] rgbKey = Encoding.UTF8.GetBytes(decryptKey);
52                 byte[] rgbIV = Keys;
53                 byte[] inputByteArray = Convert.FromBase64String(decryptString);
54                 DESCryptoServiceProvider DCSP = new DESCryptoServiceProvider();
55                 MemoryStream mStream = new MemoryStream();
56                 CryptoStream cStream = new CryptoStream(mStream, DCSP.CreateDecryptor(rgbKey, rgbIV), CryptoStreamMode.Write);
57                 cStream.Write(inputByteArray, 0, inputByteArray.Length);
58                 cStream.FlushFinalBlock();
59                 return Encoding.UTF8.GetString(mStream.ToArray());
60             }
61             catch
62             {
63                 return decryptString;
64             }
65         }
66     }

现在应该所有的程序密码都进行了加密了吧！之前好像CSDN把所有的密码都泄露出来了，好像那时还是明文的，这好像还是去年的事情吧！

以上简单介绍了升级和登陆模块，也许做好这块的功能是一个项目起到至关重要的作用。