[VB.NET] โค้ดการเก็บข้อมูลรหัสผ่านด้วยการเข้ารหัสแบบ Long Binary (CRUD)

thongkorn

โค้ดโปรแกรมการเก็บข้อมูลรหัสผ่านด้วยการเข้ารหัสแบบ Long Binary หรือชนิดข้อมูลแบบ OLE Object สำหรับ MS Access โดยจะใช้การเข้ารหัสแบบ Rijndael (AES) Algorithm ... สำหรับการจัดการกับฐานข้อมูลเรียกว่า Parameterized Query หรือการใช้คำสั่งที่มีการส่งผ่านค่าไปยังตัวแปรพารามิเตอร์ (Parameterization) โดยตรงในการเขียน SQL Query ซึ่งเป็นวิธีที่ช่วยป้องกันการโจมตีแบบ SQL Injection ได้ดี ... โค้ดชุดนี้แอดมินเพิ่มความสวยงามด้วยการใช้ MaterialSkin เข้ามาช่วย ...

มาดูโค้ดฉบับเต็มจากฟอร์มหลัก (frmPassCodeBook.vb)

1. Imports System.Data.OleDb
   
2. Imports MaterialSkin
   
3. 
   
4. Public Class frmPassCodeBook
   
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 frmPassCodeBook_Load(sender As System.Object, e As System.EventArgs) Handles MyBase.Load
    
13.         '// Initialize MaterialSkin .Net Framework 4.0
    
14.         Dim SkinManager As MaterialSkinManager = MaterialSkinManager.Instance
    
15.         SkinManager.AddFormToManage(Me)
    
16.         SkinManager.Theme = MaterialSkinManager.Themes.LIGHT
    
17.         '//SkinManager.ColorScheme = New MaterialSkin.ColorScheme(Primary.Amber500, Primary.BlueGrey900, Primary.BlueGrey500, Accent.LightBlue200, TextShade.WHITE)
    
18.         SkinManager.ColorScheme = New MaterialSkin.ColorScheme(Primary.Green600, Primary.Green700, Primary.Green200, Accent.Red100, TextShade.WHITE)
    
19. 
    
20.         '// ตั้งค่าขนาดของฟอร์มแบบ Run Time
    
21.         Me.Size = New Size(1020, 720)
    
22.         '// คำนวณตำแหน่งที่กึ่งกลางของหน้าจอ
    
23.         Me.Location = New Point((Screen.PrimaryScreen.WorkingArea.Width - Me.Width) \ 2,
    
24.         (Screen.PrimaryScreen.WorkingArea.Height - Me.Height) \ 2)
    
25. 
    
26.         '// โหลดข้อมูลลงตารางกริด
    
27.         dgvData.DataSource = RetrieveData()
    
28.         Call SetupDataGridView(dgvData)
    
29.         lblRecordCount.Text = "[Total: " & dgvData.Rows.Count & " Record.]"
    
30.         '// เริ่มโหมดว่างเปล่าเพื่อทำรายการใหม่ หรือเลือกรายการขึ้นมาแก้ไข
    
31.         Call NewMode()
    
32.     End Sub
    
33. 
    
34.     ' / ------------------------------------------------------------------------------------------------
    
35.     ' / Add New Mode
    
36.     ' / ------------------------------------------------------------------------------------------------
    
37.     Private Sub NewMode()
    
38.         '// Clear all TextBox.
    
39.         For Each c In GroupBox1.Controls
    
40.             If TypeOf c Is TextBox Then
    
41.                 DirectCast(c, TextBox).Clear()
    
42.                 DirectCast(c, TextBox).Enabled = False
    
43.             End If
    
44.             If TypeOf c Is DateTimePicker Then DirectCast(c, DateTimePicker).Enabled = False
    
45.         Next
    
46.         itemAdd.Enabled = True
    
47.         itemSave.Enabled = False
    
48.         itemDelete.Enabled = True
    
49.         itemDelete.Text = "DELETE DATA"
    
50.         itemExit.Enabled = True
    
51.     End Sub
    
52. 
    
53.     ' / ------------------------------------------------------------------------------------------------
    
54.     ' / Edit Data Mode
    
55.     ' / ------------------------------------------------------------------------------------------------
    
56.     Private Sub EditMode()
    
57.         '// Clear all TextBox
    
58.         For Each c In GroupBox1.Controls
    
59.             If TypeOf c Is TextBox Then
    
60.                 DirectCast(c, TextBox).Enabled = True
    
61.             End If
    
62.             If TypeOf c Is DateTimePicker Then DirectCast(c, DateTimePicker).Enabled = True
    
63.         Next
    
64.         itemAdd.Enabled = False
    
65.         itemSave.Enabled = True
    
66.         itemDelete.Enabled = True
    
67.         itemDelete.Text = "CANCEL"
    
68.     End Sub
    
69. 
    
70.     ' / ------------------------------------------------------------------------------------------------
    
71.     ' / Add new data.
    
72.     ' / ------------------------------------------------------------------------------------------------
    
73.     Private Sub itemAdd_Click(sender As System.Object, e As System.EventArgs) Handles itemAdd.Click
    
74.         blnNewData = True
    
75.         Call EditMode()
    
76.         txtUrl.Focus()
    
77.     End Sub
    
78. 
    
79.     ' / ------------------------------------------------------------------------------------------------
    
80.     ' / Save Data.
    
81.     ' / ------------------------------------------------------------------------------------------------
    
82.     Private Sub itemSave_Click(sender As System.Object, e As System.EventArgs) Handles itemSave.Click
    
83.         If txtUrl.Text = "" Or txtUrl.Text.Trim.Length = 0 Then
    
84.             MessageBox.Show("Please enter Url or Name.", "Report Status", MessageBoxButtons.OK, MessageBoxIcon.Warning)
    
85.             txtUrl.Focus()
    
86.             Return
    
87.         End If
    
88.         If txtLogin.Text = "" Or txtLogin.Text.Trim.Length = 0 Then
    
89.             MessageBox.Show("Please enter Login Name.", "Report Status", MessageBoxButtons.OK, MessageBoxIcon.Warning)
    
90.             txtLogin.Focus()
    
91.             Return
    
92.         End If
    
93.         If txtPassword.Text = "" Or txtPassword.Text.Trim.Length = 0 Then
    
94.             MessageBox.Show("Please enter Password.", "Report Status", MessageBoxButtons.OK, MessageBoxIcon.Warning)
    
95.             txtPassword.Focus()
    
96.             Return
    
97.         End If
    
98.         '// Details Class in modDataBase.vb
    
99.         '// กำหนดค่าให้กับตัวแปร เพื่อส่งออกไปทำการบันทึกข้อมูล
    
100.         Dim result As New Details
     
101.         With result
     
102.             .PK = PK
     
103.             .Login = txtLogin.Text
     
104.             .Encrypted = txtPassword.Text
     
105.             .Url = txtUrl.Text
     
106.             .Email = txtEmail.Text
     
107.             .Phone = txtPhone.Text
     
108.             .DateModified = dtpDateModified.Value
     
109.         End With
     
110.         '// blnNewData = True คือการเพิ่มข้อมูล
     
111.         If blnNewData Then
     
112.             '// หาค่า Primary Key จาก SetupNewPK อยู่ใน modDataBase.vb
     
113.             result.PK = SetupNewPK("SELECT MAX(UserData.PK) AS MaxPK FROM UserData")
     
114.             '// ส่งค่า blnNewData = True บอกว่าเป็นการเพิ่มข้อมูลใหม่ และค่าของ Control บนฟอร์มไปด้วย
     
115.             Call SaveData(blnNewData, result)
     
116.             Call NewMode()
     
117. 
     
118.             '// การแก้ไขข้อมูล
     
119.         Else
     
120.             'blnNewData = False
     
121.             '// ส่งค่า blnNewData = False บอกว่าเป็นการเพิ่มข้อมูลใหม่ และส่งค่าของ Control บนฟอร์มไปด้วย
     
122.             Call SaveData(blnNewData, result)
     
123.             Call NewMode()
     
124.         End If
     
125.         '// Refresh data.
     
126.         Call btnRefresh_Click(sender, e)
     
127.     End Sub
     
128. 
     
129.     ' / ------------------------------------------------------------------------------------------------
     
130.     ' / DELETE DATA
     
131.     ' / ------------------------------------------------------------------------------------------------
     
132.     Private Sub itemDelete_Click(sender As System.Object, e As System.EventArgs) Handles itemDelete.Click
     
133.         If itemDelete.Text = "DELETE DATA" Then
     
134.             '// เช็คก่อนว่ามีรายแถวอยู่หรือไม่ หรือมีการคลิ๊กเลือกแถวรายการหรือเปล่า
     
135.             If dgvData.RowCount = 0 Or dgvData.CurrentRow Is Nothing Then Exit Sub
     
136.             Dim Url As String = dgvData.Item(1, dgvData.CurrentRow.Index).Value
     
137.             '// ถามยืนยันการลบข้อมูล
     
138.             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)
     
139.             If Result = DialogResult.Yes Then
     
140.                 '// Receive Primary Key value to confirm the deletion.
     
141.                 PK = dgvData.Item(0, dgvData.CurrentRow.Index).Value
     
142.                 '// ลบข้อมูล (modDataBase.vb)
     
143.                 Call DeleteData(PK)
     
144.                 '// ทำการแสดงผลรายการใหม่
     
145.                 Call btnRefresh_Click(sender, e)
     
146.                 Call NewMode()
     
147.             End If
     
148.         ElseIf itemDelete.Text = "CANCEL" Then
     
149.             '// เริ่มโหมดว่างเปล่าเพื่อทำรายการใหม่ หรือเลือกรายการขึ้นมาแก้ไข
     
150.             Call NewMode()
     
151.         End If
     
152.     End Sub
     
153. 
     
154.     ' / ------------------------------------------------------------------------------------------------
     
155.     ' / แสดงผลข้อมูลทั้งหมด
     
156.     ' / ------------------------------------------------------------------------------------------------
     
157.     Private Sub btnRefresh_Click(sender As System.Object, e As System.EventArgs) Handles btnRefresh.Click
     
158.         dgvData.DataSource = RetrieveData()
     
159.         lblRecordCount.Text = "[Total: " & dgvData.Rows.Count & " Record.]"
     
160.     End Sub
     
161. 
     
162.     ' / ------------------------------------------------------------------------------------------------
     
163.     ' / การค้นหาข้อมูล
     
164.     ' / ------------------------------------------------------------------------------------------------
     
165.     Private Sub txtSearch_KeyPress(sender As Object, e As System.Windows.Forms.KeyPressEventArgs) Handles txtSearch.KeyPress
     
166.         '// Undesirable characters for the database ex.  ', * or %
     
167.         txtSearch.Text = txtSearch.Text.Replace("'", "").Replace("*", "").Replace("%", "")
     
168.         If Trim(txtSearch.Text) = "" Or Len(Trim(txtSearch.Text)) = 0 Then Exit Sub
     
169.         '// RetrieveData(True) It means searching for information.
     
170.         If e.KeyChar = Chr(13) Then '// Press Enter
     
171.             '// No beep.
     
172.             e.Handled = True
     
173.             '//
     
174.             dgvData.DataSource = RetrieveData(True, txtSearch.Text)
     
175.             lblRecordCount.Text = "[Total: " & dgvData.Rows.Count & " Record.]"
     
176.             txtSearch.Clear()
     
177.         End If
     
178.     End Sub
     
179. 
     
180.     ' / ------------------------------------------------------------------------------------------------
     
181.     ' / ดับเบิ้ลคิ๊กเมาส์ในตารางกริด จากนั้นค้นหา Primary Key เพื่อนำข้อมูลไปแสดงผล
     
182.     ' / ------------------------------------------------------------------------------------------------
     
183.     Private Sub dgvData_CellDoubleClick(sender As Object, e As System.Windows.Forms.DataGridViewCellEventArgs) Handles dgvData.CellDoubleClick
     
184.         If e.RowIndex >= 0 Then
     
185.             '// เก็บค่านี้เอาไว้เพื่อใช้ในการอ้างอิงตอนที่ทำการอัพเดตข้อมูล
     
186.             PK = CInt(dgvData.Rows(e.RowIndex).Cells("PK").Value)
     
187.             blnNewData = False    '// กำหนดให้เป็นการแก้ไขข้อมูล
     
188.             '// ส่งค่า PK เพื่อเป็นคีย์ไปทำการค้นข้อมูล แล้วคืนข้อมูลกลับมาในรูปแบบ Class
     
189.             '// Class Details อยู่ใน modDataBase.vb
     
190.             Dim result As Details = RetrieveDetails(PK)
     
191.             txtLogin.Text = result.Login
     
192.             txtPassword.Text = result.Encrypted
     
193.             txtUrl.Text = result.Url
     
194.             txtEmail.Text = result.Email
     
195.             txtPhone.Text = result.Phone
     
196.             dtpDateModified.Text = Format(result.DateModified, "dd/MM/yyyy")
     
197.             '// โหมดในการแก้ไข
     
198.             Call EditMode()
     
199.             txtUrl.Focus()
     
200.         End If
     
201.     End Sub
     
202. 
     
203.     '// Function to handle the KeyPress event for TextBox controls.
     
204.     Private Sub TextBox_KeyPress(sender As Object, e As KeyPressEventArgs) Handles txtLogin.KeyPress, txtPassword.KeyPress, txtUrl.KeyPress, txtEmail.KeyPress, txtPhone.KeyPress
     
205.         If e.KeyChar = ChrW(Keys.Enter) Then
     
206.             e.Handled = True '// Prevent the Enter key from adding a new line in the TextBox.
     
207.             SendKeys.Send("{TAB}") '// Send the TAB key to move to the next control.
     
208.         End If
     
209.     End Sub
     
210. 
     
211.     Private Sub dtpDateModified_KeyDown(sender As Object, e As System.Windows.Forms.KeyEventArgs) Handles dtpDateModified.KeyDown
     
212.         If e.KeyCode = Keys.Enter Then
     
213.             e.Handled = True
     
214.             SendKeys.Send("{TAB}")
     
215.         End If
     
216.     End Sub
     
217. 
     
218.     Private Sub frmPassCodeBook_FormClosed(sender As Object, e As System.Windows.Forms.FormClosedEventArgs) Handles Me.FormClosed
     
219.         Me.Dispose()
     
220.         GC.SuppressFinalize(Me)
     
221.         End
     
222.     End Sub
     
223. 
     
224.     Private Sub itemExit_Click(sender As System.Object, e As System.EventArgs) Handles itemExit.Click
     
225.         Me.Close()
     
226.     End Sub
     
227. 
     
228. 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.     ' / ------------------------------------------------------------------------------------------------
   
7.     '// การใช้ Key และ IV
   
8.     '// Key ควรมีขนาด 16, 24 หรือ 32 bytes เพื่อความปลอดภัย
   
9.     '// IV (Initialization Vector) ควรมีขนาด 16 bytes สำหรับการเข้ารหัส AES
   
10.     Dim key As String = "1234567890123456" '// Key 16 bytes
    
11.     Dim iv As String = "6543210987654321"  '// IV 16 bytes
    
12.     ' / ------------------------------------------------------------------------------------------------
    
13. 
    
14.     Public Class Details
    
15.         '// Auto Implemented Properties for Data Fields.
    
16.         Public Property PK As Integer
    
17.         Public Property Login As String
    
18.         Public Property Encrypted As String
    
19.         Public Property Url As String
    
20.         Public Property Email As String
    
21.         Public Property Phone As String
    
22.         Public Property DateModified As Date
    
23.     End Class
    
24. 
    
25.     ' / ------------------------------------------------------------------------------------------------
    
26.     '// Save data to MS Access database.
    
27.     ' / ------------------------------------------------------------------------------------------------
    
28.     Sub SaveData(newdata As Boolean, details As Details)
    
29.         Dim sql As String
    
30.         Try
    
31.             Using Conn As New OleDbConnection(ConnString)
    
32.                 Conn.Open()
    
33.                 '// การเพิ่มข้อมูลใหม่
    
34.                 If newdata Then
    
35.                     sql = "INSERT INTO UserData (PK, url, Login, Encrypted, email, phone, DateModified) VALUES (@PK, @Url, @Email, @Phone, @Login, @Encrypted, @DateModified)"
    
36. 
    
37.                     '// การแก้ไข
    
38.                 Else
    
39.                     '// Update the record if it exists.
    
40.                     sql = "UPDATE UserData SET PK = @PK, url = @Url, Login = @Login, Encrypted = @Encrypted, email = @Email, Phone = @Phone, DateModified = @DateModified WHERE PK = @PK"
    
41.                 End If
    
42.                 '// Update.
    
43.                 Using Cmd As New OleDbCommand(sql, Conn)
    
44.                     With Cmd.Parameters
    
45.                         .AddWithValue("@PK", details.PK)
    
46.                         .AddWithValue("@url", details.Url)
    
47.                         .AddWithValue("@login", details.Login)
    
48.                         '// การเข้ารหัสด้วย Rijndael (AES) (modEncryption.vb)
    
49.                         .AddWithValue("@Encrypted", EncryptPassword(details.Encrypted, key, iv))
    
50.                         .AddWithValue("@email", details.Email)
    
51.                         .AddWithValue("@phone", details.Phone)
    
52.                         .AddWithValue("@DateModified", Format(details.DateModified, "dd/MM/yyyy"))
    
53.                     End With
    
54.                     Cmd.ExecuteNonQuery()
    
55.                 End Using
    
56.                 MessageBox.Show("UPDATE COMPLETE.", "Report Status", MessageBoxButtons.OK, MessageBoxIcon.Information)
    
57.             End Using
    
58.         Catch ex As Exception
    
59.             MessageBox.Show("UPDATE NOT SUCCESSFUL!", "Report Status", MessageBoxButtons.OK, MessageBoxIcon.Warning)
    
60.         End Try
    
61.     End Sub
    
62. 
    
63.     ' / ------------------------------------------------------------------------------------------------
    
64.     '// Retrieve data from MS Access database and display in DataGridView.
    
65.     '// blnSearch = True คือการค้นหาข้อมูล โดยรับค่า SearchValue เป็นคีย์เวิร์ดใช้ในการค้นหา
    
66.     '// blnSearch = False ก็จะแสดงผลข้อมูลออกมาทั้งหมด
    
67.     '// ทั้ง 2 ค่าเป็น Optional คือหากไม่ระบุการส่งค่ามา ก็ให้มีค่าเป็น False หรือแสดงผลข้อมูลออกมาทั้งหมด
    
68.     ' / ------------------------------------------------------------------------------------------------
    
69.     Function RetrieveData(Optional ByVal blnSearch As Boolean = False, Optional ByVal SearchValue As String = "") As DataTable
    
70.         Using Conn As New OleDbConnection(ConnString)
    
71.             Conn.Open()
    
72.             Dim sql As String = "SELECT PK, url, login, Encrypted, email, phone, DateModified FROM UserData"
    
73.             If blnSearch Then
    
74.                 sql &= " WHERE url LIKE @SearchValue OR email LIKE @SearchValue OR phone LIKE @SearchValue OR login LIKE @SearchValue"
    
75.             End If
    
76.             sql &= " ORDER BY PK"
    
77.             '//
    
78.             Using cmd As New OleDbCommand(sql, Conn)
    
79.                 If blnSearch Then cmd.Parameters.AddWithValue("@SearchValue", "%" & SearchValue & "%")
    
80.                 '//
    
81.                 Using reader As OleDbDataReader = cmd.ExecuteReader()
    
82.                     Dim dt As New DataTable()
    
83.                     With dt.Columns
    
84.                         .Add("PK", GetType(Integer))
    
85.                         .Add("URL", GetType(String))
    
86.                         .Add("Login", GetType(String))
    
87.                         .Add("Encrypted", GetType(String))
    
88.                         .Add("Email", GetType(String))
    
89.                         .Add("Phone", GetType(String))
    
90.                         .Add("DateModified", GetType(Date))
    
91.                     End With
    
92.                     While reader.Read()
    
93.                         '// มีการเข้ารหัสของรหัสผ่านเอาไว้ตอนแสดงผลในตารางกริด
    
94.                         Dim EncryptedData As Byte() = DirectCast(reader("Encrypted"), Byte())
    
95.                         '// เพิ่มรายการแถวข้อมูลเข้าไปใน DataTable
    
96.                         dt.Rows.Add(reader("PK"), reader("url"), reader("login"), EncryptedData, reader("email"), reader("phone"), reader("DateModified"))
    
97.                     End While
    
98.                     Return dt
    
99.                 End Using
    
100.             End Using
     
101.         End Using
     
102.     End Function
     
103. 
     
104.     ' / ------------------------------------------------------------------------------------------------
     
105.     ' / คืนค่าข้อมูลจากตารางกริดไปแสดงผลลงใน Control
     
106.     ' / ------------------------------------------------------------------------------------------------
     
107.     Public Function RetrieveDetails(PK As Integer) As Details
     
108.         '// Class Details เพื่อคืนค่ากลับไปแสดงผลในฟอร์ม
     
109.         Dim result As New Details()
     
110.         Try
     
111.             Using Conn As New OleDbConnection(ConnString)
     
112.                 Conn.Open()
     
113.                 Using cmd As New OleDbCommand("SELECT url, login, Encrypted, email, phone, DateModified FROM UserData WHERE PK = @PK", Conn)
     
114.                     cmd.Parameters.AddWithValue("@PK", PK)
     
115.                     Using reader As OleDbDataReader = cmd.ExecuteReader()
     
116.                         If reader.Read() Then
     
117.                             result.Url = reader("url").ToString()
     
118.                             result.Login = reader("login").ToString()
     
119.                             '// ถอดรหัสออกมาเพื่อให้เห็น Password แบบ Plain Text.
     
120.                             result.Encrypted = DecryptPassword(DirectCast(reader("Encrypted"), Byte()), key, iv)
     
121.                             result.Email = reader("email").ToString()
     
122.                             result.Phone = reader("phone").ToString()
     
123.                             result.DateModified = reader("DateModified").ToString()
     
124.                             Return (result)
     
125.                         End If
     
126.                     End Using
     
127.                 End Using
     
128.             End Using
     
129.         Catch ex As Exception
     
130.             MessageBox.Show("Error: " & ex.Message, "Report Status", MessageBoxButtons.OK, MessageBoxIcon.Warning)
     
131.         End Try
     
132.         Return result '// Return the initialized structure if no data is found.
     
133.     End Function
     
134. 
     
135.     ' / ------------------------------------------------------------------------------------------------
     
136.     ' / DELETE DATA
     
137.     ' / ------------------------------------------------------------------------------------------------
     
138.     Public Sub DeleteData(PK As Integer)
     
139.         Try
     
140.             Using Conn As New OleDbConnection(ConnString)
     
141.                 Conn.Open()
     
142.                 Dim sql As String = "DELETE FROM UserData WHERE PK = @PK"
     
143.                 Using cmd As New OleDbCommand(sql, Conn)
     
144.                     cmd.Parameters.AddWithValue("@PK", PK)
     
145.                     Try
     
146.                         cmd.ExecuteNonQuery()
     
147.                         MessageBox.Show("DELETE COMPLETE.", "Report Status", MessageBoxButtons.OK, MessageBoxIcon.Information)
     
148.                     Catch ex As Exception
     
149.                         MessageBox.Show("Error: " & ex.Message)
     
150.                     End Try
     
151.                 End Using
     
152.             End Using
     
153.         Catch ex As Exception
     
154.             MessageBox.Show("Error: " & ex.Message, "Report Status", MessageBoxButtons.OK, MessageBoxIcon.Warning)
     
155.         End Try
     
156.     End Sub
     
157. 
     
158.     ' / ------------------------------------------------------------------------------------------------
     
159.     ' / Function to find and create the new Primary Key not to duplicate.
     
160.     ' / ------------------------------------------------------------------------------------------------
     
161.     Function SetupNewPK(ByVal sql As String) As Integer
     
162.         Using Conn As New OleDbConnection(ConnString)
     
163.             Conn.Open()
     
164.             Using cmd As New OleDbCommand(sql, Conn)
     
165.                 '/ Check if the information is available. And return it back
     
166.                 If IsDBNull(cmd.ExecuteScalar) Then
     
167.                     '// Start at 1
     
168.                     SetupNewPK = 1
     
169.                 Else
     
170.                     SetupNewPK = cmd.ExecuteScalar + 1
     
171.                 End If
     
172.             End Using
     
173.         End Using
     
174.     End Function
     
175. 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) ได้ที่นี่ ...