520 lines
19 KiB
C#
520 lines
19 KiB
C#
#if false
|
||
using System.Diagnostics;
|
||
using System.Drawing;
|
||
using System.Drawing.Imaging;
|
||
using System.IO;
|
||
using System.Security.Cryptography;
|
||
using System.Text;
|
||
|
||
using Autodesk.Revit.DB;
|
||
using Autodesk.Revit.DB.ExtensibleStorage;
|
||
using Org.BouncyCastle.Crypto;
|
||
using Org.BouncyCastle.OpenSsl;
|
||
using OpenMcdf;
|
||
using System.Windows;
|
||
|
||
namespace ShrlAlgoToolkit.Revit.Assists;
|
||
|
||
public class PublishAssist
|
||
{
|
||
#region FileAttributes
|
||
/// <summary>
|
||
/// 设置文件为只读
|
||
/// </summary>
|
||
/// <param name="filePath"></param>
|
||
/// <param name="readOnly"></param>
|
||
public static void SetReadOnly(string filePath, bool readOnly = true)
|
||
{
|
||
try
|
||
{
|
||
if (File.Exists(filePath))
|
||
{
|
||
var info = new FileInfo(filePath);
|
||
info.IsReadOnly = readOnly;
|
||
}
|
||
else
|
||
{
|
||
Debug.WriteLine("文件不存在。");
|
||
}
|
||
}
|
||
catch (UnauthorizedAccessException)
|
||
{
|
||
Debug.WriteLine("无权限修改文件属性。");
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
Debug.WriteLine($"错误: {ex.Message}");
|
||
}
|
||
}
|
||
/// <summary>
|
||
/// 设置文件为只读
|
||
/// </summary>
|
||
/// <param name="filePath"></param>
|
||
public static void SetPrivate(string filePath)
|
||
{
|
||
try
|
||
{
|
||
if (File.Exists(filePath))
|
||
{
|
||
var info = new FileInfo(filePath);
|
||
File.SetAttributes(filePath, FileAttributes.Hidden | FileAttributes.System | FileAttributes.ReadOnly);
|
||
}
|
||
else
|
||
{
|
||
Debug.WriteLine("文件不存在。");
|
||
}
|
||
}
|
||
catch (UnauthorizedAccessException)
|
||
{
|
||
Debug.WriteLine("无权限修改文件属性。");
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
Debug.WriteLine($"错误: {ex.Message}");
|
||
}
|
||
}
|
||
|
||
#endregion
|
||
//需要与addin文件里的GUID保持一致
|
||
private static readonly Guid SchemaGuid = new Guid("A519E82B-911C-4CA5-9BC5-ED509B2055D2");
|
||
|
||
/// <summary>
|
||
/// 存储图片字节数据列表 (使用 Base64 编码)
|
||
/// </summary>
|
||
public static void StoreSignatureMap(Document doc, IDictionary<string, byte[]> signatureMap)
|
||
{
|
||
ProjectInfo projectInfo = doc.ProjectInformation;
|
||
if (projectInfo == null) return;
|
||
|
||
Schema schema = GetOrCreateSchema();
|
||
Entity entity = new Entity(schema);
|
||
IDictionary<int, string> imageMap = new Dictionary<int, string>();
|
||
try
|
||
{
|
||
IDictionary<string, string> stringMap = signatureMap
|
||
.ToDictionary(kvp => kvp.Key, kvp => Convert.ToBase64String(kvp.Value));
|
||
|
||
entity.Set<IDictionary<string, string>>("SignatureMap", stringMap);
|
||
|
||
projectInfo.SetEntity(entity);
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
MessageBox.Show(ex.Message, "错误", MessageBoxButton.OK, MessageBoxImage.Error);
|
||
}
|
||
|
||
}
|
||
|
||
/// <summary>
|
||
/// 读取图片字节数据列表 (Revit 2020 兼容版, 使用 Base64 解码)
|
||
/// </summary>
|
||
public static IDictionary<string, byte[]> RetrieveSignatureMap(Document doc)
|
||
{
|
||
ProjectInfo projectInfo = doc.ProjectInformation;
|
||
var emptyDict = new Dictionary<string, byte[]>();
|
||
if (projectInfo == null) return emptyDict;
|
||
|
||
Schema schema = Schema.Lookup(SchemaGuid);
|
||
if (schema == null) return emptyDict;
|
||
|
||
Entity entity = projectInfo.GetEntity(schema);
|
||
if (entity == null || !entity.IsValid()) return emptyDict;
|
||
IDictionary<string, string> retrievedMap = entity.Get<IDictionary<string, string>>("SignatureMap");
|
||
|
||
if (retrievedMap == null) return emptyDict;
|
||
|
||
return retrievedMap.ToDictionary(kvp => kvp.Key, kvp => Convert.FromBase64String(kvp.Value));
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取或创建数据模式 (Revit 2020 兼容版)
|
||
/// </summary>
|
||
private static Schema GetOrCreateSchema()
|
||
{
|
||
Schema schema = Schema.Lookup(SchemaGuid);
|
||
if (schema == null)
|
||
{
|
||
try
|
||
{
|
||
SchemaBuilder schemaBuilder = new SchemaBuilder(SchemaGuid);
|
||
schemaBuilder.SetReadAccessLevel(AccessLevel.Application);
|
||
schemaBuilder.SetWriteAccessLevel(AccessLevel.Application);
|
||
schemaBuilder.SetVendorId("SZMC");
|
||
schemaBuilder.SetApplicationGUID(SchemaGuid);
|
||
schemaBuilder.SetSchemaName("SignatureStorage");
|
||
|
||
schemaBuilder.AddMapField("SignatureMap", typeof(string), typeof(string));
|
||
|
||
schema = schemaBuilder.Finish();
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
MessageBox.Show(ex.Message, "创建schema错误");
|
||
}
|
||
|
||
}
|
||
return schema;
|
||
}
|
||
#region ExtensionsStorage
|
||
//private const string SchemaGuid = "{FBC67C54-781F-4D0A-B31B-8F54707889E0}";
|
||
///// <summary>
|
||
///// 计算hash256
|
||
///// </summary>
|
||
///// <param name="filePath"></param>
|
||
///// <returns></returns>
|
||
//public static string ComputeFileHash(string filePath)
|
||
//{
|
||
// using (SHA256 sha256 = SHA256.Create())
|
||
// using (FileStream stream = File.OpenRead(filePath))
|
||
// {
|
||
// byte[] hashBytes = sha256.ComputeHash(stream);
|
||
// return BitConverter.ToString(hashBytes).Replace("-", string.Empty).ToLower();
|
||
// }
|
||
//}
|
||
|
||
|
||
#endregion
|
||
#region PreviewSettings
|
||
/// <summary>
|
||
/// 完全替换缩略图
|
||
/// </summary>
|
||
/// <param name="rvtFilePath"></param>
|
||
/// <param name="imgPath"></param>
|
||
internal static void ReplaceThumb(string rvtFilePath, Bitmap bmp)
|
||
{
|
||
using var cf = new CompoundFile(rvtFilePath, CFSUpdateMode.Update, CFSConfiguration.Default);
|
||
|
||
// 获取根存储
|
||
CFStorage root = cf.RootStorage;
|
||
|
||
// 更新 Windows 预览
|
||
if (root.TryGetStream("RevitPreview4.0", out var previewStream))
|
||
{
|
||
using var outMs = new MemoryStream();
|
||
bmp.Save(outMs, ImageFormat.Png);
|
||
previewStream.SetData(outMs.ToArray());
|
||
}
|
||
|
||
// 更新 Revit 预览
|
||
if (root.TryGetStream("BasicFileInfo", out var basicFileInfo))
|
||
{
|
||
// 读取现有数据
|
||
byte[] data = basicFileInfo.GetData();
|
||
|
||
// 基本文件信息头部大小(版本等信息)
|
||
const int HEADER_SIZE = 0x2C8;
|
||
|
||
if (data.Length > HEADER_SIZE)
|
||
{
|
||
// 转换图片为字节数组
|
||
using (var ms = new MemoryStream())
|
||
{
|
||
// 转换为24位色
|
||
using (var temp = new Bitmap(bmp.Width, bmp.Height, PixelFormat.Format24bppRgb))
|
||
{
|
||
using (var g = Graphics.FromImage(temp))
|
||
{
|
||
g.DrawImage(bmp, 0, 0);
|
||
}
|
||
temp.Save(ms, ImageFormat.Bmp);
|
||
}
|
||
|
||
byte[] imageBytes = ms.ToArray();
|
||
|
||
// 跳过BMP文件头(14字节)和信息头(40字节)
|
||
int startIndex = 54;
|
||
|
||
// 计算新数据大小
|
||
int newDataSize = HEADER_SIZE + (imageBytes.Length - startIndex);
|
||
byte[] newData = new byte[newDataSize];
|
||
|
||
// 复制原有头部数据
|
||
Array.Copy(data, 0, newData, 0, HEADER_SIZE);
|
||
|
||
// 复制新的图像数据
|
||
Array.Copy(imageBytes, startIndex, newData, HEADER_SIZE, imageBytes.Length - startIndex);
|
||
|
||
// 写回流
|
||
basicFileInfo.SetData(newData);
|
||
}
|
||
}
|
||
}
|
||
|
||
cf.Commit();
|
||
}
|
||
|
||
/// <summary>
|
||
/// 合并图片
|
||
/// </summary>
|
||
/// <param name="thumbnail">缩略图</param>
|
||
/// <param name="overlay">图标</param>
|
||
/// <returns></returns>
|
||
internal static Bitmap CombineImages(Bitmap thumbnail, Bitmap overlay)
|
||
{
|
||
// 创建画布
|
||
Bitmap result = new Bitmap(thumbnail.Width, thumbnail.Height);
|
||
|
||
using (Graphics graphics = Graphics.FromImage(result))
|
||
{
|
||
// 绘制缩略图
|
||
graphics.DrawImage(thumbnail, 0, 0);
|
||
|
||
// 计算叠加位置（右下角）
|
||
int padding = 5;
|
||
int x = 0 + padding;
|
||
//int x = thumbnail.Width - overlay.Width - padding;
|
||
int y = thumbnail.Height - overlay.Height - padding;
|
||
// 保持图标原始大小或按比例缩放
|
||
int iconSize = 48;
|
||
var resizedOverlay = new Bitmap(overlay, iconSize, iconSize);
|
||
|
||
// 带透明度绘制
|
||
graphics.DrawImage(
|
||
resizedOverlay,
|
||
new System.Drawing.Rectangle(x, y, iconSize, iconSize),
|
||
new System.Drawing.Rectangle(0, 0, overlay.Width, overlay.Height),
|
||
GraphicsUnit.Pixel
|
||
);
|
||
}
|
||
return result;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 添加水印
|
||
/// </summary>
|
||
/// <param name="rvtFilePath"></param>
|
||
/// <param name="bmp"></param>
|
||
/// <param name="watermark"></param>
|
||
internal static void AddWatermark(string rvtFilePath, Bitmap bmp, string watermark)
|
||
{
|
||
// 打开 RVT 文件的复合文件结构
|
||
using var cf = new CompoundFile(rvtFilePath, CFSUpdateMode.Update, CFSConfiguration.Default);
|
||
|
||
// 获取根存储
|
||
CFStorage root = cf.RootStorage;
|
||
|
||
// 检查是否存在 "RevitPreview4.0" 流
|
||
if (!root.TryGetStream("RevitPreview4.0", out var previewStream))
|
||
{
|
||
Debug.WriteLine("未找到 RevitPreview4.0 流。");
|
||
return;
|
||
}
|
||
// 在图像上绘制半透明文字水印
|
||
if (!string.IsNullOrEmpty(watermark))
|
||
{
|
||
using var g = Graphics.FromImage(bmp);
|
||
var font = new Font("微软雅黑", 20, System.Drawing.FontStyle.Bold);
|
||
var brush = new SolidBrush(System.Drawing.Color.FromArgb(128, System.Drawing.Color.Red));
|
||
g.DrawString(watermark, font, brush, new PointF(20, bmp.Height - 80));
|
||
}
|
||
|
||
// 将修改后的图像保存为 PNG，再写回流
|
||
using var outMs = new MemoryStream();
|
||
bmp.Save(outMs, ImageFormat.Png);
|
||
|
||
previewStream.SetData(outMs.ToArray());
|
||
|
||
// 提交更改
|
||
cf.Commit();
|
||
}
|
||
#endregion
|
||
|
||
#region Hash
|
||
// 公钥（硬编码）
|
||
private const string PublicKeyPem = """
|
||
-----BEGIN PUBLIC KEY-----
|
||
MIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEAyN+Z04RmVOvzI2Z8Li1t
|
||
xB9yv80Y4Pd+M0wiuW9BHccmVlm7LVlWDRZgaAykQxIt0UskZbaIQ5eR4mqywVoM
|
||
BygfWqurVtgC6dO3SSbhZiOuSA3g1xJLgje6TOQIPWDGw1Ap0sbWrCuk+bxPFjSi
|
||
XANksY3Qfl1e6YETYEEpptTXorM8enM4368/KkvEhFswBqE5/Qf2x7XUpPAfnh3Y
|
||
cl45bUu87yDeyAnX4HEuA+aZPHEgXFPmhndtqMiUsLEajQNd8zZQHKmjMbubAq9s
|
||
l10w7smZUck+bozHO0RP31ytI6n/7I0T9PXOb/rwjeXBxXD1K0kvU3DPuq2+a5sa
|
||
iwIDAQAB
|
||
-----END PUBLIC KEY-----
|
||
""";
|
||
|
||
// 私钥（硬编码）
|
||
private const string PrivateKeyPem = """
|
||
-----BEGIN RSA PRIVATE KEY-----
|
||
MIIEpAIBAAKCAQEAyN+Z04RmVOvzI2Z8Li1txB9yv80Y4Pd+M0wiuW9BHccmVlm7
|
||
LVlWDRZgaAykQxIt0UskZbaIQ5eR4mqywVoMBygfWqurVtgC6dO3SSbhZiOuSA3g
|
||
1xJLgje6TOQIPWDGw1Ap0sbWrCuk+bxPFjSiXANksY3Qfl1e6YETYEEpptTXorM8
|
||
enM4368/KkvEhFswBqE5/Qf2x7XUpPAfnh3Ycl45bUu87yDeyAnX4HEuA+aZPHEg
|
||
XFPmhndtqMiUsLEajQNd8zZQHKmjMbubAq9sl10w7smZUck+bozHO0RP31ytI6n/
|
||
7I0T9PXOb/rwjeXBxXD1K0kvU3DPuq2+a5saiwIDAQABAoIBAACdcETZkedzXBM5
|
||
6WofTAof4bWOgovvHJCipBgOiubPKPdG5jNq/bHX2/kSnbl2eWWMQUEuGIiYJccG
|
||
/6zF6GDQ3FqHZcNbp8M4gbymNZV0TfTB0KOTyDAaojQ381LgyIK1DcWBHdw2wCRZ
|
||
vo6v1BXez4evaUy5ckXQfAA9POiAH6Im7QQlPamaUJfBCfRqFNvTcL6YQngHg8RW
|
||
4xFvN8Lxhgnqftw0dY1YzNkDBz+toQ+211LTAJWYC4vZ2sHoGvR74YxZUWPS4wms
|
||
QdIq50kRwL45Z9IZcxqp1x27W0pEwBsOO2DhnM6r7mB7Mg9BMEjUFf3oM+vrx67W
|
||
w6McEXkCgYEA8q+Bb6ldNqWNa5uLmAziEqXCnN/X9o1Cr81YIeWZpx4Ovx2aR3wH
|
||
4OtWl9VmYrU7w758nhPH27q8ADxjSbIB3ShPPdfLcKBoW/5eg+K8Bexr5Fb6Lyqo
|
||
SMhFLptxOdNzYLej99XppzRk4CV3s7IZsfQd6okSdfvlqka5ExFjXDkCgYEA0+TZ
|
||
IBsc0nCZFTlcDxxMCkmj+c4+c01BGgSeoVSKIgyp5d5DSZoNjxXsrP3Nw3uq2+fh
|
||
v4DD/V76BpzQ6v7P+Xg9YbW1QO6QdsG4Cf1bnAFYaTwQwdm2To5SXixYpVnON4Xv
|
||
4fOWVL0qrn3PK6aUTJCcEOdhx1YtBl+P4gVq9OMCgYEAhtLcDOXBhE96/rI+Xi2i
|
||
FwwBz1dISo/14WcqNEKzFzXKqYBPqrEMS3dS9y02IVZgKoIUB1oj6T2XnmXuHw4G
|
||
nQ/83fWZ7ysebyUk6w42uO+8jPGJMlT3gt/IF3mB8Mc5TR0YueIcWajw9dm645jp
|
||
T4S4bgjSubtEv1FlFmPNmHkCgYAtmq5ka5FrdVJtL/gxRFzVRf4lsI7eT/039VCS
|
||
1lgYdfNHBuwidiZ+6jATF1jmvsIKLED472S/LmbPb4bDXO4z+f3z/qCxBedPt+e9
|
||
Nqs1y1BU7dmJbyr+g0EVBaRaihaI+qmjTsLHICOEhI1HDsYSKHl+Zd40gcaU/ZJB
|
||
Cft4EwKBgQDeK/3k6OJTPuzskuqMsQy6yHd7hesxvdsfUH78tZrkRB2B9fciOa2v
|
||
ulfBss9KKVCQeKaC+91ptLpMOlSJhc9tZUsWvE19vVlNmN0ezQyfSFn+E09Ouqin
|
||
0cMfedRgHigwi+T818aFnC8qPzTMqCMQyxRTYxEC43OB9VcPyzo1yA==
|
||
-----END RSA PRIVATE KEY-----
|
||
""";
|
||
private static AsymmetricKeyParameter GetPublicKey()
|
||
{
|
||
using var reader = new StringReader(PublicKeyPem);
|
||
var pemReader = new PemReader(reader);
|
||
return (AsymmetricKeyParameter)pemReader.ReadObject();
|
||
}
|
||
|
||
private static AsymmetricKeyParameter GetPrivateKey()
|
||
{
|
||
using var reader = new StringReader(PrivateKeyPem);
|
||
var pemReader = new PemReader(reader);
|
||
var keyPair = (AsymmetricCipherKeyPair)pemReader.ReadObject();
|
||
return keyPair.Private;
|
||
}
|
||
|
||
private static byte[] ComputeFileHash(string filePath)
|
||
{
|
||
using var sha256 = SHA256.Create();
|
||
using var stream = File.OpenRead(filePath);
|
||
return sha256.ComputeHash(stream);
|
||
}
|
||
|
||
private static byte[] EncryptData(byte[] data, AsymmetricKeyParameter publicKey)
|
||
{
|
||
var rsaEngine = new Org.BouncyCastle.Crypto.Engines.RsaEngine();
|
||
rsaEngine.Init(true, publicKey);
|
||
return rsaEngine.ProcessBlock(data, 0, data.Length);
|
||
}
|
||
|
||
private static byte[] DecryptData(byte[] encryptedData, AsymmetricKeyParameter privateKey)
|
||
{
|
||
var rsaEngine = new Org.BouncyCastle.Crypto.Engines.RsaEngine();
|
||
rsaEngine.Init(false, privateKey);
|
||
return rsaEngine.ProcessBlock(encryptedData, 0, encryptedData.Length);
|
||
}
|
||
private static readonly string EncryptFileName = ".hashes.enc";
|
||
/// <summary>
|
||
/// 获取相对路径
|
||
/// </summary>
|
||
/// <param name="relativeTo">相对于某个目录</param>
|
||
/// <param name="path">示例路径</param>
|
||
/// <returns></returns>
|
||
private static string GetRelativePath(string relativeTo, string path)
|
||
{
|
||
// 确保路径以目录分隔符结尾（对于目录路径）
|
||
string from = Path.GetFullPath(relativeTo).TrimEnd(Path.DirectorySeparatorChar, Path.AltDirectorySeparatorChar) + Path.DirectorySeparatorChar;
|
||
string to = Path.GetFullPath(path);
|
||
|
||
Uri fromUri = new Uri(from);
|
||
Uri toUri = new Uri(to);
|
||
|
||
// 计算相对 URI
|
||
Uri relativeUri = fromUri.MakeRelativeUri(toUri);
|
||
// 将相对 URI 转换回字符串，并替换可能的 '/' 为当前平台的分隔符
|
||
string relativePath = Uri.UnescapeDataString(relativeUri.ToString()).Replace('/', Path.DirectorySeparatorChar);
|
||
|
||
return relativePath;
|
||
}
|
||
|
||
public static string GenerateEncFile(string directoryPath)
|
||
{
|
||
var publicKey = GetPublicKey();
|
||
var files = Directory.GetFiles(directoryPath, "*.rvt", SearchOption.AllDirectories);
|
||
var encFilePath = Path.Combine(directoryPath, EncryptFileName);
|
||
|
||
using var fs = new FileStream(encFilePath, FileMode.Create, FileAccess.Write);
|
||
using var bw = new BinaryWriter(fs);
|
||
|
||
foreach (var file in files)
|
||
{
|
||
//var fileName = Path.GetFileName(file);
|
||
var fileName = GetRelativePath(directoryPath, file);
|
||
var fileHash = ComputeFileHash(file);
|
||
var nameBytes = Encoding.UTF8.GetBytes(fileName);
|
||
var combined = nameBytes.Concat(fileHash).ToArray();
|
||
var encrypted = EncryptData(combined, publicKey);
|
||
|
||
bw.Write(nameBytes.Length);
|
||
bw.Write(nameBytes);
|
||
bw.Write(encrypted.Length);
|
||
bw.Write(encrypted);
|
||
}
|
||
return encFilePath;
|
||
}
|
||
|
||
public static StringBuilder VerifyEncFile(string directoryPath, out bool hasModified)
|
||
{
|
||
hasModified = false;
|
||
var privateKey = GetPrivateKey();
|
||
var encFilePath = Path.Combine(directoryPath, EncryptFileName);
|
||
StringBuilder sb = new StringBuilder();
|
||
|
||
if (!File.Exists(encFilePath))
|
||
{
|
||
sb.AppendLine("验证文件未找到。");
|
||
return sb;
|
||
}
|
||
Dictionary<string, string> allFiles;
|
||
try
|
||
{
|
||
//转成相对路径
|
||
allFiles = Directory.GetFiles(directoryPath, "*.rvt", SearchOption.AllDirectories)
|
||
.ToDictionary(
|
||
file => GetRelativePath(directoryPath, file),
|
||
file => file,
|
||
StringComparer.OrdinalIgnoreCase);
|
||
|
||
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
MessageBox.Show(ex.Message, "验证出错！", MessageBoxButton.OK, MessageBoxImage.Error);
|
||
return sb;
|
||
}
|
||
using var fs = new FileStream(encFilePath, FileMode.Open, FileAccess.Read);
|
||
using var br = new BinaryReader(fs);
|
||
while (fs.Position < fs.Length)
|
||
{
|
||
var nameLen = br.ReadInt32();
|
||
//if (nameLen <= 0 || nameLen > 256) // 假设文件名长度不超过256字节
|
||
//{
|
||
// Console.WriteLine("无效的文件名长度。");
|
||
// continue;
|
||
//}
|
||
var nameBytes = br.ReadBytes(nameLen);
|
||
var fileName = Encoding.UTF8.GetString(nameBytes);
|
||
|
||
var encLen = br.ReadInt32();
|
||
var encrypted = br.ReadBytes(encLen);
|
||
|
||
if (!allFiles.TryGetValue(fileName, out var actualFilePath))
|
||
{
|
||
sb.AppendLine($"{fileName} 未找到");
|
||
continue;
|
||
}
|
||
|
||
var decrypted = DecryptData(encrypted, privateKey);
|
||
var expectedName = Encoding.UTF8.GetString(decrypted, 0, nameBytes.Length);
|
||
var expectedHash = decrypted.Skip(nameBytes.Length).ToArray();
|
||
|
||
var currentHash = ComputeFileHash(actualFilePath);
|
||
bool match = expectedName == fileName && currentHash.SequenceEqual(expectedHash);
|
||
if (!match)
|
||
{
|
||
hasModified = true;
|
||
sb.AppendLine($"{fileName}");
|
||
}
|
||
}
|
||
return sb;
|
||
|
||
}
|
||
#endregion
|
||
}
|
||
|
||
#endif |