Annotation of embedaddon/lighttpd/src/network_write_no_mmap.c, revision 1.1

1.1     ! misho       1: #include "first.h"
        !             2: 
        !             3: #include "network_backends.h"
        !             4: 
        !             5: #include "network.h"
        !             6: #include "fdevent.h"
        !             7: #include "log.h"
        !             8: #include "stat_cache.h"
        !             9: 
        !            10: #include "sys-socket.h"
        !            11: 
        !            12: #include <sys/time.h>
        !            13: #include <stdlib.h>
        !            14: 
        !            15: #include <fcntl.h>
        !            16: #include <sys/stat.h>
        !            17: #include <unistd.h>
        !            18: 
        !            19: #include <errno.h>
        !            20: #include <string.h>
        !            21: 
        !            22: int network_open_file_chunk(server *srv, connection *con, chunkqueue *cq) {
        !            23:        chunk* const c = cq->first;
        !            24:        off_t file_size, offset, toSend;
        !            25: 
        !            26:        force_assert(NULL != c);
        !            27:        force_assert(FILE_CHUNK == c->type);
        !            28:        force_assert(c->offset >= 0 && c->offset <= c->file.length);
        !            29: 
        !            30:        offset = c->file.start + c->offset;
        !            31:        toSend = c->file.length - c->offset;
        !            32: 
        !            33:        if (-1 == c->file.fd) {
        !            34:                stat_cache_entry *sce = NULL;
        !            35: 
        !            36:                if (HANDLER_ERROR == stat_cache_get_entry(srv, con, c->file.name, &sce)) {
        !            37:                        log_error_write(srv, __FILE__, __LINE__, "ssb", "stat-cache failed:", strerror(errno), c->file.name);
        !            38:                        return -1;
        !            39:                }
        !            40: 
        !            41:                if (-1 == (c->file.fd = open(c->file.name->ptr, O_RDONLY|O_NOCTTY))) {
        !            42:                        log_error_write(srv, __FILE__, __LINE__, "ssb", "open failed:", strerror(errno), c->file.name);
        !            43:                        return -1;
        !            44:                }
        !            45:                fd_close_on_exec(c->file.fd);
        !            46: 
        !            47:                file_size = sce->st.st_size;
        !            48:        } else {
        !            49:                struct stat st;
        !            50:                if (-1 == fstat(c->file.fd, &st)) {
        !            51:                        log_error_write(srv, __FILE__, __LINE__, "ss", "fstat failed:", strerror(errno));
        !            52:                        return -1;
        !            53:                }
        !            54:                file_size = st.st_size;
        !            55:        }
        !            56: 
        !            57:        if (offset > file_size || toSend > file_size || offset > file_size - toSend) {
        !            58:                log_error_write(srv, __FILE__, __LINE__, "sb", "file was shrinked:", c->file.name);
        !            59:                return -1;
        !            60:        }
        !            61: 
        !            62:        return 0;
        !            63: }
        !            64: 
        !            65: int network_write_file_chunk_no_mmap(server *srv, connection *con, int fd, chunkqueue *cq, off_t *p_max_bytes) {
        !            66:        chunk* const c = cq->first;
        !            67:        off_t offset, toSend;
        !            68:        ssize_t r;
        !            69: 
        !            70:        force_assert(NULL != c);
        !            71:        force_assert(FILE_CHUNK == c->type);
        !            72:        force_assert(c->offset >= 0 && c->offset <= c->file.length);
        !            73: 
        !            74:        offset = c->file.start + c->offset;
        !            75:        toSend = c->file.length - c->offset;
        !            76:        if (toSend > 64*1024) toSend = 64*1024; /* max read 64kb in one step */
        !            77:        if (toSend > *p_max_bytes) toSend = *p_max_bytes;
        !            78: 
        !            79:        if (0 == toSend) {
        !            80:                chunkqueue_remove_finished_chunks(cq);
        !            81:                return 0;
        !            82:        }
        !            83: 
        !            84:        if (0 != network_open_file_chunk(srv, con, cq)) return -1;
        !            85: 
        !            86:        buffer_string_prepare_copy(srv->tmp_buf, toSend);
        !            87: 
        !            88:        if (-1 == lseek(c->file.fd, offset, SEEK_SET)) {
        !            89:                log_error_write(srv, __FILE__, __LINE__, "ss", "lseek: ", strerror(errno));
        !            90:                return -1;
        !            91:        }
        !            92:        if (-1 == (toSend = read(c->file.fd, srv->tmp_buf->ptr, toSend))) {
        !            93:                log_error_write(srv, __FILE__, __LINE__, "ss", "read: ", strerror(errno));
        !            94:                return -1;
        !            95:        }
        !            96: 
        !            97: #if defined(__WIN32)
        !            98:        if ((r = send(fd, srv->tmp_buf->ptr, toSend, 0)) < 0) {
        !            99:                int lastError = WSAGetLastError();
        !           100:                switch (lastError) {
        !           101:                case WSAEINTR:
        !           102:                case WSAEWOULDBLOCK:
        !           103:                        break;
        !           104:                case WSAECONNRESET:
        !           105:                case WSAETIMEDOUT:
        !           106:                case WSAECONNABORTED:
        !           107:                        return -2;
        !           108:                default:
        !           109:                        log_error_write(srv, __FILE__, __LINE__, "sdd",
        !           110:                                "send failed: ", lastError, fd);
        !           111:                        return -1;
        !           112:                }
        !           113:        }
        !           114: #else /* __WIN32 */
        !           115:        if ((r = write(fd, srv->tmp_buf->ptr, toSend)) < 0) {
        !           116:                switch (errno) {
        !           117:                case EAGAIN:
        !           118:                case EINTR:
        !           119:                        break;
        !           120:                case EPIPE:
        !           121:                case ECONNRESET:
        !           122:                        return -2;
        !           123:                default:
        !           124:                        log_error_write(srv, __FILE__, __LINE__, "ssd",
        !           125:                                "write failed:", strerror(errno), fd);
        !           126:                        return -1;
        !           127:                }
        !           128:        }
        !           129: #endif /* __WIN32 */
        !           130: 
        !           131:        if (r >= 0) {
        !           132:                *p_max_bytes -= r;
        !           133:                chunkqueue_mark_written(cq, r);
        !           134:        }
        !           135: 
        !           136:        return (r > 0 && r == toSend) ? 0 : -3;
        !           137: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>