예제 #1
0
파일: request.go 프로젝트: edgarcai/xserver
func parseFlowRequestSlice(r *xio.PacketReader) (*flowRequestSlice, error) {
	var err error
	flags := uint8(0)
	if flags, err = r.Read8(); err != nil {
		return nil, errors.New("flow.read flags")
	}
	data := r.Bytes()
	return &flowRequestSlice{flags, data}, nil
}
예제 #2
0
파일: request.go 프로젝트: edgarcai/xserver
func parseAssignRequest(r *xio.PacketReader) (*assignRequest, error) {
	var err error
	yid := uint32(0)
	if yid, err = r.Read32(); err != nil {
		return nil, errors.New("assign.read yid")
	}
	coolen := uint64(0)
	if coolen, err = r.Read7BitValue64(); err != nil {
		return nil, errors.New("assign.read cookie.len")
	} else if coolen != cookies.CookieSize {
		return nil, errors.New("assign.bad cookie.len")
	}
	coobuf := make([]byte, cookies.CookieSize)
	if err = r.ReadBytes(coobuf); err != nil {
		return nil, errors.New("assign.read cookie.value")
	}
	size := uint64(0)
	if size, err = r.Read7BitValue64(); err != nil {
		return nil, errors.New("assign.read pid.size")
	} else if size == 0 || size > uint64(r.Len()) {
		return nil, errors.New("assign.bad pid.size")
	}
	pid := sha256.Sum256(r.Bytes()[:size])
	if len(pid) != 0x20 {
		return nil, errors.New("assign.bad pid.len")
	}
	if size, err = r.Read7BitValue64(); err != nil {
		return nil, errors.New("assign.read pubkey.len")
	} else if size <= 2 {
		return nil, errors.New("assign.too small pubkey.len")
	} else {
		size -= 2
		if size > uint64(r.Len()) {
			return nil, errors.New("assign.too big pubkey.len")
		}
		if err := r.Skip(2); err != nil {
			return nil, errors.New("assign.skip useless")
		}
	}
	pubkey := make([]byte, int(size))
	if err = r.ReadBytes(pubkey); err != nil {
		return nil, errors.New("assign.read pubkey")
	}
	if size, err = r.Read7BitValue64(); err != nil {
		return nil, errors.New("assign.read initiator.len")
	} else if size == 0 || size > uint64(r.Len()) {
		return nil, errors.New("assign.bad initiator.len")
	}
	initiator := make([]byte, int(size))
	if err = r.ReadBytes(initiator); err != nil {
		return nil, errors.New("assign.read initiator")
	}
	return &assignRequest{yid, string(pid[:]), coobuf, pubkey, initiator}, nil
}
예제 #3
0
파일: request.go 프로젝트: edgarcai/xserver
func parseFlowRequest(r *xio.PacketReader) (*flowRequest, error) {
	var err error
	flags := uint8(0)
	if flags, err = r.Read8(); err != nil {
		return nil, errors.New("flow.read flags")
	}
	fid := uint64(0)
	if fid, err = r.Read7BitValue64(); err != nil {
		return nil, errors.New("flow.read fid")
	}
	stage, delta := uint64(0), uint64(0)
	if stage, err = r.Read7BitValue64(); err != nil {
		return nil, errors.New("flow.read stage")
	}
	if delta, err = r.Read7BitValue64(); err != nil {
		return nil, errors.New("flow.read delta")
	}
	signature := ""
	if (flags & flagsHeader) != 0 {
		if signature, err = r.ReadString8(); err != nil {
			return nil, errors.New("flow.read signature")
		}
		for {
			if size, err := r.Read8(); err != nil {
				return nil, errors.New("flow.read header content size")
			} else if size == 0 {
				break
			} else if n := int(size); n > r.Len() {
				return nil, errors.New("flows.too big header content size")
			} else if err = r.Skip(n); err != nil {
				return nil, errors.New("flows.skip header")
			}
		}
	}
	data := r.Bytes()
	req := &flowRequest{}
	req.fid = fid
	req.signature = signature
	req.stage = stage
	req.stageack = stage - delta
	req.slices = make([]*flowRequestSlice, 0, 4)
	req.AddSlice(&flowRequestSlice{flags, data})
	return req, nil
}
예제 #4
0
파일: request.go 프로젝트: edgarcai/xserver
func ParseRequestMessage(r *xio.PacketReader) (*RequestMessage, error) {
	var err error
	code := uint8(0)
	if code, err = r.Read8(); err != nil {
		return nil, errors.New("message.read code")
	} else if code == 0xff {
		return nil, EOP
	}
	var size uint16
	if size, err = r.Read16(); err != nil {
		return nil, errors.New("message.read size")
	}
	rlen := r.Len()
	mlen := int(size)
	if rlen < mlen {
		return nil, errors.New("message.bad content length")
	} else {
		data := r.Bytes()[:mlen]
		if err := r.Skip(mlen); err != nil {
			return nil, errors.New("message.skip forward")
		}
		return &RequestMessage{code, xio.NewPacketReader(data)}, nil
	}
}