fix(mitm): add cert logging

fix(puppet): stablilize dom storage flush

core/lib/Tab.ts

@@ -257,7 +257,7 @@ export default class Tab
     const errors: Error[] = [];
 
     try {
-      await this.puppetPage.domStorageTracker.flush(5e3);
+      await this.puppetPage.domStorageTracker.finalFlush(5e3);
     } catch (error) {
       if (!error.message.includes('Target closed') && !(error instanceof CanceledPromiseError)) {
         errors.push(error);

interfaces/IPuppetDomStorageTracker.ts

@@ -10,7 +10,7 @@ export default interface IPuppetDomStorageTracker {
       storageForOrigin: IDomStorageForOrigin;
     }[]
   >;
-  flush(timeoutMs?: number): Promise<void>;
+  finalFlush(timeoutMs?: number): Promise<void>;
 }
 
 export interface IPuppetStorageEvents {

mitm-socket/go/generate_cert.go

@@ -8,8 +8,10 @@ import (
 	"crypto/x509"
 	"crypto/x509/pkix"
 	"encoding/pem"
+	"errors"
 	"fmt"
 	"math/big"
+	"log"
 	"net"
 	"os"
 	"sync/atomic"
@@ -39,6 +41,7 @@ type CertConfig struct {
 func readCertFromDisk(file string) (*x509.Certificate, error) {
 
 	bytes, err := os.ReadFile(file)
+
 	if err != nil {
 		return nil, err
 	}
@@ -76,15 +79,19 @@ func NewAuthority() (*x509.Certificate, *rsa.PrivateKey, error) {
 	var caKeyFile string = "caKey.der"
 
 	certFromDisk, err := readCertFromDisk(caFile)
-	if err == nil {
+
+	if err != nil && !errors.Is(err, os.ErrNotExist) {
+       log.Printf("Error reading cert from disk", caFile, err)
+    } else if err == nil {
 		keyFromDisk, err := readPrivateKeyFromDisk(caKeyFile)
 		if err != nil {
-			fmt.Printf("Error reading private key from disk", caKeyFile, err)
+            log.Printf("Error reading private key from disk", caKeyFile, err)
 		} else {
 			return certFromDisk, keyFromDisk, nil
 		}
 	}
 
+
 	// Generating the private key that will be used for domain certificates
 	priv, err := rsa.GenerateKey(rand.Reader, 2048)
 	if err != nil {

mitm-socket/lib/BaseIpcHandler.ts

@@ -176,7 +176,7 @@ export default abstract class BaseIpcHandler {
 
   private onChildProcessStderr(message: string): void {
     if (this.isClosing) return;
-    this.logger.info(`${this.handlerName}.stderr: ${message}`);
+    this.logger.error(`${this.handlerName}.stderr: ${message}`);
   }
 
   private spawnChild(): void {

puppet-chrome/lib/DomStorageTracker.ts

@@ -70,7 +70,8 @@ export class DomStorageTracker
     this.cancelPendingEvents('DomStorageTracker closed');
   }
 
-  public async flush(timeoutMs = 30e3): Promise<void> {
+  public async finalFlush(timeoutMs = 30e3): Promise<void> {
+    eventUtils.removeEventListeners(this.registeredEvents);
     await Promise.race([
       this.processingPromise,
       new Promise<void>(resolve => setTimeout(resolve, timeoutMs ?? 0)),
@@ -294,6 +295,8 @@ export class DomStorageTracker
     this.indexedDBContentUpdatingOrigins.add(securityOrigin);
 
     const timestamp = Date.now();
+    const resolvable = new Resolvable<void>();
+    this.processingPromise = this.processingPromise.then(() => resolvable.promise);
     try {
       const db = await this.getLatestIndexedDB(securityOrigin, databaseName);
       const objectStore = db.objectStores.find(x => x.name === objectStoreName);
@@ -329,6 +332,7 @@ export class DomStorageTracker
       });
     } finally {
       this.indexedDBContentUpdatingOrigins.delete(securityOrigin);
+      resolvable.resolve();
     }
   }