Move AesInit() call to _PG_init()

jeltz · jeltz · commit b8338686eff6 · 2025-05-05T14:01:57.000+02:00
Also clean up the order of the function slightly.
diff --git a/contrib/pg_tde/src/pg_tde.c b/contrib/pg_tde/src/pg_tde.c
@@ -73,8 +73,6 @@ tde_shmem_startup(void)
 		prev_shmem_startup_hook();
 
 	TdeShmemInit();
-	AesInit();
-
 	TDEXLogShmemInit();
 	TDEXLogSmgrInit();
 }
@@ -96,23 +94,21 @@ _PG_init(void)
 
 	check_percona_api_version();
 
+	AesInit();
 	TdeGucInit();
 	TdeEventCaptureInit();
-
 	InitializePrincipalKeyInfo();
 	InitializeKeyProviderInfo();
-
-	prev_shmem_request_hook = shmem_request_hook;
-	shmem_request_hook = tde_shmem_request;
-	prev_shmem_startup_hook = shmem_startup_hook;
-	shmem_startup_hook = tde_shmem_startup;
-
 	InstallFileKeyring();
 	InstallVaultV2Keyring();
 	InstallKmipKeyring();
 	RegisterTdeRmgr();
-
 	RegisterStorageMgr();
+
+	prev_shmem_request_hook = shmem_request_hook;
+	shmem_request_hook = tde_shmem_request;
+	prev_shmem_startup_hook = shmem_startup_hook;
+	shmem_startup_hook = tde_shmem_startup;
 }
 
 static void
