@@ -9,17 +9,15 @@ import org.wordpress.android.provider.query.QueryResult
99import org.wordpress.android.ui.prefs.AppPrefs.DeletablePrefKey
1010import org.wordpress.android.ui.prefs.AppPrefs.UndeletablePrefKey
1111import org.wordpress.android.ui.prefs.AppPrefsWrapper
12- import org.wordpress.android.util.publicdata.JetpackPublicData
12+ import org.wordpress.android.util.publicdata.ClientVerification
1313import org.wordpress.android.util.signature.SignatureNotFoundException
14- import org.wordpress.android.util.signature.SignatureUtils
1514import javax.inject.Inject
1615
1716class UserFlagsProvider : QueryContentProvider () {
1817 @Inject lateinit var appPrefsWrapper: AppPrefsWrapper
1918 @Inject lateinit var siteStore: SiteStore
20- @Inject lateinit var signatureUtils: SignatureUtils
2119 @Inject lateinit var queryResult: QueryResult
22- @Inject lateinit var jetpackPublicData : JetpackPublicData
20+ @Inject lateinit var clientVerification : ClientVerification
2321
2422 private val userFlagsKeysSet: Set <String > = setOf (
2523 DeletablePrefKey .MAIN_PAGE_INDEX .name,
@@ -70,11 +68,7 @@ class UserFlagsProvider : QueryContentProvider() {
7068 inject()
7169 return context?.let {
7270 try {
73- val callerPackageId = callingPackage
74- val callerExpectedPackageId = jetpackPublicData.currentPackageId()
75- val callerSignatureHash = signatureUtils.getSignatureHash(it, callerExpectedPackageId)
76- val callerExpectedSignatureHash = jetpackPublicData.currentPublicKeyHash()
77- if (callerPackageId == callerExpectedPackageId && callerSignatureHash == callerExpectedSignatureHash) {
71+ if (clientVerification.canTrust(callingPackage)) {
7872 val userFlagsMap = appPrefsWrapper.getAllPrefs()
7973 .filter { entry ->
8074 userFlagsKeysSet.contains(entry.key)
0 commit comments