lib/net/textproto: patch Dial

This commit is contained in:
赵英杰
2024-08-13 15:28:30 +08:00
parent 9f1100b967
commit 4a6a97ee75
5 changed files with 369 additions and 2 deletions

View File

@@ -0,0 +1,13 @@
package main
import "net/textproto"
func main() {
h := make(textproto.MIMEHeader)
h.Set("host", "www.example.com")
println(h.Get("Host"))
}
/* Expected output:
www.example.com
*/

View File

@@ -0,0 +1,7 @@
package main
import "net/textproto"
func main() {
println(textproto.CanonicalMIMEHeaderKey("host"))
}

View File

@@ -0,0 +1,51 @@
; ModuleID = 'main'
source_filename = "main"
%"github.com/goplus/llgo/internal/runtime.String" = type { ptr, i64 }
@"main.init$guard" = global i1 false, align 1
@__llgo_argc = global i32 0, align 4
@__llgo_argv = global ptr null, align 8
@0 = private unnamed_addr constant [4 x i8] c"host", align 1
define void @main.init() {
_llgo_0:
%0 = load i1, ptr @"main.init$guard", align 1
br i1 %0, label %_llgo_2, label %_llgo_1
_llgo_1: ; preds = %_llgo_0
store i1 true, ptr @"main.init$guard", align 1
call void @"net/textproto.init"()
br label %_llgo_2
_llgo_2: ; preds = %_llgo_1, %_llgo_0
ret void
}
define i32 @main(i32 %0, ptr %1) {
_llgo_0:
store i32 %0, ptr @__llgo_argc, align 4
store ptr %1, ptr @__llgo_argv, align 8
call void @"github.com/goplus/llgo/internal/runtime.init"()
call void @main.init()
%2 = alloca %"github.com/goplus/llgo/internal/runtime.String", align 8
%3 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.String", ptr %2, i32 0, i32 0
store ptr @0, ptr %3, align 8
%4 = getelementptr inbounds %"github.com/goplus/llgo/internal/runtime.String", ptr %2, i32 0, i32 1
store i64 4, ptr %4, align 4
%5 = load %"github.com/goplus/llgo/internal/runtime.String", ptr %2, align 8
%6 = call %"github.com/goplus/llgo/internal/runtime.String" @"net/textproto.CanonicalMIMEHeaderKey"(%"github.com/goplus/llgo/internal/runtime.String" %5)
call void @"github.com/goplus/llgo/internal/runtime.PrintString"(%"github.com/goplus/llgo/internal/runtime.String" %6)
call void @"github.com/goplus/llgo/internal/runtime.PrintByte"(i8 10)
ret i32 0
}
declare void @"net/textproto.init"()
declare void @"github.com/goplus/llgo/internal/runtime.init"()
declare %"github.com/goplus/llgo/internal/runtime.String" @"net/textproto.CanonicalMIMEHeaderKey"(%"github.com/goplus/llgo/internal/runtime.String")
declare void @"github.com/goplus/llgo/internal/runtime.PrintString"(%"github.com/goplus/llgo/internal/runtime.String")
declare void @"github.com/goplus/llgo/internal/runtime.PrintByte"(i8)

View File

@@ -0,0 +1,292 @@
// Copyright 2010 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
// Package textproto implements generic support for text-based request/response
// protocols in the style of HTTP, NNTP, and SMTP.
//
// The package provides:
//
// Error, which represents a numeric error response from
// a server.
//
// Pipeline, to manage pipelined requests and responses
// in a client.
//
// Reader, to read numeric response code lines,
// key: value headers, lines wrapped with leading spaces
// on continuation lines, and whole text blocks ending
// with a dot on a line by itself.
//
// Writer, to write dot-encoded text blocks.
//
// Conn, a convenient packaging of Reader, Writer, and Pipeline for use
// with a single network connection.
package textproto
import (
"bufio"
"errors"
"fmt"
"io"
"io/fs"
"strings"
"syscall"
"unsafe"
"github.com/goplus/llgo/c"
"github.com/goplus/llgo/c/net"
"github.com/goplus/llgo/c/os"
)
// An Error represents a numeric error response from a server.
type Error struct {
Code int
Msg string
}
func (e *Error) Error() string {
return fmt.Sprintf("%03d %s", e.Code, e.Msg)
}
// A ProtocolError describes a protocol violation such
// as an invalid response or a hung-up connection.
type ProtocolError string
func (p ProtocolError) Error() string {
return string(p)
}
// A Conn represents a textual network protocol connection.
// It consists of a Reader and Writer to manage I/O
// and a Pipeline to sequence concurrent requests on the connection.
// These embedded types carry methods with them;
// see the documentation of those types for details.
type Conn struct {
Reader
Writer
Pipeline
conn io.ReadWriteCloser
}
// NewConn returns a new Conn using conn for I/O.
func NewConn(conn io.ReadWriteCloser) *Conn {
return &Conn{
Reader: Reader{R: bufio.NewReader(conn)},
Writer: Writer{W: bufio.NewWriter(conn)},
conn: conn,
}
}
// Close closes the connection.
func (c *Conn) Close() error {
return c.conn.Close()
}
// Dial connects to the given address on the given network using net.Dial
// and then returns a new Conn for the connection.
func Dial(network, addr string) (*Conn, error) {
cconn, err := dialNetWork(network, addr)
if err != nil {
return nil, err
}
return NewConn(cconn), nil
}
type cConn struct {
socketFd c.Int
closed bool
}
func (conn *cConn) Read(p []byte) (n int, err error) {
if conn == nil || conn.closed {
return 0, fs.ErrClosed
}
if len(p) == 0 {
return 0, nil
}
for n < len(p) {
result := os.Read(conn.socketFd, unsafe.Pointer(&p[n:][0]), uintptr(len(p)-n))
if result < 0 {
if os.Errno == c.Int(syscall.EINTR) {
continue
}
return n, errors.New("read error")
}
if result == 0 {
return n, io.EOF
}
n += result
}
return n, nil
}
func (conn *cConn) Write(p []byte) (n int, err error) {
if conn == nil || conn.closed {
return 0, fs.ErrClosed
}
for n < len(p) {
result := os.Write(conn.socketFd, unsafe.Pointer(&p[n:][0]), uintptr(len(p)-n))
if result < 0 {
if os.Errno == c.Int(syscall.EINTR) {
continue
}
return n, errors.New("write error")
}
n += result
}
if n < len(p) {
return n, io.ErrShortWrite
}
return n, nil
}
func (conn *cConn) Close() error {
if conn == nil {
return nil
}
if conn.closed {
return fs.ErrClosed
}
conn.closed = true
result := os.Close(conn.socketFd)
if result < 0 {
return errors.New(c.GoString(c.Strerror(os.Errno)))
}
return nil
}
func dialNetWork(network, addr string) (*cConn, error) {
host, port, err := splitAddr(addr)
if err != nil {
return nil, err
}
var hints net.AddrInfo
var res *net.AddrInfo
c.Memset(unsafe.Pointer(&hints), 0, unsafe.Sizeof(hints))
hints.Family = net.AF_UNSPEC
hints.SockType = net.SOCK_STREAM
status := net.Getaddrinfo(c.AllocaCStr(host), c.AllocaCStr(port), &hints, &res)
if status != 0 {
return nil, errors.New("getaddrinfo error")
}
socketFd := net.Socket(res.Family, res.SockType, res.Protocol)
if socketFd == -1 {
net.Freeaddrinfo(res)
return nil, errors.New("socket error")
}
if net.Connect(socketFd, res.Addr, res.AddrLen) == -1 {
os.Close(socketFd)
net.Freeaddrinfo(res)
return nil, errors.New("connect error")
}
net.Freeaddrinfo(res)
return &cConn{
socketFd: socketFd,
}, nil
}
func splitAddr(addr string) (host, port string, err error) {
// Handle IPv6 addresses
if strings.HasPrefix(addr, "[") {
closeBracket := strings.LastIndex(addr, "]")
if closeBracket == -1 {
return "", "", errors.New("invalid IPv6 address: missing closing bracket")
}
host = addr[1:closeBracket]
if len(addr) > closeBracket+1 {
if addr[closeBracket+1] != ':' {
return "", "", errors.New("invalid address: colon missing after IPv6 address")
}
port = addr[closeBracket+2:]
}
} else {
// Handle IPv4 addresses or domain names
parts := strings.Split(addr, ":")
if len(parts) > 2 {
return "", "", errors.New("invalid address: too many colons")
}
host = parts[0]
if len(parts) == 2 {
port = parts[1]
}
}
if host == "" {
return "", "", errors.New("invalid address: host is empty")
}
if port == "" {
port = "80" // Default port is 80
}
return host, port, nil
}
// Cmd is a convenience method that sends a command after
// waiting its turn in the pipeline. The command text is the
// result of formatting format with args and appending \r\n.
// Cmd returns the id of the command, for use with StartResponse and EndResponse.
//
// For example, a client might run a HELP command that returns a dot-body
// by using:
//
// id, err := c.Cmd("HELP")
// if err != nil {
// return nil, err
// }
//
// c.StartResponse(id)
// defer c.EndResponse(id)
//
// if _, _, err = c.ReadCodeLine(110); err != nil {
// return nil, err
// }
// text, err := c.ReadDotBytes()
// if err != nil {
// return nil, err
// }
// return c.ReadCodeLine(250)
func (c *Conn) Cmd(format string, args ...any) (id uint, err error) {
id = c.Next()
c.StartRequest(id)
err = c.PrintfLine(format, args...)
c.EndRequest(id)
if err != nil {
return 0, err
}
return id, nil
}
// TrimString returns s without leading and trailing ASCII space.
func TrimString(s string) string {
for len(s) > 0 && isASCIISpace(s[0]) {
s = s[1:]
}
for len(s) > 0 && isASCIISpace(s[len(s)-1]) {
s = s[:len(s)-1]
}
return s
}
// TrimBytes returns b without leading and trailing ASCII space.
func TrimBytes(b []byte) []byte {
for len(b) > 0 && isASCIISpace(b[0]) {
b = b[1:]
}
for len(b) > 0 && isASCIISpace(b[len(b)-1]) {
b = b[:len(b)-1]
}
return b
}
func isASCIISpace(b byte) bool {
return b == ' ' || b == '\t' || b == '\n' || b == '\r'
}
func isASCIILetter(b byte) bool {
b |= 0x20 // make lower case
return 'a' <= b && b <= 'z'
}

View File

@@ -7,7 +7,11 @@ import (
//go:embed _overlay/go/parser/resolver.go //go:embed _overlay/go/parser/resolver.go
var go_parser_resolver string var go_parser_resolver string
//go:embed _overlay/net/textproto/textproto.go
var net_textproto string
var overlayFiles = map[string]string{ var overlayFiles = map[string]string{
"math/exp_amd64.go": "package math;", "math/exp_amd64.go": "package math;",
"go/parser/resolver.go": go_parser_resolver, "go/parser/resolver.go": go_parser_resolver,
"net/textproto/textproto.go": net_textproto,
} }