예제 #1
0
func serveError(req *web.Request, status int, reason os.Error, header web.Header) {
	header.Set(web.HeaderContentType, "text/plain; charset=utf-8")
	w := req.Responder.Respond(status, header)
	io.WriteString(w, web.StatusText(status))
	if reason != nil {
		io.WriteString(w, "\n")
		io.WriteString(w, reason.String())
	}
}
예제 #2
0
func (t *transaction) Respond(status int, header web.Header) (body io.Writer) {
	if t.hijacked {
		log.Println("twister: Respond called on hijacked connection")
		return &nullResponseBody{err: web.ErrInvalidState}
	}
	if t.respondCalled {
		log.Println("twister: Multiple calls to Respond")
		return &nullResponseBody{err: web.ErrInvalidState}
	}
	t.respondCalled = true
	t.requestErr = web.ErrInvalidState
	t.status = status
	t.header = header

	if te := header.Get(web.HeaderTransferEncoding); te != "" {
		log.Println("twister: transfer encoding not allowed")
		delete(header, web.HeaderTransferEncoding)
	}

	if !t.requestConsumed {
		t.closeAfterResponse = true
	}

	if header.Get(web.HeaderConnection) == "close" {
		t.closeAfterResponse = true
	}

	t.chunkedResponse = true
	contentLength := -1

	if status == web.StatusNotModified {
		delete(header, web.HeaderContentType)
		delete(header, web.HeaderContentLength)
		t.chunkedResponse = false
	} else if s := header.Get(web.HeaderContentLength); s != "" {
		contentLength, _ = strconv.Atoi(s)
		t.chunkedResponse = false
	} else if t.req.ProtocolVersion < web.ProtocolVersion(1, 1) {
		t.closeAfterResponse = true
	}

	if t.closeAfterResponse {
		header.Set(web.HeaderConnection, "close")
		t.chunkedResponse = false
	}

	if t.req.Method == "HEAD" {
		t.chunkedResponse = false
	}

	if t.chunkedResponse {
		header.Set(web.HeaderTransferEncoding, "chunked")
	}

	proto := "HTTP/1.0"
	if t.req.ProtocolVersion >= web.ProtocolVersion(1, 1) {
		proto = "HTTP/1.1"
	}
	statusString := strconv.Itoa(status)
	text := web.StatusText(status)

	var b bytes.Buffer
	b.WriteString(proto)
	b.WriteString(" ")
	b.WriteString(statusString)
	b.WriteString(" ")
	b.WriteString(text)
	b.WriteString("\r\n")
	header.WriteHttpHeader(&b)
	t.headerSize = b.Len()

	const bufferSize = 4096
	switch {
	case t.req.Method == "HEAD" || status == web.StatusNotModified:
		t.responseBody, _ = newNullResponseBody(t.conn, b.Bytes())
	case t.chunkedResponse:
		t.responseBody, _ = newChunkedResponseBody(t.conn, b.Bytes(), bufferSize)
	default:
		t.responseBody, _ = newIdentityResponseBody(t.conn, b.Bytes(), bufferSize, contentLength)
	}
	return t.responseBody
}