Fix bug

6block · May 22, 2024 · 21b0f45 · 21b0f45
1 parent 7d156a9
commit 21b0f45
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/internal/postrs/proof.go b/internal/postrs/proof.go
@@ -87,13 +87,15 @@ func GeneratePow(dataDir string, postServer string, priority uint, challenge []b
 		config.pow_difficulty[i] = C.uchar(b)
 	}
 
-	cPow := C.generate_pow(dataDirPtr, challengePtr, config, C.size_t(nonces), powFlags, postServerPtr, C.size_t(priority))
+	cPow := C.generate_pow(dataDirPtr, (*C.uchar)(challengePtr), config, C.size_t(nonces), powFlags, postServerPtr, C.size_t(priority))
 
-	if cPow == nil {
+	goStr := C.GoString(cPow)
+	C.free(unsafe.Pointer(cPow))
+	if &goStr == nil {
 		return nil, fmt.Errorf("got nil")
 	}
 
-	return &cPow, nil
+	return &goStr, nil
 }
 
 func GenerateProof(dataDir string, postServer string, powResponse string, challenge []byte, logger *zap.Logger, nonces, threads uint, K1, K2 uint32, powDifficulty [32]byte, powFlags PowFlags, options ...PostOptionFunc) (*shared.Proof, error) {