throw

microsoft · Dec 2, 2024 · 9cc91ed · 9cc91ed
1 parent 03d7858
commit 9cc91ed
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/standalone/api/kernels/kernel.ts b/src/standalone/api/kernels/kernel.ts
@@ -432,12 +432,15 @@ class WrappedKernelPerExtension extends DisposableBase implements Kernel {
         );
 
         try {
+            let hasAccess: boolean = false;
             for await (const output of kernelExecution.executeCode(code, this.extensionId, events, token, async () => {
                 try {
                     // Validate access before execution
                     await this.checkAccess();
+                    hasAccess = true;
                     return true;
                 } catch (e) {
+                    hasAccess = false;
                     return false;
                 }
             })) {
@@ -458,6 +461,9 @@ class WrappedKernelPerExtension extends DisposableBase implements Kernel {
                     yield output;
                 }
             }
+            if (!hasAccess) {
+                throw new KernelAPIAccessRevoked();
+            }
         } finally {
             dispose(disposables);
         }