[VB.NET] โปรแกรมการจัดเก็บข้อมูลรหัสผ่าน CRUD Full Version

thongkorn

โค้ดชุดนี้ก็ต่อมาจากในตอนที่แล้ว โค้ดการเก็บข้อมูลรหัสผ่านด้วยการเข้ารหัสแบบ Long Binary (CRUD) โดยใช้ไฟล์ฐานข้อมูล MS Access ในการจัดเก็บข้อมูล ... หากไม่นับโค้ดใน การเข้ารหัส/ถอดรหัสด้วย Rijndael (AES) Algorithm มันเป็นการเขียนโปรแกรมแบบ CRUD (Create / Read / Update / Delete) ขั้นพื้นฐานสำหรับ Windows Form ในสายภาษา VB.NET หรือ C# เพราะเป็นการติดต่อกับตารางข้อมูล (Table) เดี่ยวๆ โดยไม่มีการเชื่อมความสัมพันธ์ใดๆกันระหว่างตาราง ...

สำหรับ Full Version นี้ แอดมินเพิ่มเติมความปลอดภัยในการใช้โปรแกรมเพิ่มเข้ามา คือจะต้องใส่รหัสผ่านก่อน (ค่า Default คือ admin) และสามารถแก้ไขรหัสผ่านได้ด้วย ...

มาดูโค้ดต้นฉบับกันเถอะ ฟอร์มหลัก (frmPassCodeFull.vb)

1. Imports System.Data.OleDb
   
2. Imports MaterialSkin
   
3. 
   
4. Public Class frmPassCodeFull
   
5.     '// Add new or Edit data.
   
6.     Dim blnNewData As Boolean = False   '// Edit mode.
   
7.     Dim PK As Long  '// เก็บค่า Primary Key ทั้งการเพิ่มและแก้ไขข้อมูล
   
8. 
   
9.     ' / ------------------------------------------------------------------------------------------------
   
10.     ' / S T A R T ... H E R E
    
11.     ' / ------------------------------------------------------------------------------------------------
    
12.     Private Sub frmPassCodeFull_Load(sender As System.Object, e As System.EventArgs) Handles MyBase.Load
    
13.         '// Login เข้าสู่ระบบก่อน
    
14.         Call frmLogin.ShowDialog()
    
15.         '// blnLogin ถูกประกาศไว้ใน modDataBase.vb
    
16.         If Not blnLogin Then Me.Close()
    
17. 
    
18.         '// Initialize MaterialSkin .Net Framework 4.0
    
19.         Dim SkinManager As MaterialSkinManager = MaterialSkinManager.Instance
    
20.         SkinManager.AddFormToManage(Me)
    
21.         SkinManager.Theme = MaterialSkinManager.Themes.LIGHT
    
22.         '//SkinManager.ColorScheme = New MaterialSkin.ColorScheme(Primary.Amber500, Primary.BlueGrey900, Primary.BlueGrey500, Accent.LightBlue200, TextShade.WHITE)
    
23.         'SkinManager.ColorScheme = New MaterialSkin.ColorScheme(Primary.Green600, Primary.Green700, Primary.Green200, Accent.Red100, TextShade.WHITE)
    
24.         '//
    
25.         With cmbColorTheme
    
26.             .Items.Add("Default")
    
27.             .Items.Add("Orange")
    
28.             .Items.Add("Green")
    
29.             .Items.Add("Light Blue")
    
30.             .Items.Add("Cyan")
    
31.             .Items.Add("Gray")
    
32.         End With
    
33.         cmbColorTheme.SelectedIndex = 2
    
34. 
    
35.         '// ตั้งค่าขนาดของฟอร์มแบบ Run Time
    
36.         Me.Size = New Size(1020, 720)
    
37.         '// คำนวณตำแหน่งที่กึ่งกลางของหน้าจอ
    
38.         Me.Location = New Point((Screen.PrimaryScreen.WorkingArea.Width - Me.Width) \ 2,
    
39.         (Screen.PrimaryScreen.WorkingArea.Height - Me.Height) \ 2)
    
40. 
    
41.         '// โหลดข้อมูลลงตารางกริด
    
42.         dgvData.DataSource = RetrieveData()
    
43.         Call SetupDataGridView(dgvData)
    
44.         lblRecordCount.Text = "[Total: " & dgvData.Rows.Count & " Record.]"
    
45.         '// เริ่มโหมดว่างเปล่าเพื่อทำรายการใหม่ หรือเลือกรายการขึ้นมาแก้ไข
    
46.         Call NewMode()
    
47.     End Sub
    
48. 
    
49.     ' / ------------------------------------------------------------------------------------------------
    
50.     ' / Add New Mode
    
51.     ' / ------------------------------------------------------------------------------------------------
    
52.     Private Sub NewMode()
    
53.         '// Clear all TextBox.
    
54.         For Each c In GroupBox1.Controls
    
55.             If TypeOf c Is TextBox Then
    
56.                 DirectCast(c, TextBox).Clear()
    
57.                 DirectCast(c, TextBox).Enabled = False
    
58.             End If
    
59.             If TypeOf c Is DateTimePicker Then DirectCast(c, DateTimePicker).Enabled = False
    
60.         Next
    
61.         itemAdd.Enabled = True
    
62.         itemSave.Enabled = False
    
63.         itemDelete.Enabled = True
    
64.         itemDelete.Text = "DELETE DATA"
    
65.     End Sub
    
66. 
    
67.     ' / ------------------------------------------------------------------------------------------------
    
68.     ' / Edit Data Mode
    
69.     ' / ------------------------------------------------------------------------------------------------
    
70.     Private Sub EditMode()
    
71.         '// Clear all TextBox
    
72.         For Each c In GroupBox1.Controls
    
73.             If TypeOf c Is TextBox Then
    
74.                 DirectCast(c, TextBox).Enabled = True
    
75.             End If
    
76.             If TypeOf c Is DateTimePicker Then DirectCast(c, DateTimePicker).Enabled = True
    
77.         Next
    
78.         itemAdd.Enabled = False
    
79.         itemSave.Enabled = True
    
80.         itemDelete.Enabled = True
    
81.         itemDelete.Text = "CANCEL"
    
82.     End Sub
    
83. 
    
84.     ' / ------------------------------------------------------------------------------------------------
    
85.     ' / Add new data.
    
86.     ' / ------------------------------------------------------------------------------------------------
    
87.     Private Sub itemAdd_Click(sender As System.Object, e As System.EventArgs) Handles itemAdd.Click
    
88.         blnNewData = True
    
89.         Call EditMode()
    
90.         txtUrl.Focus()
    
91.     End Sub
    
92. 
    
93.     ' / ------------------------------------------------------------------------------------------------
    
94.     ' / Save Data.
    
95.     ' / ------------------------------------------------------------------------------------------------
    
96.     Private Sub itemSave_Click(sender As System.Object, e As System.EventArgs) Handles itemSave.Click
    
97.         If txtUrl.Text = "" Or txtUrl.Text.Trim.Length = 0 Then
    
98.             MessageBox.Show("Please enter Url or Name.", "Report Status", MessageBoxButtons.OK, MessageBoxIcon.Warning)
    
99.             txtUrl.Focus()
    
100.             Return
     
101.         End If
     
102.         If txtLogin.Text = "" Or txtLogin.Text.Trim.Length = 0 Then
     
103.             MessageBox.Show("Please enter Login Name.", "Report Status", MessageBoxButtons.OK, MessageBoxIcon.Warning)
     
104.             txtLogin.Focus()
     
105.             Return
     
106.         End If
     
107.         If txtPassword.Text = "" Or txtPassword.Text.Trim.Length = 0 Then
     
108.             MessageBox.Show("Please enter Password.", "Report Status", MessageBoxButtons.OK, MessageBoxIcon.Warning)
     
109.             txtPassword.Focus()
     
110.             Return
     
111.         End If
     
112.         '// Details Class in modDataBase.vb
     
113.         '// กำหนดค่าให้กับตัวแปร เพื่อส่งออกไปทำการบันทึกข้อมูล
     
114.         Dim result As New Details
     
115.         With result
     
116.             .PK = PK
     
117.             .Login = txtLogin.Text
     
118.             .Encrypted = txtPassword.Text
     
119.             .Url = txtUrl.Text
     
120.             .Email = txtEmail.Text
     
121.             .Phone = txtPhone.Text
     
122.             .DateModified = dtpDateModified.Value
     
123.         End With
     
124.         '// blnNewData = True คือการเพิ่มข้อมูล
     
125.         If blnNewData Then
     
126.             '// หาค่า Primary Key จาก SetupNewPK อยู่ใน modDataBase.vb
     
127.             result.PK = SetupNewPK("SELECT MAX(UserData.PK) AS MaxPK FROM UserData")
     
128.             '// ส่งค่า blnNewData = True บอกว่าเป็นการเพิ่มข้อมูลใหม่ และค่าของ Control บนฟอร์มไปด้วย
     
129.             Call SaveData(blnNewData, result)
     
130.             Call NewMode()
     
131. 
     
132.             '// การแก้ไขข้อมูล
     
133.         Else
     
134.             'blnNewData = False
     
135.             '// ส่งค่า blnNewData = False บอกว่าเป็นการแก้ไขข้อมูล และส่งค่าของ Control บนฟอร์มไปด้วย
     
136.             Call SaveData(blnNewData, result)
     
137.             Call NewMode()
     
138.         End If
     
139.         '// Refresh data.
     
140.         Call btnRefresh_Click(sender, e)
     
141.     End Sub
     
142. 
     
143.     ' / ------------------------------------------------------------------------------------------------
     
144.     ' / DELETE DATA
     
145.     ' / ------------------------------------------------------------------------------------------------
     
146.     Private Sub itemDelete_Click(sender As System.Object, e As System.EventArgs) Handles itemDelete.Click
     
147.         If itemDelete.Text = "DELETE DATA" Then
     
148.             '// เช็คก่อนว่ามีรายแถวอยู่หรือไม่ หรือมีการคลิ๊กเลือกแถวรายการหรือเปล่า
     
149.             If dgvData.RowCount = 0 Or dgvData.CurrentRow Is Nothing Then Exit Sub
     
150.             Dim Url As String = dgvData.Item(1, dgvData.CurrentRow.Index).Value
     
151.             '// ถามยืนยันการลบข้อมูล
     
152.             Dim Result As Byte = MessageBox.Show("Are you sure you want to delete the data?" & vbCrLf & "URL/Name: " & Url, "Confirm Deletion", MessageBoxButtons.YesNo, MessageBoxIcon.Question, MessageBoxDefaultButton.Button2)
     
153.             If Result = DialogResult.Yes Then
     
154.                 '// Receive Primary Key value to confirm the deletion.
     
155.                 PK = dgvData.Item(0, dgvData.CurrentRow.Index).Value
     
156.                 '// ลบข้อมูล (modDataBase.vb)
     
157.                 Call DeleteData(PK)
     
158.                 '// ทำการแสดงผลรายการใหม่
     
159.                 Call btnRefresh_Click(sender, e)
     
160.                 Call NewMode()
     
161.             End If
     
162.         ElseIf itemDelete.Text = "CANCEL" Then
     
163.             '// เริ่มโหมดว่างเปล่าเพื่อทำรายการใหม่ หรือเลือกรายการขึ้นมาแก้ไข
     
164.             Call NewMode()
     
165.         End If
     
166.     End Sub
     
167. 
     
168.     ' / ------------------------------------------------------------------------------------------------
     
169.     ' / แสดงผลข้อมูลทั้งหมด
     
170.     ' / ------------------------------------------------------------------------------------------------
     
171.     Private Sub btnRefresh_Click(sender As System.Object, e As System.EventArgs) Handles btnRefresh.Click
     
172.         dgvData.DataSource = RetrieveData()
     
173.         lblRecordCount.Text = "[Total: " & dgvData.Rows.Count & " Record.]"
     
174.     End Sub
     
175. 
     
176.     ' / ------------------------------------------------------------------------------------------------
     
177.     ' / การค้นหาข้อมูล
     
178.     ' / ------------------------------------------------------------------------------------------------
     
179.     Private Sub txtSearch_KeyPress(sender As Object, e As System.Windows.Forms.KeyPressEventArgs) Handles txtSearch.KeyPress
     
180.         '// Undesirable characters for the database ex.  ', * or %
     
181.         txtSearch.Text = txtSearch.Text.Replace("'", "").Replace("*", "").Replace("%", "")
     
182.         If Trim(txtSearch.Text) = "" Or Len(Trim(txtSearch.Text)) = 0 Then Exit Sub
     
183.         '// RetrieveData(True) It means searching for information.
     
184.         If e.KeyChar = Chr(13) Then '// Press Enter
     
185.             '// No beep.
     
186.             e.Handled = True
     
187.             '//
     
188.             dgvData.DataSource = RetrieveData(True, txtSearch.Text)
     
189.             lblRecordCount.Text = "[Total: " & dgvData.Rows.Count & " Record.]"
     
190.             txtSearch.Clear()
     
191.         End If
     
192.     End Sub
     
193. 
     
194.     ' / ------------------------------------------------------------------------------------------------
     
195.     ' / ดับเบิ้ลคิ๊กเมาส์ในตารางกริด จากนั้นค้นหา Primary Key เพื่อนำข้อมูลไปแสดงผล
     
196.     ' / ------------------------------------------------------------------------------------------------
     
197.     Private Sub dgvData_CellDoubleClick(sender As Object, e As System.Windows.Forms.DataGridViewCellEventArgs) Handles dgvData.CellDoubleClick
     
198.         If e.RowIndex >= 0 Then
     
199.             '// เก็บค่านี้เอาไว้เพื่อใช้ในการอ้างอิงตอนที่ทำการอัพเดตข้อมูล
     
200.             PK = CInt(dgvData.Rows(e.RowIndex).Cells("PK").Value)
     
201.             blnNewData = False    '// กำหนดให้เป็นการแก้ไขข้อมูล
     
202.             '// ส่งค่า PK เพื่อเป็นคีย์ไปทำการค้นข้อมูล แล้วคืนข้อมูลกลับมาในรูปแบบ Class
     
203.             '// Class Details อยู่ใน modDataBase.vb
     
204.             Dim result As Details = RetrieveDetails(PK)
     
205.             txtLogin.Text = result.Login
     
206.             txtPassword.Text = result.Encrypted
     
207.             txtUrl.Text = result.Url
     
208.             txtEmail.Text = result.Email
     
209.             txtPhone.Text = result.Phone
     
210.             dtpDateModified.Text = Format(result.DateModified, "dd/MM/yyyy")
     
211.             '// โหมดในการแก้ไข
     
212.             Call EditMode()
     
213.             txtUrl.Focus()
     
214.         End If
     
215.     End Sub
     
216. 
     
217.     '// เปลี่ยนรหัสผ่าน
     
218.     Private Sub btnChangePassword_Click(sender As System.Object, e As System.EventArgs) Handles btnChangePassword.Click
     
219.         blnChangePassword = True
     
220.         frmLogin.ShowDialog()
     
221.     End Sub
     
222. 
     
223.     '// Function to handle the KeyPress event for TextBox controls.
     
224.     Private Sub TextBox_KeyPress(sender As Object, e As KeyPressEventArgs) Handles txtLogin.KeyPress, txtPassword.KeyPress, txtUrl.KeyPress, txtEmail.KeyPress, txtPhone.KeyPress
     
225.         If e.KeyChar = ChrW(Keys.Enter) Then
     
226.             e.Handled = True '// Prevent the Enter key from adding a new line in the TextBox.
     
227.             SendKeys.Send("{TAB}") '// Send the TAB key to move to the next control.
     
228.         End If
     
229.     End Sub
     
230. 
     
231.     Private Sub dtpDateModified_KeyDown(sender As Object, e As System.Windows.Forms.KeyEventArgs) Handles dtpDateModified.KeyDown
     
232.         If e.KeyCode = Keys.Enter Then
     
233.             e.Handled = True
     
234.             SendKeys.Send("{TAB}")
     
235.         End If
     
236.     End Sub
     
237. 
     
238.     '// Select Color Theme
     
239.     Private Sub cmbColorTheme_SelectedIndexChanged(sender As System.Object, e As System.EventArgs) Handles cmbColorTheme.SelectedIndexChanged
     
240.         Select Case cmbColorTheme.SelectedIndex
     
241.             Case 0
     
242.                 SkinManager.ColorScheme = New MaterialSkin.ColorScheme(Primary.BlueGrey800, Primary.BlueGrey900, Primary.BlueGrey500, Accent.LightBlue200, TextShade.WHITE)
     
243.             Case 1
     
244.                 SkinManager.ColorScheme = New MaterialSkin.ColorScheme(Primary.Amber500, Primary.BlueGrey900, Primary.BlueGrey500, Accent.LightBlue200, TextShade.WHITE)
     
245.             Case 2
     
246.                 SkinManager.ColorScheme = New MaterialSkin.ColorScheme(Primary.Green600, Primary.Green700, Primary.Green200, Accent.Red100, TextShade.WHITE)
     
247.             Case 3
     
248.                 SkinManager.ColorScheme = New MaterialSkin.ColorScheme(Primary.LightBlue600, Primary.LightBlue700, Primary.Green200, Accent.LightGreen700, TextShade.WHITE)
     
249.             Case 4
     
250.                 SkinManager.ColorScheme = New MaterialSkin.ColorScheme(Primary.Cyan500, Primary.Cyan700, Primary.Cyan100, Accent.Blue100, TextShade.WHITE)
     
251.             Case 5
     
252.                 SkinManager.ColorScheme = New MaterialSkin.ColorScheme(Primary.Grey600, Primary.Grey700, Primary.Grey100, Accent.LightGreen200, TextShade.WHITE)
     
253.         End Select
     
254.     End Sub
     
255. 
     
256.     Private Sub frmPassCodeFull_FormClosed(sender As Object, e As System.Windows.Forms.FormClosedEventArgs) Handles Me.FormClosed
     
257.         Me.Dispose()
     
258.         GC.SuppressFinalize(Me)
     
259.         End
     
260.     End Sub
     
261. 
     
262.     Private Sub itemExit_Click(sender As System.Object, e As System.EventArgs) Handles btnExit.Click
     
263.         Me.Close()
     
264.     End Sub
     
265. 
     
266. End Class

คัดลอกไปที่คลิปบอร์ด

โค้ดในการป้อนรหัสผ่านก่อนเข้าใช้โปรแกรม (frmLogin.vb)

1. Imports MaterialSkin
   
2. 
   
3. Public Class frmLogin
   
4. 
   
5.     ' / ------------------------------------------------------------------------------------------------
   
6.     ' / S T A R T ... H E R E
   
7.     ' / ------------------------------------------------------------------------------------------------
   
8.     Private Sub frmLogin_Load(sender As System.Object, e As System.EventArgs) Handles MyBase.Load
   
9.         '// รหัสผ่านเริ่มต้น
   
10.         'txtPassword.Text = "admin"
    
11.         chkShowPass.Checked = True
    
12.         Me.MaximizeBox = False
    
13.         Me.MinimizeBox = False
    
14.         '// ปรับระยะของฟอร์มใหม่
    
15.         '// การเข้าสู่ระบบ
    
16.         If Not blnChangePassword Then
    
17.             Me.MinimumSize = New Point(413, 355)
    
18.             Me.MaximumSize = New Point(413, 355)
    
19.             btnLogin.Location = New Point(146, 288)
    
20.             btnExit.Location = New Point(217, 288)
    
21.             btnLogin.Text = "Login"
    
22.             lblPassword.Text = "PASSWORD"
    
23.             lblNewPassword.Visible = False
    
24.             txtNewPassword.Visible = False
    
25.             '// การเปลี่ยนรหัสผ่าน
    
26.         Else
    
27.             Me.MinimumSize = New Point(413, 390)
    
28.             Me.MaximumSize = New Point(413, 390)
    
29.             btnLogin.Location = New Point(146, 341)
    
30.             btnExit.Location = New Point(217, 341)
    
31.             btnLogin.Text = "Save"
    
32.             lblPassword.Text = "OLD PASSWORD"
    
33.             lblNewPassword.Visible = True
    
34.             txtNewPassword.Visible = True
    
35.             Me.Text = "CHANGE PASSWORD"
    
36.             lblLogin.Text = "Change Password"
    
37.         End If
    
38.         '//
    
39.     End Sub
    
40. 
    
41.     ' / ------------------------------------------------------------------------------------------------
    
42.     ' / ตรวจสอบรหัสผ่าน
    
43.     ' / ------------------------------------------------------------------------------------------------
    
44.     Private Sub btnLogin_Click(sender As System.Object, e As System.EventArgs) Handles btnLogin.Click
    
45.         '// เป็นการ Login เข้าสู่โปรแกรม
    
46.         If Not blnChangePassword Then
    
47.             '// เรียกไปฟังค์ชั่น CheckPassword ใน modDataBase.vb เพื่อตรวจสอบรหัสผ่านของ Admin
    
48.             If CheckPassword(txtPassword.Text) Then
    
49.                 '// Redirect to admin main form.
    
50.                 blnLogin = True
    
51.                 frmPassCodeFull.Show()
    
52.                 Me.Close()
    
53.             Else
    
54.                 MessageBox.Show("Login failed. Please check your credentials.", "Report Status", MessageBoxButtons.OK, MessageBoxIcon.Warning)
    
55.                 txtPassword.Focus()
    
56.                 Return
    
57.             End If
    
58. 
    
59.             '// เปลี่ยนรหัสผ่าน Admin
    
60.         Else
    
61.             '// เรียกไปฟังค์ชั่น CheckPassword ใน modDataBase.vb เพื่อตรวจสอบรหัสผ่านของ Admin
    
62.             If Not CheckPassword(txtPassword.Text) Then
    
63.                 MessageBox.Show("The old password is incorrect.", "Report Status", MessageBoxButtons.OK, MessageBoxIcon.Error)
    
64.                 txtPassword.Focus()
    
65.                 Return
    
66.             End If
    
67.             '// ยังไม่ได้ป้อนรหัสผ่านใหม่
    
68.             If txtNewPassword.Text.Trim = String.Empty Or txtNewPassword.Text.Trim.Length = 0 Then
    
69.                 MessageBox.Show("Please enter a new password.", "Report Status", MessageBoxButtons.OK, MessageBoxIcon.Error)
    
70.                 txtNewPassword.Focus()
    
71.                 Return
    
72.             End If
    
73.             '// ทำการบันทึกรหัสผ่านใหม่
    
74.             Call SavePassword(txtNewPassword.Text)
    
75.             Me.Close()
    
76.         End If
    
77.     End Sub
    
78. 
    
79.     '// แสดงรหัสผ่าน
    
80.     Private Sub chkShowPass_CheckedChanged(sender As System.Object, e As System.EventArgs) Handles chkShowPass.CheckedChanged
    
81.         If chkShowPass.Checked Then
    
82.             txtPassword.UseSystemPasswordChar = False
    
83.             txtNewPassword.UseSystemPasswordChar = False
    
84.         Else
    
85.             txtPassword.UseSystemPasswordChar = True
    
86.             txtNewPassword.UseSystemPasswordChar = True
    
87.         End If
    
88.         txtPassword.Focus()
    
89.     End Sub
    
90. 
    
91.     '// Function to handle the KeyPress event for TextBox controls.
    
92.     Private Sub TextBox_KeyPress(sender As Object, e As KeyPressEventArgs) Handles txtPassword.KeyPress, txtNewPassword.KeyPress
    
93.         If e.KeyChar = ChrW(Keys.Enter) Then
    
94.             e.Handled = True '// Prevent the Enter key from adding a new line in the TextBox.
    
95.             SendKeys.Send("{TAB}") '// Send the TAB key to move to the next control.
    
96.         End If
    
97.     End Sub
    
98. 
    
99.     Private Sub btnExit_Click(sender As System.Object, e As System.EventArgs) Handles btnExit.Click
    
100.         Me.Close()
     
101.     End Sub
     
102. 
     
103.     Private Sub frmLogin_FormClosed(sender As Object, e As System.Windows.Forms.FormClosedEventArgs) Handles Me.FormClosed
     
104.         Me.Dispose()
     
105.         GC.SuppressFinalize(Me)
     
106.     End Sub
     
107. 
     
108. End Class

คัดลอกไปที่คลิปบอร์ด

โค้ดในส่วนของการติดต่อกับฐานข้อมูล (modDataBase.vb)

1. Imports System.Data.OleDb
   
2. 
   
3. Module modDataBase
   
4.     Dim ConnString As String = "Provider=Microsoft.ACE.OLEDB.12.0;Data Source=" & MyPath(Application.StartupPath) & "\PassCodeBook.accdb;"
   
5. 
   
6.     Public blnLogin As Boolean = False  '// ส่งค่าแจ้งการ Login
   
7.     Public blnChangePassword As Boolean = False '// แจ้งการเปลี่ยนรหัสผ่าน
   
8.     ' / ------------------------------------------------------------------------------------------------
   
9.     '// การใช้ Key และ IV
   
10.     '// Key ควรมีขนาด 16, 24 หรือ 32 bytes เพื่อความปลอดภัย
    
11.     '// IV (Initialization Vector) ควรมีขนาด 16 bytes สำหรับการเข้ารหัส AES
    
12.     Dim key As String = "1234567890123456" '// Key 16 bytes
    
13.     Dim iv As String = "6543210987654321"  '// IV 16 bytes
    
14.     ' / ------------------------------------------------------------------------------------------------
    
15. 
    
16.     Public Class Details
    
17.         '// Auto Implemented Properties for Data Fields.
    
18.         Public Property PK As Integer
    
19.         Public Property Login As String
    
20.         Public Property Encrypted As String
    
21.         Public Property Url As String
    
22.         Public Property Email As String
    
23.         Public Property Phone As String
    
24.         Public Property DateModified As Date
    
25.     End Class
    
26. 
    
27.     ' / ------------------------------------------------------------------------------------------------
    
28.     '// Save data to MS Access database.
    
29.     ' / ------------------------------------------------------------------------------------------------
    
30.     Sub SaveData(newdata As Boolean, details As Details)
    
31.         Dim sql As String
    
32.         Try
    
33.             Using Conn As New OleDbConnection(ConnString)
    
34.                 Conn.Open()
    
35.                 '// การเพิ่มข้อมูลใหม่
    
36.                 If newdata Then
    
37.                     sql = "INSERT INTO UserData (PK, url, Login, Encrypted, email, phone, DateModified) VALUES (@PK, @Url, @Email, @Phone, @Login, @Encrypted, @DateModified)"
    
38. 
    
39.                     '// การแก้ไข
    
40.                 Else
    
41.                     '// Update the record if it exists.
    
42.                     sql = "UPDATE UserData SET PK = @PK, url = @Url, Login = @Login, Encrypted = @Encrypted, email = @Email, Phone = @Phone, DateModified = @DateModified WHERE PK = @PK"
    
43.                 End If
    
44.                 '// Update.
    
45.                 Using Cmd As New OleDbCommand(sql, Conn)
    
46.                     With Cmd.Parameters
    
47.                         .AddWithValue("@PK", details.PK)
    
48.                         .AddWithValue("@url", details.Url)
    
49.                         .AddWithValue("@login", details.Login)
    
50.                         '// การเข้ารหัสด้วย Rijndael (AES) (modEncryption.vb)
    
51.                         .AddWithValue("@Encrypted", EncryptPassword(details.Encrypted, key, iv))
    
52.                         .AddWithValue("@email", details.Email)
    
53.                         .AddWithValue("@phone", details.Phone)
    
54.                         .AddWithValue("@DateModified", Format(details.DateModified, "dd/MM/yyyy"))
    
55.                     End With
    
56.                     Cmd.ExecuteNonQuery()
    
57.                 End Using
    
58.                 MessageBox.Show("UPDATE COMPLETE.", "Report Status", MessageBoxButtons.OK, MessageBoxIcon.Information)
    
59.             End Using
    
60.         Catch ex As Exception
    
61.             MessageBox.Show("UPDATE NOT SUCCESSFUL!", "Report Status", MessageBoxButtons.OK, MessageBoxIcon.Warning)
    
62.         End Try
    
63.     End Sub
    
64. 
    
65.     ' / ------------------------------------------------------------------------------------------------
    
66.     '// Retrieve data from MS Access database and display in DataGridView.
    
67.     '// blnSearch = True คือการค้นหาข้อมูล โดยรับค่า SearchValue เป็นคีย์เวิร์ดใช้ในการค้นหา
    
68.     '// blnSearch = False ก็จะแสดงผลข้อมูลออกมาทั้งหมด
    
69.     '// ทั้ง 2 ค่าเป็น Optional คือหากไม่ระบุการส่งค่ามา ก็ให้มีค่าเป็น False หรือแสดงผลข้อมูลออกมาทั้งหมด
    
70.     ' / ------------------------------------------------------------------------------------------------
    
71.     Function RetrieveData(Optional ByVal blnSearch As Boolean = False, Optional ByVal SearchValue As String = "") As DataTable
    
72.         Using Conn As New OleDbConnection(ConnString)
    
73.             Conn.Open()
    
74.             Dim sql As String = "SELECT PK, url, login, Encrypted, email, phone, DateModified FROM UserData"
    
75.             If blnSearch Then
    
76.                 sql &= " WHERE url LIKE @SearchValue OR email LIKE @SearchValue OR phone LIKE @SearchValue OR login LIKE @SearchValue"
    
77.             End If
    
78.             sql &= " ORDER BY PK"
    
79.             '//
    
80.             Using cmd As New OleDbCommand(sql, Conn)
    
81.                 If blnSearch Then cmd.Parameters.AddWithValue("@SearchValue", "%" & SearchValue & "%")
    
82.                 '//
    
83.                 Using reader As OleDbDataReader = cmd.ExecuteReader()
    
84.                     Dim dt As New DataTable()
    
85.                     With dt.Columns
    
86.                         .Add("PK", GetType(Integer))
    
87.                         .Add("URL", GetType(String))
    
88.                         .Add("Login", GetType(String))
    
89.                         '.Add("Encrypted", GetType(String))
    
90.                         .Add("Email", GetType(String))
    
91.                         .Add("Phone", GetType(String))
    
92.                         .Add("DateModified", GetType(Date))
    
93.                     End With
    
94.                     While reader.Read()
    
95.                         '// มีการเข้ารหัสของรหัสผ่านเอาไว้ตอนแสดงผลในตารางกริด
    
96.                         Dim EncryptedData As Byte() = DirectCast(reader("Encrypted"), Byte())
    
97.                         '// เพิ่มรายการแถวข้อมูลเข้าไปใน DataTable
    
98.                         dt.Rows.Add(reader("PK"), reader("url"), reader("login"), reader("email"), reader("phone"), reader("DateModified"))
    
99.                     End While
    
100.                     Return dt
     
101.                 End Using
     
102.             End Using
     
103.         End Using
     
104.     End Function
     
105. 
     
106.     ' / ------------------------------------------------------------------------------------------------
     
107.     ' / คืนค่าข้อมูลจากตารางกริดไปแสดงผลลงใน Control
     
108.     ' / ------------------------------------------------------------------------------------------------
     
109.     Public Function RetrieveDetails(PK As Integer) As Details
     
110.         '// Class Details เพื่อคืนค่ากลับไปแสดงผลในฟอร์ม
     
111.         Dim result As New Details()
     
112.         Try
     
113.             Using Conn As New OleDbConnection(ConnString)
     
114.                 Conn.Open()
     
115.                 Using cmd As New OleDbCommand("SELECT url, login, Encrypted, email, phone, DateModified FROM UserData WHERE PK = @PK", Conn)
     
116.                     cmd.Parameters.AddWithValue("@PK", PK)
     
117.                     Using reader As OleDbDataReader = cmd.ExecuteReader()
     
118.                         If reader.Read() Then
     
119.                             result.Url = reader("url").ToString()
     
120.                             result.Login = reader("login").ToString()
     
121.                             '// ถอดรหัสออกมาเพื่อให้เห็น Password แบบ Plain Text.
     
122.                             result.Encrypted = DecryptPassword(DirectCast(reader("Encrypted"), Byte()), key, iv)
     
123.                             result.Email = reader("email").ToString()
     
124.                             result.Phone = reader("phone").ToString()
     
125.                             result.DateModified = reader("DateModified").ToString()
     
126.                             Return (result)
     
127.                         End If
     
128.                     End Using
     
129.                 End Using
     
130.             End Using
     
131.         Catch ex As Exception
     
132.             MessageBox.Show("Error: " & ex.Message, "Report Status", MessageBoxButtons.OK, MessageBoxIcon.Warning)
     
133.         End Try
     
134.         Return result '// Return the initialized structure if no data is found.
     
135.     End Function
     
136. 
     
137.     ' / ------------------------------------------------------------------------------------------------
     
138.     ' / DELETE DATA
     
139.     ' / ------------------------------------------------------------------------------------------------
     
140.     Public Sub DeleteData(PK As Integer)
     
141.         Try
     
142.             Using Conn As New OleDbConnection(ConnString)
     
143.                 Conn.Open()
     
144.                 Dim sql As String = "DELETE FROM UserData WHERE PK = @PK"
     
145.                 Using cmd As New OleDbCommand(sql, Conn)
     
146.                     cmd.Parameters.AddWithValue("@PK", PK)
     
147.                     Try
     
148.                         cmd.ExecuteNonQuery()
     
149.                         MessageBox.Show("DELETE COMPLETE.", "Report Status", MessageBoxButtons.OK, MessageBoxIcon.Information)
     
150.                     Catch ex As Exception
     
151.                         MessageBox.Show("Error: " & ex.Message)
     
152.                     End Try
     
153.                 End Using
     
154.             End Using
     
155.         Catch ex As Exception
     
156.             MessageBox.Show("Error: " & ex.Message, "Report Status", MessageBoxButtons.OK, MessageBoxIcon.Warning)
     
157.         End Try
     
158.     End Sub
     
159. 
     
160.     ' / ------------------------------------------------------------------------------------------------
     
161.     ' / Function to find and create the new Primary Key not to duplicate.
     
162.     ' / ------------------------------------------------------------------------------------------------
     
163.     Function SetupNewPK(ByVal sql As String) As Integer
     
164.         Using Conn As New OleDbConnection(ConnString)
     
165.             Conn.Open()
     
166.             Using cmd As New OleDbCommand(sql, Conn)
     
167.                 '/ Check if the information is available. And return it back
     
168.                 If IsDBNull(cmd.ExecuteScalar) Then
     
169.                     '// Start at 1
     
170.                     SetupNewPK = 1
     
171.                 Else
     
172.                     SetupNewPK = cmd.ExecuteScalar + 1
     
173.                 End If
     
174.             End Using
     
175.         End Using
     
176.     End Function
     
177. 
     
178.     ' / ------------------------------------------------------------------------------------------------
     
179.     ' / Function to check password for login system.
     
180.     ' / ------------------------------------------------------------------------------------------------
     
181.     Public Function CheckPassword(password As String) As Boolean
     
182.         CheckPassword = False
     
183.         Using Conn As New OleDbConnection(ConnString)
     
184.             Conn.Open()
     
185.             '// มีข้อมูลสำหรับ Admin อยู่แค่รายการเดียว
     
186.             Using Cmd As New OleDbCommand("SELECT TOP 1 * FROM AdminData ", Conn)
     
187.                 Using reader As OleDbDataReader = Cmd.ExecuteReader()
     
188.                     If reader.Read() Then
     
189.                         '// ถอดรหัสผ่านแล้วนำค่ามาเปรียบเทียบกับค่าที่ป้อนเข้าไป
     
190.                         If password = DecryptPassword(DirectCast(reader("Encrypted"), Byte()), key, iv) Then Return True
     
191.                     Else
     
192.                         Return False ' ไม่พบข้อมูลผู้ใช้
     
193.                     End If
     
194.                 End Using
     
195.             End Using
     
196.         End Using
     
197.     End Function
     
198. 
     
199.     ' / ------------------------------------------------------------------------------------------------
     
200.     '// Save new Admin Password.
     
201.     ' / ------------------------------------------------------------------------------------------------
     
202.     Sub SavePassword(ByVal NewPassword As String)
     
203.         Try
     
204.             Using Conn As New OleDbConnection(ConnString)
     
205.                 Conn.Open()
     
206.                 '// Update the record if it exists.
     
207.                 Dim sql As String = "UPDATE AdminData SET PK = @PK, Encrypted = @Encrypted WHERE PK = @PK"
     
208.                 '// Update.
     
209.                 Using Cmd As New OleDbCommand(Sql, Conn)
     
210.                     With Cmd.Parameters
     
211.                         .AddWithValue("@PK", 1)
     
212.                         '// การเข้ารหัสด้วย Rijndael (AES) (modEncryption.vb)
     
213.                         .AddWithValue("@Encrypted", EncryptPassword(NewPassword, key, iv))
     
214.                     End With
     
215.                     Cmd.ExecuteNonQuery()
     
216.                 End Using
     
217.                 MessageBox.Show("UPDATE COMPLETE.", "Report Status", MessageBoxButtons.OK, MessageBoxIcon.Information)
     
218.             End Using
     
219.         Catch ex As Exception
     
220.             MessageBox.Show("UPDATE NOT SUCCESSFUL!", "Report Status", MessageBoxButtons.OK, MessageBoxIcon.Warning)
     
221.         End Try
     
222.     End Sub
     
223. End Module

คัดลอกไปที่คลิปบอร์ด

โค้ดในส่วนของการเข้ารหัส/ถอดรหัส (modEncryption.vb)

1. Imports System.Security.Cryptography
   
2. Imports System.Text
   
3. Imports System.IO
   
4. 
   
5. Module modEncryption
   
6. 
   
7.     '// Rijndael (AES) ในการเข้ารหัสข้อมูล (Encryption) โดยการใช้ Key และ IV (Initialization Vector).
   
8.     Public Function EncryptPassword(password As String, key As String, iv As String) As Byte()
   
9.         Dim aes As New RijndaelManaged()
   
10.         aes.Key = Encoding.UTF8.GetBytes(key)
    
11.         aes.IV = Encoding.UTF8.GetBytes(iv)
    
12.         aes.Padding = PaddingMode.PKCS7
    
13.         aes.Mode = CipherMode.CBC
    
14. 
    
15.         Dim encryptor As ICryptoTransform = aes.CreateEncryptor(aes.Key, aes.IV)
    
16. 
    
17.         Using ms As New MemoryStream()
    
18.             Using cs As New CryptoStream(ms, encryptor, CryptoStreamMode.Write)
    
19.                 Using sw As New StreamWriter(cs)
    
20.                     sw.Write(password)
    
21.                 End Using
    
22.             End Using
    
23.             Return ms.ToArray()
    
24.         End Using
    
25.     End Function
    
26. 
    
27.     '// Rijndael (AES) ในการถอดรหัสข้อมูล (Decryption)
    
28.     Public Function DecryptPassword(encryptedPassword As Byte(), key As String, iv As String) As String
    
29.         Dim aes As New RijndaelManaged()
    
30.         aes.Key = Encoding.UTF8.GetBytes(key)
    
31.         aes.IV = Encoding.UTF8.GetBytes(iv)
    
32.         aes.Padding = PaddingMode.PKCS7
    
33.         aes.Mode = CipherMode.CBC
    
34. 
    
35.         Dim decryptor As ICryptoTransform = aes.CreateDecryptor(aes.Key, aes.IV)
    
36. 
    
37.         Using ms As New MemoryStream(encryptedPassword)
    
38.             Using cs As New CryptoStream(ms, decryptor, CryptoStreamMode.Read)
    
39.                 Using sr As New StreamReader(cs)
    
40.                     Return sr.ReadToEnd()
    
41.                 End Using
    
42.             End Using
    
43.         End Using
    
44.     End Function
    
45. End Module

คัดลอกไปที่คลิปบอร์ด

โค้ดในส่วนของฟังค์ชั่น (modFunction.vb)

1. Module modFunction
   
2.     Function MyPath(ByVal AppPath As String) As String
   
3.         '/ Return Value
   
4.         MyPath = AppPath.ToLower.Replace("\bin\debug", "").Replace("\bin\release", "").Replace("\bin\x86\debug", "").Replace("\bin\x86\release", "")
   
5.         '// If not found folder then put the \ (BackSlash) at the end.
   
6.         If Microsoft.VisualBasic.Right(MyPath, 1) <> Chr(92) Then MyPath = MyPath & Chr(92)
   
7.     End Function
   
8. 
   
9.     ' / --------------------------------------------------------------------------------
   
10.     '// Initialize DataGridView @Run Time
    
11.     Public Sub SetupDataGridView(DGV As DataGridView)
    
12.         With DGV
    
13.             .RowHeadersVisible = False
    
14.             .AllowUserToAddRows = False
    
15.             .AllowUserToDeleteRows = False
    
16.             .AllowUserToResizeRows = False
    
17.             .MultiSelect = False
    
18.             .SelectionMode = DataGridViewSelectionMode.FullRowSelect
    
19.             .ReadOnly = True
    
20.             .Font = New Font("Tahoma", 11, FontStyle.Regular)
    
21.             .RowTemplate.Height = 28
    
22.             .RowTemplate.Resizable = DataGridViewTriState.False
    
23.             .Columns(0).Visible = False '// Primary Key
    
24.             '// Autosize Column
    
25.             .AutoSizeColumnsMode = DataGridViewAutoSizeColumnsMode.Fill
    
26.             '// Even-Odd Color
    
27.             .AlternatingRowsDefaultCellStyle.BackColor = Color.AliceBlue
    
28.             '// COLUMN HEADER
    
29.             '// Adjust Header Styles
    
30.             With .ColumnHeadersDefaultCellStyle
    
31.                 .BackColor = Color.Navy
    
32.                 .ForeColor = Color.Black ' Color.White
    
33.                 .Font = New Font("Tahoma", 11, FontStyle.Bold)
    
34.             End With
    
35.             '// Before you can adjust the height of the row.
    
36.             .ColumnHeadersHeightSizeMode = DataGridViewColumnHeadersHeightSizeMode.DisableResizing
    
37.             .ColumnHeadersHeight = 30
    
38.             '// Accept color background changes.
    
39.             .EnableHeadersVisualStyles = False
    
40.             For iCol = 0 To DGV.Columns.Count - 1
    
41.                 '// Calculates odd and even numbers. If any integer divide by 2, then answer 1 is odd number.
    
42.                 If iCol Mod 2 = 0 Then
    
43.                     DGV.Columns(iCol).HeaderCell.Style.BackColor = Color.Gold
    
44.                     '// Integer divide by 2, then answer 1 is Even number.
    
45.                 Else
    
46.                     DGV.Columns(iCol).HeaderCell.Style.BackColor = Color.Orange
    
47.                 End If
    
48.             Next
    
49.         End With
    
50.     End Sub
    
51. 
    
52. End Module

คัดลอกไปที่คลิปบอร์ด

ดาวน์โหลดโค้ดต้นฉบับ VB.NET (2010) ได้ที่นี่ ...