Replace bson with bson2 in wire

.github/settings.yml

@@ -55,7 +55,7 @@ labels:
     description: Issues about query pushdown
   - name: area/types
     color: "#0354F4"
-    description: Issues about data types, BSON, etc
+    description: Issues about data types, BSON, wire protocol, etc
 
   # backend
   - name: backend/pg

.golangci.yml

@@ -37,7 +37,6 @@ linters-settings:
       bson:
         files:
           - $all
-          - "!**/internal/wire/*.go"
           - "!**/internal/bson2/*_test.go"
         deny:
           - pkg: github.com/FerretDB/FerretDB/internal/bson$

Taskfile.yml

@@ -324,10 +324,7 @@ tasks:
     desc: "Run unit benchmarks"
     cmds:
       - go test -list='Benchmark.*' ./...
-      - go test -count=10 -bench=BenchmarkArray    -benchtime={{.BENCH_TIME}} ./internal/bson/          | tee -a new.txt
-      - go test -count=10 -bench=BenchmarkDocument -benchtime={{.BENCH_TIME}} ./internal/bson/          | tee -a new.txt
-      - go test -count=10 -bench=BenchmarkArray    -benchtime={{.BENCH_TIME}} ./internal/handler/sjson/ | tee -a new.txt
-      - go test -count=10 -bench=BenchmarkDocument -benchtime={{.BENCH_TIME}} ./internal/handler/sjson/ | tee -a new.txt
+      - go test -count=10 -bench=BenchmarkDocument -benchtime={{.BENCH_TIME}} ./internal/bson2/         | tee -a new.txt
       - bin/benchstat{{exeExt}} old.txt new.txt
 
   # That's not quite correct: https://github.com/golang/go/issues/15513

internal/bson2/bson2.go

@@ -49,11 +49,6 @@ import (
 	"github.com/FerretDB/FerretDB/internal/util/lazyerrors"
 )
 
-// If true, the usage of float64 NaN values is disallowed.
-// They mess up many things too much, starting with simple equality tests.
-// But allowing them simplifies fuzzing where we currently compare converted [*types.Document]s.
-var noNaN = true
-
 type (
 	// ScalarType represents a BSON scalar type.
 	//
@@ -218,6 +213,12 @@ func convertToTypes(v any) (any, error) {
 	case string:
 		return v, nil
 	case Binary:
+		// Special case to prevent it from being stored as null in sjson / logged as null in fjson.
+		// TODO https://github.com/FerretDB/FerretDB/issues/260
+		if v.B == nil {
+			v.B = []byte{}
+		}
+
 		return types.Binary{
 			B:       v.B,
 			Subtype: types.BinarySubtype(v.Subtype),

internal/bson2/bson2_test.go

@@ -469,11 +469,6 @@ var decodeTestCases = []decodeTestCase{
 }
 
 func TestNormal(t *testing.T) {
-	prev := noNaN
-	noNaN = false
-
-	t.Cleanup(func() { noNaN = prev })
-
 	for _, tc := range normalTestCases {
 		t.Run(tc.name, func(t *testing.T) {
 			t.Run("bson", func(t *testing.T) {
@@ -560,11 +555,6 @@ func TestNormal(t *testing.T) {
 }
 
 func TestDecode(t *testing.T) {
-	prev := noNaN
-	noNaN = false
-
-	t.Cleanup(func() { noNaN = prev })
-
 	for _, tc := range decodeTestCases {
 		if tc.decodeDeepErr == nil {
 			tc.decodeDeepErr = tc.decodeErr
@@ -622,11 +612,6 @@ func TestDecode(t *testing.T) {
 }
 
 func BenchmarkDocument(b *testing.B) {
-	prev := noNaN
-	noNaN = false
-
-	b.Cleanup(func() { noNaN = prev })
-
 	for _, tc := range normalTestCases {
 		b.Run(tc.name, func(b *testing.B) {
 			b.Run("bson", func(b *testing.B) {
@@ -854,11 +839,6 @@ func testRawDocument(t *testing.T, rawDoc RawDocument) {
 }
 
 func FuzzDocument(f *testing.F) {
-	prev := noNaN
-	noNaN = false
-
-	f.Cleanup(func() { noNaN = prev })
-
 	for _, tc := range normalTestCases {
 		f.Add([]byte(tc.raw))
 	}

internal/bson2/decode.go

@@ -16,7 +16,6 @@ package bson2
 
 import (
 	"encoding/binary"
-	"math"
 
 	"github.com/cristalhq/bson/bsonproto"
 
@@ -67,10 +66,6 @@ func decodeScalarField(b []byte, t tag) (v any, size int, err error) {
 		v = f
 		size = bsonproto.SizeFloat64
 
-		if noNaN && math.IsNaN(f) {
-			err = lazyerrors.Errorf("got NaN value: %w", ErrDecodeInvalidInput)
-		}
-
 	case tagString:
 		var s string
 		s, err = bsonproto.DecodeString(b)

internal/bson2/document.go

@@ -19,7 +19,6 @@ import (
 	"encoding/binary"
 	"errors"
 	"log/slog"
-	"math"
 
 	"github.com/FerretDB/FerretDB/internal/types"
 	"github.com/FerretDB/FerretDB/internal/util/iterator"
@@ -140,12 +139,6 @@ func (doc *Document) add(name string, value any) error {
 		return lazyerrors.Errorf("%q: %w", name, err)
 	}
 
-	if f, ok := value.(float64); ok {
-		if noNaN && math.IsNaN(f) {
-			return lazyerrors.New("invalid float64 value NaN")
-		}
-	}
-
 	doc.fields = append(doc.fields, field{
 		name:  name,
 		value: value,

internal/bson2/encode.go

@@ -17,7 +17,6 @@ package bson2
 import (
 	"bytes"
 	"fmt"
-	"math"
 	"time"
 
 	"github.com/cristalhq/bson/bsonproto"
@@ -117,10 +116,6 @@ func encodeField(buf *bytes.Buffer, name string, v any) error {
 func encodeScalarField(buf *bytes.Buffer, name string, v any) error {
 	switch v := v.(type) {
 	case float64:
-		if noNaN && math.IsNaN(v) {
-			return lazyerrors.Errorf("got NaN value")
-		}
-
 		buf.WriteByte(byte(tagFloat64))
 	case string:
 		buf.WriteByte(byte(tagString))

internal/bson2/raw_document.go

@@ -84,7 +84,7 @@ func (raw RawDocument) decode(mode decodeMode) (*Document, error) {
 		return nil, lazyerrors.Errorf("len(raw) = %d, l = %d: %w", rl, l, ErrDecodeInvalidInput)
 	}
 
-	res := MakeDocument(1)
+	res := MakeDocument(0)
 
 	offset := 4
 

internal/handler/handlererrors/error.go

@@ -357,8 +357,8 @@ type ProtoErr interface {
 // ProtocolError converts any error to wire protocol error.
 //
 // Nil panics (it never should be passed),
-// *CommandError or *WriteErrors (possibly wrapped) are returned unwrapped,
-// *wire.ValidationError (possibly wrapped) is returned as CommandError with BadValue code,
+// [*CommandError] or [*WriteErrors] (possibly wrapped) are returned unwrapped,
+// [*wire.ValidationError] (possibly wrapped) is returned as CommandError with BadValue code,
 // any other values (including lazy errors) are returned as CommandError with InternalError code.
 func ProtocolError(err error) ProtoErr {
 	if err == nil {