Added CheckAppBetaPassword to retrieve encryption keys for encrypted_gid_2 depots

DepotDownloader/ContentDownloader.cs

@@ -247,16 +247,53 @@ namespace DepotDownloader
                         Config.BetaPassword = password = Console.ReadLine();
                     }
 
-                    byte[] input = Util.DecodeHexString(node_encrypted["encrypted_gid"].Value);
-                    byte[] manifest_bytes = CryptoHelper.VerifyAndDecryptPassword(input, password);
+                    var encrypted_v1 = node_encrypted["encrypted_gid"];
+                    var encrypted_v2 = node_encrypted["encrypted_gid_2"];
 
-                    if (manifest_bytes == null)
+                    if (encrypted_v1 != KeyValue.Invalid)
                     {
-                        Console.WriteLine("Password was invalid for branch {0}", branch);
+                        byte[] input = Util.DecodeHexString(encrypted_v1.Value);
+                        byte[] manifest_bytes = CryptoHelper.VerifyAndDecryptPassword(input, password);
+
+                        if (manifest_bytes == null)
+                        {
+                            Console.WriteLine("Password was invalid for branch {0}", branch);
+                            return INVALID_MANIFEST_ID;
+                        }
+
+                        return BitConverter.ToUInt64(manifest_bytes, 0);
+                    }
+                    else if (encrypted_v2 != KeyValue.Invalid)
+                    {
+                        // Submit the password to Steam now to get encryption keys
+                        steam3.CheckAppBetaPassword(appId, Config.BetaPassword);
+
+                        if (!steam3.AppBetaPasswords.ContainsKey(branch))
+                        {
+                            Console.WriteLine("Password was invalid for branch {0}", branch);
+                            return INVALID_MANIFEST_ID;
+                        }
+
+                        byte[] input = Util.DecodeHexString(encrypted_v2.Value);
+                        byte[] manifest_bytes;
+                        try
+                        {
+                            manifest_bytes = CryptoHelper.SymmetricDecryptECB(input, steam3.AppBetaPasswords[branch]);
+                        }
+                        catch (Exception e)
+                        {
+                            Console.WriteLine("Failed to decrypt branch {0}", branch);
+                            return INVALID_MANIFEST_ID;
+                        }
+
+                        return BitConverter.ToUInt64(manifest_bytes, 0);
+                    }
+                    else
+                    {
+                        Console.WriteLine("Unhandled depot encryption for depotId {0}", depotId);
                         return INVALID_MANIFEST_ID;
                     }
 
-                    return BitConverter.ToUInt64(manifest_bytes, 0);
                 }
 
                 return INVALID_MANIFEST_ID;

DepotDownloader/Steam3Session.cs

@@ -33,6 +33,7 @@ namespace DepotDownloader
         public Dictionary<Tuple<uint, string>, SteamApps.CDNAuthTokenCallback> CDNAuthTokens { get; private set; }
         public Dictionary<uint, SteamApps.PICSProductInfoCallback.PICSProductInfo> AppInfo { get; private set; }
         public Dictionary<uint, SteamApps.PICSProductInfoCallback.PICSProductInfo> PackageInfo { get; private set; }
+        public Dictionary<string, byte[]> AppBetaPasswords { get; private set; }
 
         public SteamClient steamClient;
         public SteamUser steamUser;
@@ -73,6 +74,7 @@ namespace DepotDownloader
             this.CDNAuthTokens = new Dictionary<Tuple<uint, string>, SteamApps.CDNAuthTokenCallback>();
             this.AppInfo = new Dictionary<uint, SteamApps.PICSProductInfoCallback.PICSProductInfo>();
             this.PackageInfo = new Dictionary<uint, SteamApps.PICSProductInfoCallback.PICSProductInfo>();
+            this.AppBetaPasswords = new Dictionary<string, byte[]>();
 
             this.steamClient = new SteamClient();
 
@@ -309,6 +311,27 @@ namespace DepotDownloader
             }, () => { return completed; });
         }
 
+        public void CheckAppBetaPassword(uint appid, string password)
+        {
+            bool completed = false;
+            Action<SteamApps.CheckAppBetaPasswordCallback> cbMethod = (appPassword) =>
+            {
+                completed = true;
+
+                Console.WriteLine("Retrieved {0} beta keys with result: {1}", appPassword.BetaPasswords.Count, appPassword.Result);
+
+                foreach (var entry in appPassword.BetaPasswords)
+                {
+                    AppBetaPasswords.Add(entry.Key, entry.Value);
+                }
+            };
+
+            WaitUntilCallback(() =>
+            {
+                callbacks.Subscribe(steamApps.CheckAppBetaPassword(appid, password), cbMethod);
+            }, () => { return completed; });
+        }
+
         void Connect()
         {
             bAborted = false;