Skip to content

Commit a039705

Browse files
committed
Replace client verification logic with extracted class in UserFlagsProvider
1 parent f4cfdba commit a039705

1 file changed

Lines changed: 3 additions & 9 deletions

File tree

WordPress/src/main/java/org/wordpress/android/userflags/provider/UserFlagsProvider.kt

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -9,17 +9,15 @@ import org.wordpress.android.provider.query.QueryResult
99
import org.wordpress.android.ui.prefs.AppPrefs.DeletablePrefKey
1010
import org.wordpress.android.ui.prefs.AppPrefs.UndeletablePrefKey
1111
import org.wordpress.android.ui.prefs.AppPrefsWrapper
12-
import org.wordpress.android.util.publicdata.JetpackPublicData
12+
import org.wordpress.android.util.publicdata.ClientVerification
1313
import org.wordpress.android.util.signature.SignatureNotFoundException
14-
import org.wordpress.android.util.signature.SignatureUtils
1514
import javax.inject.Inject
1615

1716
class 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

Comments
 (0)