| #include "network_backends.h" |
| |
| #ifdef USE_SOLARIS_SENDFILEV |
| |
| #include <sys/types.h> |
| #include <sys/socket.h> |
| #include <sys/stat.h> |
| #include <sys/time.h> |
| #include <sys/resource.h> |
| |
| #include <netinet/in.h> |
| #include <netinet/tcp.h> |
| |
| #include <errno.h> |
| #include <fcntl.h> |
| #include <unistd.h> |
| #include <netdb.h> |
| #include <string.h> |
| #include <stdlib.h> |
| #include <limits.h> |
| |
| #include "network.h" |
| #include "fdevent.h" |
| #include "log.h" |
| #include "stat_cache.h" |
| |
| #ifndef UIO_MAXIOV |
| #define UIO_MAXIOV IOV_MAX |
| #endif |
| |
| /** |
| * a very simple sendfilev() interface for solaris which can be optimised a lot more |
| * as solaris sendfilev() supports 'sending everythin in one syscall()' |
| * |
| * If you want such an interface and need the performance, just give me an account on |
| * a solaris box. |
| * - jan@kneschke.de |
| */ |
| |
| |
| int network_write_chunkqueue_solarissendfilev(server *srv, connection *con, int fd, chunkqueue *cq) { |
| chunk *c; |
| size_t chunks_written = 0; |
| |
| for(c = cq->first; c; c = c->next, chunks_written++) { |
| int chunk_finished = 0; |
| |
| switch(c->type) { |
| case MEM_CHUNK: { |
| char * offset; |
| size_t toSend; |
| ssize_t r; |
| |
| size_t num_chunks, i; |
| struct iovec chunks[UIO_MAXIOV]; |
| chunk *tc; |
| |
| size_t num_bytes = 0; |
| |
| /* we can't send more then SSIZE_MAX bytes in one chunk */ |
| |
| /* build writev list |
| * |
| * 1. limit: num_chunks < UIO_MAXIOV |
| * 2. limit: num_bytes < SSIZE_MAX |
| */ |
| for(num_chunks = 0, tc = c; tc && tc->type == MEM_CHUNK && num_chunks < UIO_MAXIOV; num_chunks++, tc = tc->next); |
| |
| for(tc = c, i = 0; i < num_chunks; tc = tc->next, i++) { |
| if (tc->mem->used == 0) { |
| chunks[i].iov_base = tc->mem->ptr; |
| chunks[i].iov_len = 0; |
| } else { |
| offset = tc->mem->ptr + tc->offset; |
| toSend = tc->mem->used - 1 - tc->offset; |
| |
| chunks[i].iov_base = offset; |
| |
| /* protect the return value of writev() */ |
| if (toSend > SSIZE_MAX || |
| num_bytes + toSend > SSIZE_MAX) { |
| chunks[i].iov_len = SSIZE_MAX - num_bytes; |
| |
| num_chunks = i + 1; |
| break; |
| } else { |
| chunks[i].iov_len = toSend; |
| } |
| |
| num_bytes += toSend; |
| } |
| } |
| |
| if ((r = writev(fd, chunks, num_chunks)) < 0) { |
| switch (errno) { |
| case EAGAIN: |
| case EINTR: |
| r = 0; |
| break; |
| case EPIPE: |
| case ECONNRESET: |
| return -2; |
| default: |
| log_error_write(srv, __FILE__, __LINE__, "ssd", |
| "writev failed:", strerror(errno), fd); |
| |
| return -1; |
| } |
| } |
| |
| /* check which chunks have been written */ |
| cq->bytes_out += r; |
| |
| for(i = 0, tc = c; i < num_chunks; i++, tc = tc->next) { |
| if (r >= (ssize_t)chunks[i].iov_len) { |
| /* written */ |
| r -= chunks[i].iov_len; |
| tc->offset += chunks[i].iov_len; |
| |
| if (chunk_finished) { |
| /* skip the chunks from further touches */ |
| chunks_written++; |
| c = c->next; |
| } else { |
| /* chunks_written + c = c->next is done in the for()*/ |
| chunk_finished++; |
| } |
| } else { |
| /* partially written */ |
| |
| tc->offset += r; |
| chunk_finished = 0; |
| |
| break; |
| } |
| } |
| |
| break; |
| } |
| case FILE_CHUNK: { |
| ssize_t r; |
| off_t offset; |
| size_t toSend, written; |
| sendfilevec_t fvec; |
| 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 (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; |
| } |
| |
| fvec.sfv_fd = ifd; |
| fvec.sfv_flag = 0; |
| fvec.sfv_off = offset; |
| fvec.sfv_len = toSend; |
| |
| /* Solaris sendfilev() */ |
| if (-1 == (r = sendfilev(fd, &fvec, 1, &written))) { |
| if (errno != EAGAIN) { |
| log_error_write(srv, __FILE__, __LINE__, "ssd", "sendfile: ", strerror(errno), errno); |
| |
| close(ifd); |
| return -1; |
| } |
| |
| r = 0; |
| } |
| |
| close(ifd); |
| c->offset += written; |
| cq->bytes_out += written; |
| |
| if (c->offset == c->file.length) { |
| chunk_finished = 1; |
| } |
| |
| break; |
| } |
| default: |
| |
| log_error_write(srv, __FILE__, __LINE__, "ds", c, "type not known"); |
| |
| return -1; |
| } |
| |
| if (!chunk_finished) { |
| /* not finished yet */ |
| |
| break; |
| } |
| } |
| |
| return chunks_written; |
| } |
| |
| #endif |