version 1.1.1.2, 2014/06/15 20:20:06
|
version 1.1.1.3, 2016/11/02 10:35:00
|
Line 1
|
Line 1
|
|
#include "first.h" |
|
|
#include "network_backends.h" |
#include "network_backends.h" |
|
|
#include "network.h" |
#include "network.h" |
#include "fdevent.h" |
|
#include "log.h" |
#include "log.h" |
#include "stat_cache.h" |
|
|
|
#include "sys-socket.h" |
#include "sys-socket.h" |
|
|
#include <sys/types.h> |
|
#include <sys/stat.h> |
|
#include <sys/time.h> |
|
#include <errno.h> |
|
#include <fcntl.h> |
|
#include <unistd.h> |
#include <unistd.h> |
|
|
|
#include <errno.h> |
#include <string.h> |
#include <string.h> |
#include <stdlib.h> |
|
|
|
#ifdef HAVE_SYS_FILIO_H | int network_write_mem_chunk(server *srv, connection *con, int fd, chunkqueue *cq, off_t *p_max_bytes) { |
# include <sys/filio.h> | chunk* const c = cq->first; |
#endif | off_t c_len; |
| ssize_t r; |
| UNUSED(con); |
|
|
#ifdef HAVE_SYS_RESOURCE_H | force_assert(NULL != c); |
# include <sys/resource.h> | force_assert(MEM_CHUNK == c->type); |
#endif | force_assert(c->offset >= 0 && c->offset <= (off_t)buffer_string_length(c->mem)); |
|
|
int network_write_chunkqueue_write(server *srv, connection *con, int fd, chunkqueue *cq, off_t max_bytes) { | c_len = buffer_string_length(c->mem) - c->offset; |
chunk *c; | if (c_len > *p_max_bytes) c_len = *p_max_bytes; |
|
|
for(c = cq->first; (max_bytes > 0) && (NULL != c); c = c->next) { | if (0 == c_len) { |
int chunk_finished = 0; | chunkqueue_remove_finished_chunks(cq); |
| return 0; |
| } |
|
|
switch(c->type) { | #if defined(__WIN32) |
case MEM_CHUNK: { | if ((r = send(fd, c->mem->ptr + c->offset, c_len, 0)) < 0) { |
char * offset; | int lastError = WSAGetLastError(); |
off_t toSend; | switch (lastError) { |
ssize_t r; | case WSAEINTR: |
| case WSAEWOULDBLOCK: |
if (c->mem->used == 0) { | |
chunk_finished = 1; | |
break; | |
} | |
| |
offset = c->mem->ptr + c->offset; | |
toSend = c->mem->used - 1 - c->offset; | |
if (toSend > max_bytes) toSend = max_bytes; | |
| |
#ifdef __WIN32 | |
if ((r = send(fd, offset, toSend, 0)) < 0) { | |
/* no error handling for windows... */ | |
log_error_write(srv, __FILE__, __LINE__, "ssd", "send failed: ", strerror(errno), fd); | |
| |
return -1; | |
} | |
#else | |
if ((r = write(fd, offset, toSend)) < 0) { | |
switch (errno) { | |
case EAGAIN: | |
case EINTR: | |
r = 0; | |
break; | |
case EPIPE: | |
case ECONNRESET: | |
return -2; | |
default: | |
log_error_write(srv, __FILE__, __LINE__, "ssd", | |
"write failed:", strerror(errno), fd); | |
| |
return -1; | |
} | |
} | |
#endif | |
| |
c->offset += r; | |
cq->bytes_out += r; | |
max_bytes -= r; | |
| |
if (c->offset == (off_t)c->mem->used - 1) { | |
chunk_finished = 1; | |
} | |
| |
break; |
break; |
|
case WSAECONNRESET: |
|
case WSAETIMEDOUT: |
|
case WSAECONNABORTED: |
|
return -2; |
|
default: |
|
log_error_write(srv, __FILE__, __LINE__, "sdd", |
|
"send failed: ", lastError, fd); |
|
return -1; |
} |
} |
case FILE_CHUNK: { | } |
#ifdef USE_MMAP | |
char *p = NULL; | |
#endif | |
ssize_t r; | |
off_t offset; | |
off_t toSend; | |
stat_cache_entry *sce = NULL; | |
int ifd; | |
| |
if (HANDLER_ERROR == stat_cache_get_entry(srv, con, c->file.name, &sce)) { | |
log_error_write(srv, __FILE__, __LINE__, "sb", | |
strerror(errno), c->file.name); | |
return -1; | |
} | |
| |
offset = c->file.start + c->offset; | |
toSend = c->file.length - c->offset; | |
| |
if (toSend > max_bytes) toSend = max_bytes; | |
| |
if (offset > sce->st.st_size) { | |
log_error_write(srv, __FILE__, __LINE__, "sb", "file was shrinked:", c->file.name); | |
| |
return -1; | |
} | |
| |
if (-1 == (ifd = open(c->file.name->ptr, O_RDONLY))) { | |
log_error_write(srv, __FILE__, __LINE__, "ss", "open failed: ", strerror(errno)); | |
| |
return -1; | |
} | |
| |
#ifdef USE_MMAP | |
if (MAP_FAILED == (p = mmap(0, sce->st.st_size, PROT_READ, MAP_SHARED, ifd, 0))) { | |
log_error_write(srv, __FILE__, __LINE__, "ss", "mmap failed: ", strerror(errno)); | |
| |
close(ifd); | |
| |
return -1; | |
} | |
close(ifd); | |
| |
if ((r = write(fd, p + offset, toSend)) <= 0) { | |
switch (errno) { | |
case EAGAIN: | |
case EINTR: | |
r = 0; | |
break; | |
case EPIPE: | |
case ECONNRESET: | |
munmap(p, sce->st.st_size); | |
return -2; | |
default: | |
log_error_write(srv, __FILE__, __LINE__, "ssd", | |
"write failed:", strerror(errno), fd); | |
munmap(p, sce->st.st_size); | |
| |
return -1; | |
} | |
} | |
| |
munmap(p, sce->st.st_size); | |
#else /* USE_MMAP */ | |
buffer_prepare_copy(srv->tmp_buf, toSend); | |
| |
if (-1 == lseek(ifd, offset, SEEK_SET)) { | |
log_error_write(srv, __FILE__, __LINE__, "ss", "lseek: ", strerror(errno)); | |
close(ifd); | |
return -1; | |
} | |
if (-1 == (toSend = read(ifd, srv->tmp_buf->ptr, toSend))) { | |
log_error_write(srv, __FILE__, __LINE__, "ss", "read: ", strerror(errno)); | |
close(ifd); | |
return -1; | |
} | |
close(ifd); | |
| |
#ifdef __WIN32 | |
if ((r = send(fd, srv->tmp_buf->ptr, toSend, 0)) < 0) { | |
/* no error handling for windows... */ | |
log_error_write(srv, __FILE__, __LINE__, "ssd", "send failed: ", strerror(errno), fd); | |
| |
return -1; | |
} | |
#else /* __WIN32 */ |
#else /* __WIN32 */ |
if ((r = write(fd, srv->tmp_buf->ptr, toSend)) < 0) { | if ((r = write(fd, c->mem->ptr + c->offset, c_len)) < 0) { |
switch (errno) { | switch (errno) { |
case EAGAIN: | case EAGAIN: |
case EINTR: | case EINTR: |
r = 0; | break; |
break; | case EPIPE: |
case EPIPE: | case ECONNRESET: |
case ECONNRESET: | return -2; |
return -2; | default: |
default: | log_error_write(srv, __FILE__, __LINE__, "ssd", |
log_error_write(srv, __FILE__, __LINE__, "ssd", | "write failed:", strerror(errno), fd); |
"write failed:", strerror(errno), fd); | return -1; |
| } |
return -1; | } |
} | |
} | |
#endif /* __WIN32 */ |
#endif /* __WIN32 */ |
#endif /* USE_MMAP */ |
|
|
|
c->offset += r; | if (r >= 0) { |
cq->bytes_out += r; | *p_max_bytes -= r; |
max_bytes -= r; | chunkqueue_mark_written(cq, r); |
| } |
|
|
if (c->offset == c->file.length) { | return (r > 0 && r == c_len) ? 0 : -3; |
chunk_finished = 1; | } |
} | |
|
|
|
int network_write_chunkqueue_write(server *srv, connection *con, int fd, chunkqueue *cq, off_t max_bytes) { |
|
while (max_bytes > 0 && NULL != cq->first) { |
|
int r = -1; |
|
|
|
switch (cq->first->type) { |
|
case MEM_CHUNK: |
|
r = network_write_mem_chunk(srv, con, fd, cq, &max_bytes); |
break; |
break; |
|
case FILE_CHUNK: |
|
r = network_write_file_chunk_mmap(srv, con, fd, cq, &max_bytes); |
|
break; |
} |
} |
default: |
|
|
|
log_error_write(srv, __FILE__, __LINE__, "ds", c, "type not known"); | if (-3 == r) return 0; |
| if (0 != r) return r; |
| } |
|
|
return -1; | return 0; |
} | } |
|
|
if (!chunk_finished) { | #if defined(USE_SENDFILE) |
/* not finished yet */ | int network_write_chunkqueue_sendfile(server *srv, connection *con, int fd, chunkqueue *cq, off_t max_bytes) { |
| while (max_bytes > 0 && NULL != cq->first) { |
| int r = -1; |
|
|
|
switch (cq->first->type) { |
|
case MEM_CHUNK: |
|
r = network_writev_mem_chunks(srv, con, fd, cq, &max_bytes); |
break; |
break; |
|
case FILE_CHUNK: |
|
r = network_write_file_chunk_sendfile(srv, con, fd, cq, &max_bytes); |
|
break; |
} |
} |
|
|
|
if (-3 == r) return 0; |
|
if (0 != r) return r; |
} |
} |
|
|
return 0; |
return 0; |
} |
|
|
|
#if 0 |
|
network_write_init(void) { |
|
p->write = network_write_write_chunkset; |
|
} |
} |
#endif |
#endif |