Add SQLCipher encrypted SQLite support

Author: emako

QuickLook.Plugin/QuickLook.Plugin.DbViewer/DbViewerPanel.xaml.cs

@@ -16,6 +16,7 @@ public partial class DbViewerPanel : UserControl
 {
     private const int DefaultPageSize = 200;
     private string _path;
+    private string _sqlitePassword;
     private DatabaseType _databaseType;
     private string _currentObjectName;
     private int _totalCount;
@@ -29,10 +30,14 @@ public DbViewerPanel()
         pagination.PageSize = DefaultPageSize;
     }
 
-    public void LoadDatabase(string path)
+    public void LoadDatabase(string path, string password = null)
     {
         _path = path;
-        _databaseType = DetectDatabaseType(path);
+        _sqlitePassword = password;
+        // When a password is supplied the caller already verified it is SQLite.
+        _databaseType = password != null
+            ? DatabaseType.SQLite
+            : Plugin.DetectDatabaseType(path);
         _hasLoadedData = false;
 
         if (_databaseType == DatabaseType.Unknown)
@@ -155,41 +160,19 @@ private void LoadCurrentPage()
         pageInfoTextBlock.Text = $"{_totalCount} 行 / 每页 {DefaultPageSize} 行 / 第 {pagination.CurrentPage} 页，共 {pagination.PageCount} 页";
     }
 
-    private static DatabaseType DetectDatabaseType(string path)
+    private SqliteConnection OpenSqliteConnection(string path)
     {
-        try
-        {
-            using var stream = File.OpenRead(path);
-            var header = new byte[59]; // need 32+27 bytes for LiteDB magic
-            var count = stream.Read(header, 0, header.Length);
-
-            if (count >= 16)
-            {
-                var text = System.Text.Encoding.ASCII.GetString(header, 0, 16);
-                if (text.StartsWith("SQLite format 3", StringComparison.Ordinal))
-                    return DatabaseType.SQLite;
-            }
-
-            // LiteDB 5.x: magic string "** This is a LiteDB file **" at offset 32
-            if (count >= 59)
-            {
-                var liteDbMagic = System.Text.Encoding.ASCII.GetString(header, 32, 27);
-                if (liteDbMagic == "** This is a LiteDB file **")
-                    return DatabaseType.LiteDb;
-            }
-        }
-        catch
-        {
-        }
-
-        return DatabaseType.Unknown;
+        var csb = new SqliteConnectionStringBuilder { DataSource = path };
+        if (!string.IsNullOrEmpty(_sqlitePassword))
+            csb.Password = _sqlitePassword;
+        return new SqliteConnection(csb.ToString());
     }
 
-    private static string[] LoadSqliteNames(string path)
+    private string[] LoadSqliteNames(string path)
     {
         var items = new List<string>();
 
-        using var connection = new SqliteConnection($"Data Source={path}");
+        using var connection = OpenSqliteConnection(path);
         connection.Open();
 
         using var command = connection.CreateCommand();
@@ -201,18 +184,18 @@ private static string[] LoadSqliteNames(string path)
             items.Add(reader.GetString(0));
         }
 
-        return items.ToArray();
+        return [.. items];
     }
 
     private static string[] LoadLiteDbNames(string path)
     {
         using var db = new LiteDatabase(path);
-        return db.GetCollectionNames().OrderBy(name => name).ToArray();
+        return [.. db.GetCollectionNames().OrderBy(name => name)];
     }
 
-    private static int GetSqliteRowCount(string path, string tableName)
+    private int GetSqliteRowCount(string path, string tableName)
     {
-        using var connection = new SqliteConnection($"Data Source={path}");
+        using var connection = OpenSqliteConnection(path);
         connection.Open();
 
         using var command = connection.CreateCommand();
@@ -224,12 +207,12 @@ private static int GetLiteDbRowCount(string path, string collectionName)
     {
         using var db = new LiteDatabase(path);
         var collection = db.GetCollection(collectionName);
-        return (int)collection.Count();
+        return collection.Count();
     }
 
-    private static DataTable LoadSqlitePage(string path, string tableName, int page, int pageSize)
+    private DataTable LoadSqlitePage(string path, string tableName, int page, int pageSize)
     {
-        using var connection = new SqliteConnection($"Data Source={path}");
+        using var connection = OpenSqliteConnection(path);
         connection.Open();
 
         using var command = connection.CreateCommand();
@@ -250,9 +233,9 @@ private static DataTable LoadLiteDbPage(string path, string collectionName, int
         return ConvertBsonDocumentsToTable(docs);
     }
 
-    private static DataTable LoadSqliteTable(string path, string tableName)
+    private DataTable LoadSqliteTable(string path, string tableName)
     {
-        using var connection = new SqliteConnection($"Data Source={path}");
+        using var connection = OpenSqliteConnection(path);
         connection.Open();
 
         using var command = connection.CreateCommand();
@@ -311,11 +294,4 @@ private static object ConvertBsonValue(BsonValue value)
 
         return value.RawValue?.ToString() ?? value.ToString() ?? string.Empty;
     }
-
-    private enum DatabaseType
-    {
-        Unknown,
-        SQLite,
-        LiteDb,
-    }
 }

QuickLook.Plugin/QuickLook.Plugin.DbViewer/PasswordControl.xaml

@@ -0,0 +1,54 @@
+<UserControl x:Class="QuickLook.Plugin.DbViewer.PasswordControl"
+             xmlns="http://schemas.microsoft.com/winfx/2006/xaml/presentation"
+             xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
+             xmlns:d="http://schemas.microsoft.com/expression/blend/2008"
+             xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006"
+             d:DesignHeight="450"
+             d:DesignWidth="800"
+             mc:Ignorable="d">
+    <Grid>
+        <Border HorizontalAlignment="Center"
+                VerticalAlignment="Center"
+                Background="{DynamicResource CardBackground}"
+                BorderBrush="{DynamicResource CardBorderBrush}"
+                BorderThickness="1"
+                CornerRadius="5">
+            <Grid Margin="24">
+                <StackPanel>
+                    <TextBlock x:Name="titleTextBlock"
+                               FontSize="18"
+                               FontWeight="DemiBold"
+                               Text="输入密码" />
+                    <TextBlock x:Name="hintTextBlock"
+                               Margin="0,20,0,0"
+                               Text="此数据库已加密，请输入密码以打开。" />
+                    <PasswordBox x:Name="passwordBox"
+                                 MinWidth="400"
+                                 Margin="0,16,0,0" />
+                    <TextBlock x:Name="passwordErrorTextBlock"
+                               MinWidth="400"
+                               Margin="0,16,0,0"
+                               Foreground="Red"
+                               Text="密码错误，请重试"
+                               Visibility="Collapsed" />
+                    <Grid MinWidth="400" Margin="0,16,0,0">
+                        <Grid.ColumnDefinitions>
+                            <ColumnDefinition Width="*" />
+                            <ColumnDefinition Width="*" />
+                        </Grid.ColumnDefinitions>
+                        <Button x:Name="openButton"
+                                Grid.Column="0"
+                                Margin="0,0,4,0"
+                                HorizontalAlignment="Stretch"
+                                Content="打开" />
+                        <Button x:Name="cancelButton"
+                                Grid.Column="1"
+                                Margin="4,0,0,0"
+                                HorizontalAlignment="Stretch"
+                                Content="取消" />
+                    </Grid>
+                </StackPanel>
+            </Grid>
+        </Border>
+    </Grid>
+</UserControl>

QuickLook.Plugin/QuickLook.Plugin.DbViewer/PasswordControl.xaml.cs

@@ -0,0 +1,43 @@
+using System;
+using System.Windows;
+using System.Windows.Controls;
+
+namespace QuickLook.Plugin.DbViewer;
+
+public partial class PasswordControl : UserControl
+{
+    public event Func<string, bool> PasswordRequested;
+
+    public string Password => passwordBox.Dispatcher.Invoke(() => passwordBox.Password);
+
+    public PasswordControl()
+    {
+        InitializeComponent();
+
+        openButton.Click += OpenButton_Click;
+        cancelButton.Click += CancelButton_Click;
+        passwordBox.KeyDown += (s, e) =>
+        {
+            if (e.Key == System.Windows.Input.Key.Enter)
+                OpenButton_Click(s, null);
+        };
+    }
+
+    private void OpenButton_Click(object sender, RoutedEventArgs e)
+    {
+        var pw = Password;
+        if (string.IsNullOrEmpty(pw))
+            return;
+
+        passwordErrorTextBlock.Visibility = Visibility.Collapsed;
+
+        bool accepted = PasswordRequested?.Invoke(pw) ?? false;
+        if (!accepted)
+            passwordErrorTextBlock.Dispatcher.Invoke(() => passwordErrorTextBlock.Visibility = Visibility.Visible);
+    }
+
+    private void CancelButton_Click(object sender, RoutedEventArgs e)
+    {
+        Window.GetWindow(this)?.Close();
+    }
+}

QuickLook.Plugin/QuickLook.Plugin.DbViewer/Plugin.cs

@@ -1,3 +1,4 @@
+using Microsoft.Data.Sqlite;
 using QuickLook.Common.Commands;
 using QuickLook.Common.Controls;
 using QuickLook.Common.Plugin;
@@ -12,6 +13,7 @@ namespace QuickLook.Plugin.DbViewer;
 public sealed partial class Plugin : IViewer, IMoreMenu
 {
     private DbViewerPanel _panel;
+    private PasswordControl _passwordControl;
     private string _currentPath;
 
     public int Priority => 0;
@@ -20,6 +22,14 @@ public sealed partial class Plugin : IViewer, IMoreMenu
 
     public void Init()
     {
+        try
+        {
+            SQLitePCL.Batteries_V2.Init();
+        }
+        catch
+        {
+            // Provider may already be initialised by another component; proceed.
+        }
     }
 
     public bool CanHandle(string path)
@@ -45,22 +55,59 @@ public void Prepare(string path, ContextObject context)
     public void View(string path, ContextObject context)
     {
         _currentPath = path;
-        _panel = new DbViewerPanel();
 
+        var dbType = DetectDatabaseType(path);
+
+        if (dbType == DatabaseType.EncryptedSQLite)
+        {
+            _passwordControl = new PasswordControl();
+            _passwordControl.PasswordRequested += password =>
+            {
+                // Verify the password
+                try
+                {
+                    var csb = new SqliteConnectionStringBuilder { DataSource = path, Password = password };
+                    using var conn = new SqliteConnection(csb.ToString());
+                    conn.Open();
+                    using var cmd = conn.CreateCommand();
+                    cmd.CommandText = "SELECT 1";
+                    cmd.ExecuteScalar();
+                }
+                catch
+                {
+                    return false;
+                }
+
+                // Password accepted — switch to the data panel
+                _panel = new DbViewerPanel();
+                context.ViewerContent = _panel;
+                context.Title = Path.GetFileName(path);
+                _panel.LoadDatabase(path, password);
+                context.IsBusy = false;
+                return true;
+            };
+
+            context.ViewerContent = _passwordControl;
+            context.Title = $"[ENCRYPTED] {Path.GetFileName(path)}";
+            context.IsBusy = false;
+            return;
+        }
+
+        _panel = new DbViewerPanel();
         context.ViewerContent = _panel;
         context.Title = Path.GetFileName(path);
         _panel.LoadDatabase(path);
-
         context.IsBusy = false;
     }
 
     public void Cleanup()
     {
         GC.SuppressFinalize(this);
         _panel = null;
+        _passwordControl = null;
     }
 
-    private static DatabaseType DetectDatabaseType(string path)
+    internal static DatabaseType DetectDatabaseType(string path)
     {
         try
         {
@@ -88,6 +135,13 @@ private static DatabaseType DetectDatabaseType(string path)
             // Ignore invalid file reads during detection.
         }
 
+        // Header-based detection failed.
+        // Encrypted SQLite files have no recognisable header; use the extension
+        // as a heuristic so we can show the password dialog.
+        var ext = Path.GetExtension(path).ToLowerInvariant();
+        if (ext is ".sqlite" or ".sqlite3" or ".db" or ".db3" or ".sdb")
+            return DatabaseType.EncryptedSQLite;
+
         return DatabaseType.Unknown;
     }
 
@@ -117,11 +171,12 @@ private void ExportToExcel()
 
         _panel.ExportCurrentTableToExcel(_currentPath);
     }
+}
 
-    private enum DatabaseType
-    {
-        Unknown,
-        SQLite,
-        LiteDb,
-    }
+internal enum DatabaseType
+{
+    Unknown,
+    SQLite,
+    EncryptedSQLite,
+    LiteDb,
 }