컴퓨터실 40대 PC의 스토리지 고갈과 네트워크 마비 0초 차단: sLMdown의 대용량 AI 모델 90초 스톨 가드 스트리밍 및 사전 디스크 쿼터 거버넌스 아키텍처

if (model.ExpectedBytes > 0)
{
    try
    {
        var driveInfo = new DriveInfo(Path.GetPathRoot(ModelBasePath)!);
        long freeBytes = driveInfo.AvailableFreeSpace;
        long requiredBytes = (long)(model.ExpectedBytes * 1.1); // 10% 안전 마진 확보

        if (freeBytes < requiredBytes)
        {
            string freeGb = $"{freeBytes / (1024.0 * 1024.0 * 1024.0):F1} GB";
            string needGb = $"{requiredBytes / (1024.0 * 1024.0 * 1024.0):F1} GB";
            
            // 디스크 포화 방어 모달 팝업 및 다운로드 즉각 차단
            var dialog = new ContentDialog
            {
                Title = "💾 디스크 용량 부족",
                Content = $"{model.DisplayName} 다운로드에 약 {needGb}가 필요합니다.\n\n" +
                          $"현재 여유 공간: {freeGb} ({driveInfo.Name})\n\n" +
                          $"디스크 공간을 확보한 후 다시 시도해 주세요.",
                CloseButtonText = "확인",
                XamlRoot = RootGrid.XamlRoot
            };
            await dialog.ShowAsync();
            return;
        }

        // 3GB 이상 대형 모델일 경우 학생의 실수 방지를 위한 2차 확인 다이얼로그
        if (model.ExpectedBytes >= 3_000_000_000L)
        {
            var confirmDialog = new ContentDialog
            {
                Title = "📥 대용량 모델 다운로드",
                Content = $"{model.DisplayName}을 다운로드합니다. 네트워크 상태에 따라 시간이 소요될 수 있습니다.",
                PrimaryButtonText = "다운로드",
                CloseButtonText = "취소",
                XamlRoot = RootGrid.XamlRoot
            };
            if (await confirmDialog.ShowAsync() != ContentDialogResult.Primary) return;
        }
    }
    catch { /* 드라이브 조회 예외 무음 격리 */ }
}

private const int DownloadStallTimeoutMs = 90_000; // 90초 무응답 시 스톨 판정

private static async Task ReadWithStallGuardAsync(
    Stream stream, byte[] buffer, CancellationTokenSource idleCts, CancellationToken userToken)
{
    // 읽기 시작 전 90초 타이머 무장
    idleCts.CancelAfter(DownloadStallTimeoutMs);
    try
    {
        int n = await stream.ReadAsync(buffer, 0, buffer.Length, idleCts.Token);
        // 바이트 수신 성공 시 즉시 워치독 해제 (다음 루프에서 재무장)
        idleCts.CancelAfter(Timeout.Infinite);
        return n;
    }
    catch (OperationCanceledException) when (!userToken.IsCancellationRequested)
    {
        // 사용자가 취소한 것이 아니라 워치독이 타임아웃된 경우 스톨 예외 발생
        throw new IOException($"다운로드 정지 감지: {DownloadStallTimeoutMs / 1000}초간 데이터가 없어 중단합니다.");
    }
}

// 1. 임시 파일 버퍼 (.downloading) 생성
string tempPath = localPath + ".downloading";
using (var fileStream = new FileStream(tempPath, FileMode.Create, FileAccess.Write, FileShare.None, 65536))
{
    while ((bytesRead = await ReadWithStallGuardAsync(contentStream, buffer, idleCts, model.Cts.Token)) > 0)
    {
        await fileStream.WriteAsync(buffer, 0, bytesRead, model.Cts.Token);
        downloadedBytes += bytesRead;
    }
}

// 2. 바이트 무결성 검증: 서버 헤더 명시 크기보다 덜 받았으면 잘린 파일이므로 승격 거부
if (contentLength.HasValue && downloadedBytes < contentLength.Value)
    throw new Exception($"네트워크 연결 조기 끊김 (수신: {downloadedBytes} / 예상: {contentLength.Value})");

// 3. 100% 무결성이 검증된 경우에만 원자적 이동(Atomic Move)
if (File.Exists(localPath)) File.Delete(localPath);
File.Move(tempPath, localPath);

// Datacenter 티어: 다운로드 및 실행 완전 차단, 정보 모달만 제공
if (model.Tier == "datacenter")
{
    var dialog = new ContentDialog
    {
        Title = $"🏢 {model.DisplayName}",
        Content = $"{model.Description}\n\n" +
                  $"모델 크기: {model.SizeText} ({model.ParamTag})\n\n" +
                  $"이 모델은 NVIDIA H100 80GB 이상의 데이터센터 GPU에서만 실행 가능합니다.\n" +
                  $"로컬 PC에서의 다운로드 및 실행은 학내망 보호를 위해 지원되지 않습니다.",
        CloseButtonText = "확인",
        XamlRoot = RootGrid.XamlRoot
    };
    await dialog.ShowAsync();
    return;
}