短信过滤 APP 开发

本文字数：7033字

预计阅读时间：42分钟

一直想开发一个自己的短信过滤 APP，但是一直没有具体实施，现在终于静下心来，边开发边记录下整体的开发过程。

垃圾短信样本

遇到的第一个问题是，既然要过滤垃圾短信，那首先要识别哪些是垃圾短信？如何识别呢？

参考之前训练识别钢管计数的经验，决定通过 CoreML 训练 Text 模型来识别，那问题来了，要训练模型的短信数据集怎么来？

一开始打算网上找到垃圾短信样本，但找了好久没找到，于是就想到用自己和家人手机里的短信，毕竟手机里短信一般不删除，也有小几千条，而且垃圾短信、推销、广告之类的应有尽有。

所以问题就变成了，如何导出 iPhone 短信？

这里笔者也查了好久，找到的第三方软件基本都是需要收费，最终发现了一个免费导出的方案。

首先不加密备份手机到电脑，如下图，选中Back up all the data on your iPhone to this Mac，点击Back Up Now，等待备份完成，备份完成后，再点击Manage Backups ：

备份界面

Manage Backups 点击后，界面如下，可以看到已备份的记录，右键选择 Show In Finder，在文件夹中打开：

管理备份

然后可以看到备份所在目录已打开，这时候需要找到文件名为 3d0d7e5fb2ce288813306e4d4636395e047a3d28的文件，这个文件就是短信备份的数据库文件。然后问题来了，怎么找呢？看到备份目录一个个文件夹是不是懵，这怎么找，很简单，搜索，点击右上角的搜索，直接把这个文件名输入即可，注意搜索的范围是当前文件夹：

备份文件夹

搜索结果如下：

短信备份数据库文件

然后把这个文件单独拷贝到另一个地方，比如桌面，再用数据库软件打开，比如SQLPro for SQLLite，打开如下：

短信数据库文件打开

然后观察这个文件后发现，手机号和短信记录分布在不同表中，需要写一个 SQL 查出需要的内容，SQL 内容如下，参考 SQL to extract messages from backup，选中上图中Query，输入命令如下：

SELECT datetime(message.date, 'unixepoch', '+31 years', '-6 hours') as Timestamp, handle.id, message.text,
    case when message.is_from_me then 'From me' else 'To me' end as Sender
FROM message, handle WHERE message.handle_id = handle.ROWID AND message.text NOT NULL;

然后点击右上角的执行, 可以看到，把短信都筛选出来了：

然后选中所有 row，右键选择Export result set as 导出CSV，即可导出 excel 格式的文件：

导出短信

这样就获取到了所需的短信样本。

垃圾短信训练识别

有了样本之后，再来看如何训练识别，打算使用苹果的 CoreML识别，那么如何使用？样本格式的要求是什么样？训练需要多久？

先来看，创建一个文字训练的CoreML工程，选中 Xcode，点击Open Developer Tool，选中 CoreML 打开，如下图：

XcodeDevelperTool

然后选择文件夹，并点击新建New Document, 如下：

New Document

然后选中 Text Classification，如下图：

Text Classification

接着输入项目的名字和描述：

项目描述

点击右下角创建，进入主界面，如下：

主界面

点击 Traing Data 的详细说明，可以看到 CoreML 要求的文字识别的格式，支持 JSON 和 CSV 文件，格式如下：

格式

JSON 格式如下：

// JSON file
[
    {
        "text": "The movie was fantastic!",
        "label": "positive"
    }, {
        "text": "Very boring. Fell asleep.",
        "label": "negative"
    }, {
        "text": "It was just OK.",
        "label": "neutral"
    } ...
]

而 CSV 格式则是，一列 text ，一列 label，

text	label
这是一条普通短信	label1
这是一条垃圾短信	label2

由于再前一步中，已经将短信导出为 CSV 格式，所以这里就需要把格式改为上图中格式即可，只剩下一个问题需要解决，即：label 有哪些取值？

要看 label 有哪些取值，需要先看系统短信的过滤逻辑是什么样？支持的过滤分类有哪些？否则自己想实现的分类，分组好了，最后发现系统不支持就尴尬了。

短信过滤分类

系统短信的过滤逻辑

参考 SMS and MMS Message Filtering，可以看到，开发者是没有权限创建新分组的，只能是针对收到未知联系人的 SMS 或者 MMS ，拦截返回指定的分类。

这里需要注意的是，根据文档的说法，短信过滤不支持 iMessage 和通讯录中联系人短信的过滤，仅支持未知联系人的 SMS 和 MMS。

短信过滤，又分为本地判断过滤和服务端判断过滤，示意图如下：

短信本地过滤

短信服务端过滤

按照文档的说法，即使是服务端过滤，APP 也是不能直接访问网络的，系统会和设置的服务器交互；而且 App Extension 不能通过共享 Group 写数据，故而短信仅能在 App Extension中获取到，不能存储，不能上传，从而保证隐私和安全。关于服务端过滤更多的实现，可以参考 Creating a Message Filter App Extension。

再来看支持的过滤类型，ILMessageFilterAction大分类支持五种：

none 没有足够信息，不能判断，会展示信息，或进一步请求服务端判断过滤
allow 正常展示信息
junk 阻止正常展示信息，显示在垃圾短信分类下
promotion 阻止正常展示信息，显示在推送信息分类下
transation 阻止正常展示信息，显示在交易信息分类下

而其中又可以细分子分类，ILMessageFilterSubAction，具体含义可以参考ILMessageFilterSubAction：

none
promotion 支持的子分类有
- others
- offers
- coupons
transation 支持的子分类有
- others
- finance
- orders
- reminders
- health
- weather
- carrier
- rewards
- publicServices

这里仅针对大分类做处理，具体的子分类不做详细过滤，所以需要训练的 label 有哪些取值就很明确了，过滤垃圾短信、推广信息、交易信息，至于 none 和 allow 不做区分，统一处理为 allow，所以总共需要训练的 label 取值有以下这些：

allow
junk
promotion
transation

然后就是针对导出短信的 CSV 文件，针对每条短信，添加对应的 label，这里只能手工，样本的大小和 label 定义决定后续识别的准确度，同时为了后续子分类的实现，建议实事求是，不要把比如 promotion 里的分到 junk 里。。。

每条短信样本都标记好了之后，就可以导入Create ML来训练，生成需要的模型，步骤如下——

首先导入数据集：

导入数据集

然后点击左上角的 Train：

训练Train

等训练好了之后，可以点击 Preview ，模拟短信文本，看输出的预测，如下图:

效果检测

最后，导出模型，供 APP 使用：

导出模型

APP开发

新建项目，然后使用 "new bing生成图片" 来设计 APPIcon，再用 ChatGPT-4，来生成 APP 名字。然后添加 Message Filter Extension Target，如下图：

Message Filter Extension Target

在 MessageFilterExtension.swift 中，能看到苹果已经帮忙实现了基本的框架，只需要在框架对应// TODO: 的地方，加入对应的过滤逻辑即可。

然后导入训练结果集到项目中，注意 Target 要勾选主工程和 Message Filter Extension 的 Target，因为需要在这个 Target 中使用模型来实现过滤。

具体使用如下：

import Foundation
import IdentityLookup
import CoreML

import IdentityLookup

enum SMSFilterActionType: String {
    case transation
    case promotion
    case allow
    case junk
    
    func formatFilterAction() -> ILMessageFilterAction {
        switch self {
        case .transation:
            return ILMessageFilterAction.transaction
        case .promotion:
            return ILMessageFilterAction.promotion
        case .allow:
            return ILMessageFilterAction.allow
        case .junk:
            return ILMessageFilterAction.junk
        }
    }
}

struct SMSFilterUtil {
    static func filter(with messageBody: String) -> ILMessageFilterAction {
        var filterAction: ILMessageFilterAction = .none
        let configuration = MLModelConfiguration()
        do {
            let model = try SmsClassifier(configuration: configuration)
            let resultLabel = try model.prediction(text: messageBody).label
            if let resultFilterAction = SMSFilterActionType(rawValue: resultLabel)?.formatFilterAction() {
                filterAction = resultFilterAction
            }
        } catch {
            print(error)
        }
        return filterAction
    }
}

然后在MessageFilterExtension.Swift 中 offlineAction(for queryRequest: ILMessageFilterQueryRequest) -> (ILMessageFilterAction, ILMessageFilterSubAction) 方法调用，如下：

@available(iOSApplicationExtension 16.0, *)
  private func offlineAction(for queryRequest: ILMessageFilterQueryRequest) -> (ILMessageFilterAction, ILMessageFilterSubAction) {
      guard let messageBody = queryRequest.messageBody else {
          return (.none, .none)
      }
      let action = MWSMSFilterUtil.filter(with: messageBody)
      return (action, .none)
  }

这里需要注意下 APP 最低版本设置，ILMessageFilterSubAction只有 iOS 16 以上的手机才支持，而 ILMessageFilterSubAction 则是 iOS 14 以上。

如果想实现更精细的 SubAction 的过滤，则上面短信数据集的 label需要改为更精细的 label，然后训练出模型，再用来判断。

另外，ILMessageFilterQueryRequest 中可以获取到 sender 和 messageBody，所以如果想实现自定义规则，比如针对某个手机号设置对应的规则，则需要从 APP 中设置对应的规则，然后通过 Group 共享到 Extension ，然后在上面的方法里通过规则匹配。

总结

相信通过上面的步骤，大家都能开发出自己的短信过滤 APP。

上面的步骤是通过固定的训练模型来匹配的逻辑，步骤是：