From 1d993dc647096f3d2cac40ffb79d3c44cb1191dd Mon Sep 17 00:00:00 2001
From: Ali <>
Date: Tue, 21 Feb 2023 22:16:08 +0400
Subject: [PATCH] Fail early if the app can't store data

---
 .../TelegramUI/Sources/AppDelegate.swift      | 34 +++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/submodules/TelegramUI/Sources/AppDelegate.swift b/submodules/TelegramUI/Sources/AppDelegate.swift
index 66d305f6fc..d4892b7102 100644
--- a/submodules/TelegramUI/Sources/AppDelegate.swift
+++ b/submodules/TelegramUI/Sources/AppDelegate.swift
@@ -509,6 +509,40 @@ private func extractAccountManagerState(records: AccountRecordsView<TelegramAcco
         
         TempBox.initializeShared(basePath: rootPath, processType: "app", launchSpecificId: Int64.random(in: Int64.min ... Int64.max))
         
+        let writeAbilityTestFile = TempBox.shared.tempFile(fileName: "test.bin")
+        if let testFile = ManagedFile(queue: nil, path: writeAbilityTestFile.path, mode: .readwrite) {
+            let bufferSize = 128 * 1024
+            let randomBuffer = malloc(bufferSize)!
+            defer {
+                free(randomBuffer)
+            }
+            arc4random_buf(randomBuffer, bufferSize)
+            var writtenBytes = 0
+            var success = true
+            while writtenBytes < 1024 * 1024 {
+                let actualBytes = testFile.write(randomBuffer, count: bufferSize)
+                writtenBytes += actualBytes
+                if actualBytes != bufferSize {
+                    success = false
+                    break
+                }
+            }
+            testFile._unsafeClose()
+            TempBox.shared.dispose(writeAbilityTestFile)
+            
+            if !success {
+                let alertController = UIAlertController(title: nil, message: "There is not enough free space on the device.", preferredStyle: .alert)
+                alertController.addAction(UIAlertAction(title: "OK", style: .default, handler: { _ in
+                    preconditionFailure()
+                }))
+                self.mainWindow?.presentNative(alertController)
+                
+                return true
+            }
+        } else {
+            preconditionFailure()
+        }
+        
         let legacyLogs: [String] = [
             "broadcast-logs",
             "siri-logs",