2022-06-10 21:38:30 +09:00

890 lines
22 KiB
Smali

.class public Lcom/google/firebase/ml/modeldownloader/internal/ModelFileManager;
.super Ljava/lang/Object;
.source "ModelFileManager.java"
# static fields
.field public static final CUSTOM_MODEL_ROOT_PATH:Ljava/lang/String; = "com.google.firebase.ml.custom.models"
.field private static final INVALID_INDEX:I = -0x1
.field private static final TAG:Ljava/lang/String; = "FirebaseModelFileManage"
# instance fields
.field private final context:Landroid/content/Context;
.field private final firebaseApp:Lcom/google/firebase/FirebaseApp;
.field private final sharedPreferencesUtil:Lcom/google/firebase/ml/modeldownloader/internal/SharedPreferencesUtil;
# direct methods
.method public constructor <init>(Lcom/google/firebase/FirebaseApp;)V
.locals 1
.param p1 # Lcom/google/firebase/FirebaseApp;
.annotation build Landroidx/annotation/NonNull;
.end annotation
.end param
.line 1
invoke-direct {p0}, Ljava/lang/Object;-><init>()V
.line 2
invoke-virtual {p1}, Lcom/google/firebase/FirebaseApp;->getApplicationContext()Landroid/content/Context;
move-result-object v0
iput-object v0, p0, Lcom/google/firebase/ml/modeldownloader/internal/ModelFileManager;->context:Landroid/content/Context;
.line 3
iput-object p1, p0, Lcom/google/firebase/ml/modeldownloader/internal/ModelFileManager;->firebaseApp:Lcom/google/firebase/FirebaseApp;
.line 4
new-instance v0, Lcom/google/firebase/ml/modeldownloader/internal/SharedPreferencesUtil;
invoke-direct {v0, p1}, Lcom/google/firebase/ml/modeldownloader/internal/SharedPreferencesUtil;-><init>(Lcom/google/firebase/FirebaseApp;)V
iput-object v0, p0, Lcom/google/firebase/ml/modeldownloader/internal/ModelFileManager;->sharedPreferencesUtil:Lcom/google/firebase/ml/modeldownloader/internal/SharedPreferencesUtil;
return-void
.end method
.method public static getInstance()Lcom/google/firebase/ml/modeldownloader/internal/ModelFileManager;
.locals 2
.annotation build Landroidx/annotation/NonNull;
.end annotation
.line 1
invoke-static {}, Lcom/google/firebase/FirebaseApp;->getInstance()Lcom/google/firebase/FirebaseApp;
move-result-object v0
const-class v1, Lcom/google/firebase/ml/modeldownloader/internal/ModelFileManager;
invoke-virtual {v0, v1}, Lcom/google/firebase/FirebaseApp;->get(Ljava/lang/Class;)Ljava/lang/Object;
move-result-object v0
check-cast v0, Lcom/google/firebase/ml/modeldownloader/internal/ModelFileManager;
return-object v0
.end method
.method public static getInstance(Lcom/google/firebase/FirebaseApp;)Lcom/google/firebase/ml/modeldownloader/internal/ModelFileManager;
.locals 1
.param p0 # Lcom/google/firebase/FirebaseApp;
.annotation build Landroidx/annotation/NonNull;
.end annotation
.end param
.annotation build Landroidx/annotation/NonNull;
.end annotation
.line 2
const-class v0, Lcom/google/firebase/ml/modeldownloader/internal/ModelFileManager;
invoke-virtual {p0, v0}, Lcom/google/firebase/FirebaseApp;->get(Ljava/lang/Class;)Ljava/lang/Object;
move-result-object p0
check-cast p0, Lcom/google/firebase/ml/modeldownloader/internal/ModelFileManager;
return-object p0
.end method
.method private getLatestCachedModelVersion(Ljava/io/File;)I
.locals 6
.param p1 # Ljava/io/File;
.annotation build Landroidx/annotation/NonNull;
.end annotation
.end param
.annotation build Landroidx/annotation/WorkerThread;
.end annotation
.line 1
invoke-virtual {p1}, Ljava/io/File;->listFiles()[Ljava/io/File;
move-result-object p1
const/4 v0, -0x1
if-eqz p1, :cond_1
.line 2
array-length v1, p1
if-nez v1, :cond_0
goto :goto_2
.line 3
:cond_0
array-length v1, p1
const/4 v2, 0x0
move v3, v2
:goto_0
if-ge v3, v1, :cond_1
aget-object v4, p1, v3
.line 4
:try_start_0
invoke-virtual {v4}, Ljava/io/File;->getName()Ljava/lang/String;
move-result-object v5
invoke-static {v5}, Ljava/lang/Integer;->parseInt(Ljava/lang/String;)I
move-result v5
invoke-static {v0, v5}, Ljava/lang/Math;->max(II)I
move-result v0
:try_end_0
.catch Ljava/lang/NumberFormatException; {:try_start_0 .. :try_end_0} :catch_0
goto :goto_1
:catch_0
const/4 v5, 0x1
new-array v5, v5, [Ljava/lang/Object;
.line 5
invoke-virtual {v4}, Ljava/io/File;->getName()Ljava/lang/String;
move-result-object v4
aput-object v4, v5, v2
const-string v4, "Contains non-integer file name %s"
invoke-static {v4, v5}, Ljava/lang/String;->format(Ljava/lang/String;[Ljava/lang/Object;)Ljava/lang/String;
move-result-object v4
const-string v5, "FirebaseModelFileManage"
invoke-static {v5, v4}, Landroid/util/Log;->d(Ljava/lang/String;Ljava/lang/String;)I
:goto_1
add-int/lit8 v3, v3, 0x1
goto :goto_0
:cond_1
:goto_2
return v0
.end method
.method private getModelDirUnsafe(Ljava/lang/String;)Ljava/io/File;
.locals 3
.param p1 # Ljava/lang/String;
.annotation build Landroidx/annotation/NonNull;
.end annotation
.end param
.annotation build Landroidx/annotation/Nullable;
.end annotation
.line 1
new-instance v0, Ljava/io/File;
iget-object v1, p0, Lcom/google/firebase/ml/modeldownloader/internal/ModelFileManager;->context:Landroid/content/Context;
invoke-virtual {v1}, Landroid/content/Context;->getNoBackupFilesDir()Ljava/io/File;
move-result-object v1
const-string v2, "com.google.firebase.ml.custom.models"
invoke-direct {v0, v1, v2}, Ljava/io/File;-><init>(Ljava/io/File;Ljava/lang/String;)V
.line 2
new-instance v1, Ljava/io/File;
iget-object v2, p0, Lcom/google/firebase/ml/modeldownloader/internal/ModelFileManager;->firebaseApp:Lcom/google/firebase/FirebaseApp;
invoke-virtual {v2}, Lcom/google/firebase/FirebaseApp;->getPersistenceKey()Ljava/lang/String;
move-result-object v2
invoke-direct {v1, v0, v2}, Ljava/io/File;-><init>(Ljava/io/File;Ljava/lang/String;)V
.line 3
new-instance v0, Ljava/io/File;
invoke-direct {v0, v1, p1}, Ljava/io/File;-><init>(Ljava/io/File;Ljava/lang/String;)V
return-object v0
.end method
# virtual methods
.method public declared-synchronized deleteAllModels(Ljava/lang/String;)Z
.locals 0
.param p1 # Ljava/lang/String;
.annotation build Landroidx/annotation/NonNull;
.end annotation
.end param
.annotation build Landroidx/annotation/WorkerThread;
.end annotation
monitor-enter p0
.line 1
:try_start_0
invoke-direct {p0, p1}, Lcom/google/firebase/ml/modeldownloader/internal/ModelFileManager;->getModelDirUnsafe(Ljava/lang/String;)Ljava/io/File;
move-result-object p1
.line 2
invoke-virtual {p0, p1}, Lcom/google/firebase/ml/modeldownloader/internal/ModelFileManager;->deleteRecursively(Ljava/io/File;)Z
move-result p1
:try_end_0
.catchall {:try_start_0 .. :try_end_0} :catchall_0
monitor-exit p0
return p1
:catchall_0
move-exception p1
monitor-exit p0
throw p1
.end method
.method public deleteNonLatestCustomModels()V
.locals 5
.annotation system Ldalvik/annotation/Throws;
value = {
Lcom/google/firebase/ml/modeldownloader/FirebaseMlException;
}
.end annotation
const-string v0, ""
.line 1
invoke-virtual {p0, v0}, Lcom/google/firebase/ml/modeldownloader/internal/ModelFileManager;->getDirImpl(Ljava/lang/String;)Ljava/io/File;
move-result-object v0
.line 2
invoke-virtual {v0}, Ljava/io/File;->isDirectory()Z
move-result v1
if-eqz v1, :cond_1
.line 3
invoke-virtual {v0}, Ljava/io/File;->listFiles()[Ljava/io/File;
move-result-object v0
array-length v1, v0
const/4 v2, 0x0
:goto_0
if-ge v2, v1, :cond_1
aget-object v3, v0, v2
.line 4
invoke-virtual {v3}, Ljava/io/File;->getName()Ljava/lang/String;
move-result-object v3
.line 5
iget-object v4, p0, Lcom/google/firebase/ml/modeldownloader/internal/ModelFileManager;->sharedPreferencesUtil:Lcom/google/firebase/ml/modeldownloader/internal/SharedPreferencesUtil;
invoke-virtual {v4, v3}, Lcom/google/firebase/ml/modeldownloader/internal/SharedPreferencesUtil;->getCustomModelDetails(Ljava/lang/String;)Lcom/google/firebase/ml/modeldownloader/CustomModel;
move-result-object v4
if-eqz v4, :cond_0
.line 6
invoke-virtual {v4}, Lcom/google/firebase/ml/modeldownloader/CustomModel;->getLocalFilePath()Ljava/lang/String;
move-result-object v4
invoke-virtual {p0, v3, v4}, Lcom/google/firebase/ml/modeldownloader/internal/ModelFileManager;->deleteOldModels(Ljava/lang/String;Ljava/lang/String;)V
:cond_0
add-int/lit8 v2, v2, 0x1
goto :goto_0
:cond_1
return-void
.end method
.method public declared-synchronized deleteOldModels(Ljava/lang/String;Ljava/lang/String;)V
.locals 7
.param p1 # Ljava/lang/String;
.annotation build Landroidx/annotation/NonNull;
.end annotation
.end param
.param p2 # Ljava/lang/String;
.annotation build Landroidx/annotation/NonNull;
.end annotation
.end param
.annotation build Landroidx/annotation/WorkerThread;
.end annotation
monitor-enter p0
.line 1
:try_start_0
invoke-direct {p0, p1}, Lcom/google/firebase/ml/modeldownloader/internal/ModelFileManager;->getModelDirUnsafe(Ljava/lang/String;)Ljava/io/File;
move-result-object p1
.line 2
invoke-virtual {p1}, Ljava/io/File;->exists()Z
move-result v0
:try_end_0
.catchall {:try_start_0 .. :try_end_0} :catchall_0
if-nez v0, :cond_0
.line 3
monitor-exit p0
return-void
.line 4
:cond_0
:try_start_1
new-instance v0, Ljava/io/File;
invoke-direct {v0, p2}, Ljava/io/File;-><init>(Ljava/lang/String;)V
.line 5
invoke-virtual {v0}, Ljava/io/File;->getName()Ljava/lang/String;
move-result-object p2
invoke-static {p2}, Ljava/lang/Integer;->parseInt(Ljava/lang/String;)I
move-result p2
.line 6
invoke-virtual {p1}, Ljava/io/File;->listFiles()[Ljava/io/File;
move-result-object p1
.line 7
array-length v0, p1
const/4 v1, 0x0
const/4 v2, 0x1
move v3, v1
move v4, v2
:goto_0
if-ge v3, v0, :cond_3
aget-object v5, p1, v3
:try_end_1
.catchall {:try_start_1 .. :try_end_1} :catchall_0
.line 8
:try_start_2
invoke-virtual {v5}, Ljava/io/File;->getName()Ljava/lang/String;
move-result-object v6
invoke-static {v6}, Ljava/lang/Integer;->parseInt(Ljava/lang/String;)I
move-result v6
:try_end_2
.catch Ljava/lang/NumberFormatException; {:try_start_2 .. :try_end_2} :catch_0
.catchall {:try_start_2 .. :try_end_2} :catchall_0
goto :goto_1
:catch_0
const v6, 0x7fffffff
:goto_1
if-ge v6, p2, :cond_2
if-eqz v4, :cond_1
.line 9
:try_start_3
invoke-virtual {v5}, Ljava/io/File;->delete()Z
move-result v4
:try_end_3
.catchall {:try_start_3 .. :try_end_3} :catchall_0
if-eqz v4, :cond_1
move v4, v2
goto :goto_2
:cond_1
move v4, v1
:cond_2
:goto_2
add-int/lit8 v3, v3, 0x1
goto :goto_0
.line 10
:cond_3
monitor-exit p0
return-void
:catchall_0
move-exception p1
monitor-exit p0
throw p1
.end method
.method public deleteRecursively(Ljava/io/File;)Z
.locals 7
.param p1 # Ljava/io/File;
.annotation build Landroidx/annotation/Nullable;
.end annotation
.end param
const/4 v0, 0x0
if-nez p1, :cond_0
return v0
.line 1
:cond_0
invoke-virtual {p1}, Ljava/io/File;->isDirectory()Z
move-result v1
const/4 v2, 0x1
if-eqz v1, :cond_2
.line 2
invoke-virtual {p1}, Ljava/io/File;->listFiles()[Ljava/io/File;
move-result-object v1
array-length v3, v1
move v4, v0
move v5, v2
:goto_0
if-ge v4, v3, :cond_3
aget-object v6, v1, v4
if-eqz v5, :cond_1
.line 3
invoke-virtual {p0, v6}, Lcom/google/firebase/ml/modeldownloader/internal/ModelFileManager;->deleteRecursively(Ljava/io/File;)Z
move-result v5
if-eqz v5, :cond_1
move v5, v2
goto :goto_1
:cond_1
move v5, v0
:goto_1
add-int/lit8 v4, v4, 0x1
goto :goto_0
:cond_2
move v5, v2
:cond_3
if-eqz v5, :cond_4
.line 4
invoke-virtual {p1}, Ljava/io/File;->delete()Z
move-result p1
if-eqz p1, :cond_4
move v0, v2
:cond_4
return v0
.end method
.method public getDirImpl(Ljava/lang/String;)Ljava/io/File;
.locals 3
.param p1 # Ljava/lang/String;
.annotation build Landroidx/annotation/NonNull;
.end annotation
.end param
.annotation build Landroidx/annotation/VisibleForTesting;
.end annotation
.annotation build Landroidx/annotation/WorkerThread;
.end annotation
.annotation system Ldalvik/annotation/Throws;
value = {
Lcom/google/firebase/ml/modeldownloader/FirebaseMlException;
}
.end annotation
.line 1
invoke-direct {p0, p1}, Lcom/google/firebase/ml/modeldownloader/internal/ModelFileManager;->getModelDirUnsafe(Ljava/lang/String;)Ljava/io/File;
move-result-object p1
.line 2
invoke-virtual {p1}, Ljava/io/File;->exists()Z
move-result v0
if-nez v0, :cond_1
.line 3
invoke-virtual {p1}, Ljava/io/File;->mkdirs()Z
move-result v0
if-eqz v0, :cond_0
goto :goto_0
.line 4
:cond_0
new-instance v0, Lcom/google/firebase/ml/modeldownloader/FirebaseMlException;
new-instance v1, Ljava/lang/StringBuilder;
invoke-direct {v1}, Ljava/lang/StringBuilder;-><init>()V
const-string v2, "Failed to create model folder: "
invoke-virtual {v1, v2}, Ljava/lang/StringBuilder;->append(Ljava/lang/String;)Ljava/lang/StringBuilder;
invoke-virtual {v1, p1}, Ljava/lang/StringBuilder;->append(Ljava/lang/Object;)Ljava/lang/StringBuilder;
invoke-virtual {v1}, Ljava/lang/StringBuilder;->toString()Ljava/lang/String;
move-result-object p1
const/16 v1, 0xd
invoke-direct {v0, p1, v1}, Lcom/google/firebase/ml/modeldownloader/FirebaseMlException;-><init>(Ljava/lang/String;I)V
throw v0
.line 5
:cond_1
invoke-virtual {p1}, Ljava/io/File;->isDirectory()Z
move-result v0
if-eqz v0, :cond_2
:goto_0
return-object p1
.line 6
:cond_2
new-instance v0, Lcom/google/firebase/ml/modeldownloader/FirebaseMlException;
new-instance v1, Ljava/lang/StringBuilder;
invoke-direct {v1}, Ljava/lang/StringBuilder;-><init>()V
const-string v2, "Can not create model folder, since an existing file has the same name: "
invoke-virtual {v1, v2}, Ljava/lang/StringBuilder;->append(Ljava/lang/String;)Ljava/lang/StringBuilder;
invoke-virtual {v1, p1}, Ljava/lang/StringBuilder;->append(Ljava/lang/Object;)Ljava/lang/StringBuilder;
invoke-virtual {v1}, Ljava/lang/StringBuilder;->toString()Ljava/lang/String;
move-result-object p1
const/4 v1, 0x6
invoke-direct {v0, p1, v1}, Lcom/google/firebase/ml/modeldownloader/FirebaseMlException;-><init>(Ljava/lang/String;I)V
throw v0
.end method
.method public getModelFileDestination(Lcom/google/firebase/ml/modeldownloader/CustomModel;)Ljava/io/File;
.locals 2
.param p1 # Lcom/google/firebase/ml/modeldownloader/CustomModel;
.annotation build Landroidx/annotation/NonNull;
.end annotation
.end param
.annotation build Landroidx/annotation/Nullable;
.end annotation
.annotation build Landroidx/annotation/VisibleForTesting;
.end annotation
.annotation system Ldalvik/annotation/Throws;
value = {
Lcom/google/firebase/ml/modeldownloader/FirebaseMlException;
}
.end annotation
.line 1
invoke-virtual {p1}, Lcom/google/firebase/ml/modeldownloader/CustomModel;->getName()Ljava/lang/String;
move-result-object p1
invoke-virtual {p0, p1}, Lcom/google/firebase/ml/modeldownloader/internal/ModelFileManager;->getDirImpl(Ljava/lang/String;)Ljava/io/File;
move-result-object p1
.line 2
invoke-direct {p0, p1}, Lcom/google/firebase/ml/modeldownloader/internal/ModelFileManager;->getLatestCachedModelVersion(Ljava/io/File;)I
move-result v0
.line 3
new-instance v1, Ljava/io/File;
add-int/lit8 v0, v0, 0x1
invoke-static {v0}, Ljava/lang/String;->valueOf(I)Ljava/lang/String;
move-result-object v0
invoke-direct {v1, p1, v0}, Ljava/io/File;-><init>(Ljava/io/File;Ljava/lang/String;)V
return-object v1
.end method
.method public declared-synchronized moveModelToDestinationFolder(Lcom/google/firebase/ml/modeldownloader/CustomModel;Landroid/os/ParcelFileDescriptor;)Ljava/io/File;
.locals 4
.param p1 # Lcom/google/firebase/ml/modeldownloader/CustomModel;
.annotation build Landroidx/annotation/NonNull;
.end annotation
.end param
.param p2 # Landroid/os/ParcelFileDescriptor;
.annotation build Landroidx/annotation/NonNull;
.end annotation
.end param
.annotation build Landroidx/annotation/Nullable;
.end annotation
.annotation build Landroidx/annotation/WorkerThread;
.end annotation
.annotation system Ldalvik/annotation/Throws;
value = {
Lcom/google/firebase/ml/modeldownloader/FirebaseMlException;
}
.end annotation
monitor-enter p0
.line 1
:try_start_0
invoke-virtual {p0, p1}, Lcom/google/firebase/ml/modeldownloader/internal/ModelFileManager;->getModelFileDestination(Lcom/google/firebase/ml/modeldownloader/CustomModel;)Ljava/io/File;
move-result-object p1
.line 2
invoke-virtual {p1}, Ljava/io/File;->getParentFile()Ljava/io/File;
move-result-object v0
.line 3
invoke-virtual {v0}, Ljava/io/File;->exists()Z
move-result v1
if-nez v1, :cond_0
.line 4
invoke-virtual {v0}, Ljava/io/File;->mkdirs()Z
:try_end_0
.catchall {:try_start_0 .. :try_end_0} :catchall_4
.line 5
:cond_0
:try_start_1
new-instance v0, Landroid/os/ParcelFileDescriptor$AutoCloseInputStream;
invoke-direct {v0, p2}, Landroid/os/ParcelFileDescriptor$AutoCloseInputStream;-><init>(Landroid/os/ParcelFileDescriptor;)V
:try_end_1
.catch Ljava/io/IOException; {:try_start_1 .. :try_end_1} :catch_0
.catchall {:try_start_1 .. :try_end_1} :catchall_4
.line 6
:try_start_2
new-instance p2, Ljava/io/FileOutputStream;
invoke-direct {p2, p1}, Ljava/io/FileOutputStream;-><init>(Ljava/io/File;)V
:try_end_2
.catchall {:try_start_2 .. :try_end_2} :catchall_2
const/16 v1, 0x1000
:try_start_3
new-array v1, v1, [B
.line 7
:goto_0
invoke-virtual {v0, v1}, Ljava/io/FileInputStream;->read([B)I
move-result v2
const/4 v3, -0x1
if-eq v2, v3, :cond_1
const/4 v3, 0x0
.line 8
invoke-virtual {p2, v1, v3, v2}, Ljava/io/FileOutputStream;->write([BII)V
goto :goto_0
.line 9
:cond_1
invoke-virtual {p2}, Ljava/io/FileOutputStream;->getFD()Ljava/io/FileDescriptor;
move-result-object v1
invoke-virtual {v1}, Ljava/io/FileDescriptor;->sync()V
:try_end_3
.catchall {:try_start_3 .. :try_end_3} :catchall_0
.line 10
:try_start_4
invoke-virtual {p2}, Ljava/io/FileOutputStream;->close()V
:try_end_4
.catchall {:try_start_4 .. :try_end_4} :catchall_2
:try_start_5
invoke-virtual {v0}, Ljava/io/FileInputStream;->close()V
:try_end_5
.catch Ljava/io/IOException; {:try_start_5 .. :try_end_5} :catch_0
.catchall {:try_start_5 .. :try_end_5} :catchall_4
.line 11
monitor-exit p0
return-object p1
:catchall_0
move-exception v1
.line 12
:try_start_6
invoke-virtual {p2}, Ljava/io/FileOutputStream;->close()V
:try_end_6
.catchall {:try_start_6 .. :try_end_6} :catchall_1
goto :goto_1
:catchall_1
move-exception p2
:try_start_7
invoke-virtual {v1, p2}, Ljava/lang/Throwable;->addSuppressed(Ljava/lang/Throwable;)V
:goto_1
throw v1
:try_end_7
.catchall {:try_start_7 .. :try_end_7} :catchall_2
:catchall_2
move-exception p2
:try_start_8
invoke-virtual {v0}, Ljava/io/FileInputStream;->close()V
:try_end_8
.catchall {:try_start_8 .. :try_end_8} :catchall_3
goto :goto_2
:catchall_3
move-exception v0
:try_start_9
invoke-virtual {p2, v0}, Ljava/lang/Throwable;->addSuppressed(Ljava/lang/Throwable;)V
:goto_2
throw p2
:try_end_9
.catch Ljava/io/IOException; {:try_start_9 .. :try_end_9} :catch_0
.catchall {:try_start_9 .. :try_end_9} :catchall_4
:catch_0
move-exception p2
:try_start_a
const-string v0, "FirebaseModelFileManage"
.line 13
new-instance v1, Ljava/lang/StringBuilder;
invoke-direct {v1}, Ljava/lang/StringBuilder;-><init>()V
const-string v2, "Failed to copy downloaded model file to destination folder: "
invoke-virtual {v1, v2}, Ljava/lang/StringBuilder;->append(Ljava/lang/String;)Ljava/lang/StringBuilder;
invoke-virtual {p2}, Ljava/io/IOException;->toString()Ljava/lang/String;
move-result-object p2
invoke-virtual {v1, p2}, Ljava/lang/StringBuilder;->append(Ljava/lang/String;)Ljava/lang/StringBuilder;
invoke-virtual {v1}, Ljava/lang/StringBuilder;->toString()Ljava/lang/String;
move-result-object p2
invoke-static {v0, p2}, Landroid/util/Log;->d(Ljava/lang/String;Ljava/lang/String;)I
.line 14
invoke-virtual {p1}, Ljava/io/File;->delete()Z
:try_end_a
.catchall {:try_start_a .. :try_end_a} :catchall_4
const/4 p1, 0x0
.line 15
monitor-exit p0
return-object p1
:catchall_4
move-exception p1
monitor-exit p0
throw p1
.end method