Return to stream.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / lighttpd / src

Annotation of embedaddon/lighttpd/src/stream.c, revision 1.1.1.2

1.1.1.2 ! misho       1: #include "first.h"
        !             2: 
1.1       misho       3: #include "stream.h"
                      4: 
                      5: #include <sys/types.h>
                      6: #include <sys/stat.h>
                      7: 
                      8: #include <unistd.h>
                      9: #include <fcntl.h>
                     10: 
                     11: #include "sys-mmap.h"
                     12: 
                     13: #ifndef O_BINARY
                     14: # define O_BINARY 0
                     15: #endif
                     16: 
1.1.1.2 ! misho      17: /* don't want to block when open()ing a fifo */
        !            18: #if defined(O_NONBLOCK)
        !            19: # define FIFO_NONBLOCK O_NONBLOCK
        !            20: #else
        !            21: # define FIFO_NONBLOCK 0
        !            22: #endif
        !            23: 
        !            24: int stream_open(stream *f, const buffer *fn) {
        !            25: 
        !            26: #if !defined(__WIN32)
        !            27: 
1.1       misho      28:        struct stat st;
                     29:        int fd;
                     30: 
                     31:        f->start = NULL;
1.1.1.2 ! misho      32:        f->size = 0;
        !            33:        f->mapped = 0;
1.1       misho      34: 
1.1.1.2 ! misho      35:        if (-1 == (fd = open(fn->ptr, O_RDONLY | O_BINARY | FIFO_NONBLOCK))) {
1.1       misho      36:                return -1;
                     37:        }
                     38: 
1.1.1.2 ! misho      39:        if (-1 == fstat(fd, &st)) {
        !            40:                close(fd);
1.1       misho      41:                return -1;
                     42:        }
                     43: 
1.1.1.2 ! misho      44:        if (0 == st.st_size) {
        !            45:                /* empty file doesn't need a mapping */
        !            46:                close(fd);
        !            47:                return 0;
        !            48:        }
1.1       misho      49: 
1.1.1.2 ! misho      50:        f->start = mmap(NULL, st.st_size, PROT_READ, MAP_SHARED, fd, 0);
1.1       misho      51: 
                     52:        if (MAP_FAILED == f->start) {
1.1.1.2 ! misho      53:                f->start = malloc((size_t)st.st_size);
        !            54:                if (NULL == f->start
        !            55:                    || st.st_size != read(fd, f->start, (size_t)st.st_size)) {
        !            56:                        free(f->start);
        !            57:                        f->start = NULL;
        !            58:                        close(fd);
        !            59:                        return -1;
        !            60:                }
        !            61:        } else {
        !            62:                f->mapped = 1;
1.1       misho      63:        }
                     64: 
1.1.1.2 ! misho      65:        close(fd);
        !            66: 
        !            67:        f->size = st.st_size;
        !            68:        return 0;
        !            69: 
1.1       misho      70: #elif defined __WIN32
1.1.1.2 ! misho      71: 
        !            72:        HANDLE *fh, *mh;
        !            73:        void *p;
        !            74:        LARGE_INTEGER fsize;
        !            75: 
        !            76:        f->start = NULL;
        !            77:        f->size = 0;
        !            78: 
1.1       misho      79:        fh = CreateFile(fn->ptr,
                     80:                        GENERIC_READ,
                     81:                        FILE_SHARE_READ,
                     82:                        NULL,
                     83:                        OPEN_EXISTING,
                     84:                        FILE_ATTRIBUTE_READONLY,
                     85:                        NULL);
                     86: 
                     87:        if (!fh) return -1;
                     88: 
1.1.1.2 ! misho      89:        if (0 != GetFileSizeEx(fh, &fsize)) {
        !            90:                CloseHandle(fh);
        !            91:                return -1;
        !            92:        }
        !            93: 
        !            94:        if (0 == fsize) {
        !            95:                CloseHandle(fh);
        !            96:                return 0;
        !            97:        }
        !            98: 
1.1       misho      99:        mh = CreateFileMapping( fh,
                    100:                        NULL,
                    101:                        PAGE_READONLY,
1.1.1.2 ! misho     102:                        (sizeof(off_t) > 4) ? fsize >> 32 : 0,
        !           103:                        fsize & 0xffffffff,
1.1       misho     104:                        NULL);
                    105: 
                    106:        if (!mh) {
                    107: /*
                    108:                LPVOID lpMsgBuf;
                    109:                FormatMessage(
                    110:                        FORMAT_MESSAGE_ALLOCATE_BUFFER |
                    111:                        FORMAT_MESSAGE_FROM_SYSTEM,
                    112:                        NULL,
                    113:                        GetLastError(),
                    114:                        MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
                    115:                        (LPTSTR) &lpMsgBuf,
                    116:                        0, NULL );
                    117: */
1.1.1.2 ! misho     118:                CloseHandle(fh);
1.1       misho     119:                return -1;
                    120:        }
                    121: 
                    122:        p = MapViewOfFile(mh,
                    123:                        FILE_MAP_READ,
                    124:                        0,
                    125:                        0,
                    126:                        0);
                    127:        CloseHandle(mh);
                    128:        CloseHandle(fh);
                    129: 
                    130:        f->start = p;
1.1.1.2 ! misho     131:        f->size = (off_t)fsize;
        !           132:        return 0;
        !           133: 
1.1       misho     134: #endif
                    135: 
                    136: }
                    137: 
                    138: int stream_close(stream *f) {
                    139: #ifdef HAVE_MMAP
1.1.1.2 ! misho     140:        if (f->start) {
        !           141:                if (f->mapped) {
        !           142:                        f->mapped = 0;
        !           143:                        munmap(f->start, f->size);
        !           144:                } else {
        !           145:                        free(f->start);
        !           146:                }
        !           147:        }
1.1       misho     148: #elif defined(__WIN32)
                    149:        if (f->start) UnmapViewOfFile(f->start);
                    150: #endif
                    151: 
                    152:        f->start = NULL;
1.1.1.2 ! misho     153:        f->size = 0;
1.1       misho     154: 
                    155:        return 0;
                    156: }