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

目录

权限管理系统系列之序言  

权限管理系统系列之WCF通信

       之前写了两篇关于权限管理系统的博客了，由于这段时间有事比较忙就暂停了，今天继续编写权限管理系统之登陆和升级模块，登陆和升级也是每个系统之必须的模块，对于一个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哈哈~

登陆模块

用户表结构：

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

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

 

        #region 按钮
        //登录
        private void logButton_Click(object sender, EventArgs e)
        {
            if (locked)
            {
                #region 解锁
                try
                {
                    if (pswTextBox.Text == context.Password)
                    {
                        this.Hide();
                        context.ParentForm.Enabled = true;
                        context.ParentForm.Show();
                        Form[] childrens = context.ParentForm.OwnedForms;
                        if (childrens != null)
                        {
                            foreach (Form item in childrens)
                            {
                                if (item.IsDisposed)
                                    continue;
                                if (!item.IsMdiChild && !"系统已锁定".Equals(item.Text))
                                {
                                    item.Show();
                                }
                            }
                        }
                        context.ParentForm.Activate();
                        this.DialogResult = DialogResult.OK;
                    }
                    else
                    {
                        Comm.MessageBox.Info("密码错误!");
                        pswTextBox.Text = ""; ;
                        pswTextBox.Focus();
                        return;
                    }
                }
                catch (Exception exMsg)
                {
                    WriteLogAndShowMessageBox.Error("解锁出错:" + exMsg.Message, "解锁出错:" + exMsg.ToString());
                    pswTextBox.Text = ""; ;
                    pswTextBox.Focus();
                }
                #endregion 解锁
            }
            else
            {
                #region 登录
                try
                {
                    if (string.IsNullOrWhiteSpace(nameTextBox.Text))
                    {
                        Comm.MessageBox.Info("请输入账号");
                        nameTextBox.Focus();
                        return;
                    }
                    if (string.IsNullOrWhiteSpace(pswTextBox.Text))
                    {
                        Comm.MessageBox.Info("请输入密码");
                        pswTextBox.Focus();
                        return;
                    }
                    string name = nameTextBox.Text;
                    string password = pswTextBox.Text;
                    string despsw = Encrypt.EncryptDES(password, Const.EncryptKey);

                    Result result = LoginData.Login(name, despsw);
                    if (!result.success)
                    {
                        Comm.MessageBox.Info(result.msg);
                        if (result.msg.Contains("账号不存在"))
                        {
                            nameTextBox.Text = "";
                            nameTextBox.Focus();
                        }
                        else if (result.msg.Contains("密码错误"))
                        {
                            pswTextBox.Text = "";
                            pswTextBox.Focus();
                        }
                        else
                        {
                            nameTextBox.Text = "";
                            pswTextBox.Text = "";
                            nameTextBox.Focus();
                        }
                        return;
                    }
                    else
                    {
                        t_fw_user user = new t_fw_user();
                        user.s_usercode = name;
                        user.s_password = password;
                        string resData = JSON.Json2Object<string>(result.data);
                        user.s_username = resData;
                        //user.Mode = int.Parse(resData[1]);
                        //user.UserCode2 = resData[2];
                        this.Hide();
                        MainForm mainform = new MainForm(user);
                        mainform.Show();
                        mainform.Text = systemName;
                        nameTextBox.Text = "";
                        pswTextBox.Text = "";
                        Log.Info(name + " 用户登录成功!");
                        Tools.WriteOptLogToDb(Tools.OPType.Login, "", "用户:" + name + "登录", "用户登录");
                    }
                }
                catch (Exception exMsg)
                {
                    WriteLogAndShowMessageBox.Error("登录出错:" + exMsg.Message, "登录出错:" + exMsg.ToString());
                    nameTextBox.Text = "";
                    pswTextBox.Text = "";
                    nameTextBox.Focus();
                }
                #endregion 登录
            }
        }

 

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

    /// <summary>
    /// Description of Encrypt.
    /// </summary>
    public static class Encrypt
    {
        //默认密钥向量
        private static byte[] Keys = { 0x12, 0x34, 0x56, 0x78, 0x90, 0xAB, 0xCD, 0xEF };
        /// <summary>
        /// DES加密字符串
        /// </summary>
        /// <param name="encryptString">待加密的字符串</param>
        /// <param name="encryptKey">加密密钥,要求为8位</param>
        /// <returns>加密成功返回加密后的字符串，失败返回源串</returns>
        public static string EncryptDES(string encryptString, string encryptKey)
        {
            if (!string.IsNullOrEmpty(encryptString))
            {
                try
                {
                    byte[] rgbKey = Encoding.UTF8.GetBytes(encryptKey.Substring(0, 8));
                    byte[] rgbIV = Keys;
                    byte[] inputByteArray = Encoding.UTF8.GetBytes(encryptString);
                    DESCryptoServiceProvider dCSP = new DESCryptoServiceProvider();
                    MemoryStream mStream = new MemoryStream();
                    CryptoStream cStream = new CryptoStream(mStream, dCSP.CreateEncryptor(rgbKey, rgbIV), CryptoStreamMode.Write);
                    cStream.Write(inputByteArray, 0, inputByteArray.Length);
                    cStream.FlushFinalBlock();
                    return Convert.ToBase64String(mStream.ToArray());
                }
                catch
                {
                    return encryptString;
                }
            }
            else
            {
                return string.Empty;
            }
        }
        
        /// <summary>
        /// DES解密字符串
        /// </summary>
        /// <param name="decryptString">待解密的字符串</param>
        /// <param name="decryptKey">解密密钥,要求为8位,和加密密钥相同</param>
        /// <returns>解密成功返回解密后的字符串，失败返源串</returns>
        public static string DecryptDES(string decryptString, string decryptKey)
        {
            try
            {
                byte[] rgbKey = Encoding.UTF8.GetBytes(decryptKey);
                byte[] rgbIV = Keys;
                byte[] inputByteArray = Convert.FromBase64String(decryptString);
                DESCryptoServiceProvider DCSP = new DESCryptoServiceProvider();
                MemoryStream mStream = new MemoryStream();
                CryptoStream cStream = new CryptoStream(mStream, DCSP.CreateDecryptor(rgbKey, rgbIV), CryptoStreamMode.Write);
                cStream.Write(inputByteArray, 0, inputByteArray.Length);
                cStream.FlushFinalBlock();
                return Encoding.UTF8.GetString(mStream.ToArray());
            }
            catch
            {
                return decryptString;
            }
        }
    }

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

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