test case for socks udp

proxy/socks/protocol.go

@@ -344,6 +344,10 @@ type UDPReader struct {
 	reader io.Reader
 }
 
+func NewUDPReader(reader io.Reader) *UDPReader {
+	return &UDPReader{reader: reader}
+}
+
 func (r *UDPReader) Read() (*buf.Buffer, error) {
 	b := buf.NewSmall()
 	if err := b.AppendSupplier(buf.ReadFrom(r.reader)); err != nil {
@@ -363,6 +367,13 @@ type UDPWriter struct {
 	writer  io.Writer
 }
 
+func NewUDPWriter(request *protocol.RequestHeader, writer io.Writer) *UDPWriter {
+	return &UDPWriter{
+		request: request,
+		writer:  writer,
+	}
+}
+
 func (w *UDPWriter) Write(b *buf.Buffer) error {
 	eb := EncodeUDPPacket(w.request, b.Bytes())
 	b.Release()