Merge remote-tracking branch 'gop/main' into q

internal/lib/internal/bytealg/bytealg.go

@@ -21,6 +21,7 @@ import (
 	"unsafe"
 
 	"github.com/goplus/llgo/c"
+	"github.com/goplus/llgo/internal/runtime"
 )
 
 func IndexByte(b []byte, ch byte) int {
@@ -40,3 +41,67 @@ func IndexByteString(s string, ch byte) int {
 	}
 	return -1
 }
+
+func Count(b []byte, c byte) (n int) {
+	for _, x := range b {
+		if x == c {
+			n++
+		}
+	}
+	return
+}
+
+func CountString(s string, c byte) (n int) {
+	for i := 0; i < len(s); i++ {
+		if s[i] == c {
+			n++
+		}
+	}
+	return
+}
+
+// Index returns the index of the first instance of b in a, or -1 if b is not present in a.
+// Requires 2 <= len(b) <= MaxLen.
+func Index(a, b []byte) int {
+	for i := 0; i <= len(a)-len(b); i++ {
+		if equal(a[i:i+len(b)], b) {
+			return i
+		}
+	}
+	return -1
+}
+
+func equal(a, b []byte) bool {
+	if n := len(a); n == len(b) {
+		return c.Memcmp(unsafe.Pointer(unsafe.SliceData(a)), unsafe.Pointer(unsafe.SliceData(b)), uintptr(n)) == 0
+	}
+	return false
+}
+
+// IndexString returns the index of the first instance of b in a, or -1 if b is not present in a.
+// Requires 2 <= len(b) <= MaxLen.
+func IndexString(a, b string) int {
+	for i := 0; i <= len(a)-len(b); i++ {
+		if a[i:i+len(b)] == b {
+			return i
+		}
+	}
+	return -1
+}
+
+// MakeNoZero makes a slice of length and capacity n without zeroing the bytes.
+// It is the caller's responsibility to ensure uninitialized bytes
+// do not leak to the end user.
+func MakeNoZero(n int) (r []byte) {
+	s := (*sliceHead)(unsafe.Pointer(&r))
+	s.data = runtime.AllocU(uintptr(n))
+	s.len = n
+	s.cap = n
+	return
+}
+
+type sliceHead struct {
+	data unsafe.Pointer
+	len  int
+	cap  int
+}