🛡️ Sentinel: [보안 개선] 파일 쓰기 시 원자적(Atomic) 교체 적용

.jules/sentinel.md

@@ -83,3 +83,8 @@
 **Vulnerability:** 정적 HTML 생성 도구에서 매번 다른 Nonce를 동적으로 생성하여 CSP에 적용하는 것은, 캐싱 효율을 저하시킬 뿐만 아니라 정적 배포 환경(예: GitHub Pages 등)에서 올바른 보안 정책 수립을 방해할 수 있는 안티 패턴입니다.
 **Learning:** 정적으로 고정된 인라인 스타일이나 스크립트에는 난수화된 Nonce보다 콘텐츠 자체의 해시(SHA-256 등)를 사용하는 것이 안전하고 일관된 방식임을 배웠습니다.
 **Prevention:** 자동 생성되는 정적 HTML의 콘텐츠 보안 정책(CSP)에는 `style-src 'sha256-<HASH>'` 방식을 적용하고, `<style>` 태그에서 불필요한 `nonce` 속성을 제거하여 브라우저의 무결성 검증 기능을 적극 활용하십시오.
+
+## 2024-07-25 - [html4tree] 원자적 파일 교체를 통한 경합 조건(Race Condition) 방지
+**Vulnerability:** 파일 교체 시 `REPLACE_EXISTING`만 사용하면 교체 순간에 불완전한 파일 상태가 노출되거나 TOCTOU 취약점이 발생할 수 있습니다.
+**Learning:** 파일 교체 작업은 원자적(Atomic)으로 수행되어야 하며, 시스템에서 지원하지 않을 경우에만 일반 복사/이동으로 폴백해야 안전한 상태 전환을 보장할 수 있습니다.
+**Prevention:** 파일 업데이트 시 `StandardCopyOption.ATOMIC_MOVE`를 우선적으로 시도하고, 실패(`AtomicMoveNotSupportedException`)할 경우에만 `REPLACE_EXISTING`으로 폴백하도록 구현하십시오.

src/main/kotlin/html4tree/main.kt

@@ -229,12 +229,25 @@ fun process_ignore_file(curr_dir: File, dirFilesNames: Array<String>? = null): S
     return files_to_exclude
 }
 
-fun write_index_file(curr_dir: File, content: String) {
+fun write_index_file(
+    curr_dir: File,
+    content: String,
+    moveAtomic: (java.nio.file.Path, java.nio.file.Path) -> Unit = { src, dest ->
+        Files.move(src, dest, StandardCopyOption.REPLACE_EXISTING, StandardCopyOption.ATOMIC_MOVE)
+    },
+    moveFallback: (java.nio.file.Path, java.nio.file.Path) -> Unit = { src, dest ->
+        Files.move(src, dest, StandardCopyOption.REPLACE_EXISTING)
+    }
+) {
     val indexPath = curr_dir.toPath().resolve("index.html")
     val tempPath = Files.createTempFile(curr_dir.toPath(), ".index-", ".html")
     try {
         Files.write(tempPath, content.toByteArray(Charsets.UTF_8))
-        Files.move(tempPath, indexPath, StandardCopyOption.REPLACE_EXISTING)
+        try {
+            moveAtomic(tempPath, indexPath)
+        } catch (e: java.nio.file.AtomicMoveNotSupportedException) {
+            moveFallback(tempPath, indexPath)
+        }
     } finally {
         Files.deleteIfExists(tempPath)
     }

src/test/kotlin/html4tree/MainTest.kt

@@ -345,6 +345,60 @@ class MainTest {
         assertTrue(htmlContent.contains("margin: 0 auto;"))
     }
 
+    @Test
+    fun testWriteIndexFileAtomicMoveFallback() {
+        var atomicMoveAttempted = false
+        var fallbackMoveAttempted = false
+
+        write_index_file(
+            curr_dir = tempDir,
+            content = "content",
+            moveAtomic = { src, dest ->
+                atomicMoveAttempted = true
+                throw java.nio.file.AtomicMoveNotSupportedException(src.toString(), dest.toString(), "Mocked")
+            },
+            moveFallback = { src, dest ->
+                fallbackMoveAttempted = true
+                Files.move(src, dest, java.nio.file.StandardCopyOption.REPLACE_EXISTING)
+            }
+        )
+
+        assertTrue(atomicMoveAttempted)
+        assertTrue(fallbackMoveAttempted)
+        assertTrue(File(tempDir, "index.html").exists())
+    }
+
+    @Test
+    fun testWriteIndexFileDefaultLambdas() {
+        val testDir = File(tempDir, "testWriteIndexFileDefault")
+        testDir.mkdir()
+        val expectedContent = "default-lambda-content"
+        write_index_file(testDir, expectedContent)
+        val indexFile = File(testDir, "index.html")
+        assertTrue(indexFile.exists())
+        assertEquals(expectedContent, indexFile.readText())
+    }
+
+    @Test
+    fun testWriteIndexFileDefaultFallbackLambda() {
+        var atomicMoveAttempted = false
+        val testDir = File(tempDir, "testWriteIndexFileDefaultFallback")
+        testDir.mkdir()
+
+        write_index_file(
+            curr_dir = testDir,
+            content = "fallback-content",
+            moveAtomic = { src, dest ->
+                atomicMoveAttempted = true
+                throw java.nio.file.AtomicMoveNotSupportedException(src.toString(), dest.toString(), "Mocked")
+            }
+        )
+        assertTrue(atomicMoveAttempted)
+        val indexFile = File(testDir, "index.html")
+        assertTrue(indexFile.exists())
+        assertEquals("fallback-content", indexFile.readText())
+    }
+
     @Test
     fun testWriteIndexFileCleansUpTempFileOnFailure() {
         // Files.move cannot replace a non-empty directory, so this drives the