| /* Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include "win32/apr_arch_file_io.h" |
| #include "apr_file_io.h" |
| #include "apr_general.h" |
| #include "apr_strings.h" |
| #include <string.h> |
| #include "apr_arch_inherit.h" |
| #include <io.h> /* for [_open/_get]_osfhandle */ |
| |
| |
| APR_DECLARE(apr_status_t) apr_file_dup(apr_file_t **new_file, |
| apr_file_t *old_file, apr_pool_t *p) |
| { |
| #ifdef _WIN32_WCE |
| return APR_ENOTIMPL; |
| #else |
| HANDLE hproc = GetCurrentProcess(); |
| HANDLE newhand = NULL; |
| |
| if (!DuplicateHandle(hproc, old_file->filehand, |
| hproc, &newhand, 0, FALSE, |
| DUPLICATE_SAME_ACCESS)) { |
| return apr_get_os_error(); |
| } |
| |
| (*new_file) = (apr_file_t *) apr_pcalloc(p, sizeof(apr_file_t)); |
| (*new_file)->filehand = newhand; |
| (*new_file)->flags = old_file->flags & ~(APR_STD_FLAGS | APR_INHERIT); |
| (*new_file)->pool = p; |
| (*new_file)->fname = apr_pstrdup(p, old_file->fname); |
| (*new_file)->append = old_file->append; |
| (*new_file)->buffered = FALSE; |
| (*new_file)->ungetchar = old_file->ungetchar; |
| |
| #if APR_HAS_THREADS |
| if (old_file->mutex) { |
| apr_thread_mutex_create(&((*new_file)->mutex), |
| APR_THREAD_MUTEX_DEFAULT, p); |
| } |
| #endif |
| |
| apr_pool_cleanup_register((*new_file)->pool, (void *)(*new_file), file_cleanup, |
| apr_pool_cleanup_null); |
| |
| /* Create a pollset with room for one descriptor. */ |
| /* ### check return codes */ |
| (void) apr_pollset_create(&(*new_file)->pollset, 1, p, 0); |
| |
| return APR_SUCCESS; |
| #endif /* !defined(_WIN32_WCE) */ |
| } |
| |
| APR_DECLARE(apr_status_t) apr_file_dup2(apr_file_t *new_file, |
| apr_file_t *old_file, apr_pool_t *p) |
| { |
| #ifdef _WIN32_WCE |
| return APR_ENOTIMPL; |
| #else |
| DWORD stdhandle = 0; |
| HANDLE hproc = GetCurrentProcess(); |
| HANDLE newhand = NULL; |
| apr_int32_t newflags; |
| int fd; |
| |
| if (new_file->flags & APR_STD_FLAGS) |
| { |
| if ((new_file->flags & APR_STD_FLAGS) == APR_STDERR_FLAG) |
| { |
| /* Flush stderr and unset its buffer, then commit the fd-based buffer. |
| * This is typically a noop for Win2K/XP since services with NULL std |
| * handles [but valid FILE *'s, oddly enough], but is required |
| * for NT 4.0 and to use this code outside of services. |
| */ |
| fflush(stderr); |
| setvbuf(stderr, NULL, _IONBF, 0); |
| _commit(2 /* stderr */); |
| |
| /* Clone a handle can _close() without harming the source handle, |
| * open an MSVCRT-based pseudo-fd for the file handle, then dup2 |
| * and close our temporary pseudo-fd once it's been duplicated. |
| * This will incidently keep the FILE-based stderr in sync. |
| * Note the apparently redundant _O_BINARY coersions are required. |
| * Note the _dup2 will close the previous std Win32 handle. |
| */ |
| if (!DuplicateHandle(hproc, old_file->filehand, hproc, &newhand, |
| 0, FALSE, DUPLICATE_SAME_ACCESS)) { |
| return apr_get_os_error(); |
| } |
| fd = _open_osfhandle((INT_PTR)newhand, _O_WRONLY | _O_BINARY); |
| _dup2(fd, 2); |
| _close(fd); |
| _setmode(2, _O_BINARY); |
| |
| /* hPipeWrite was _close()'ed above, and _dup2()'ed |
| * to fd 2 creating a new, inherited Win32 handle. |
| * Recover that real handle from fd 2. Note that |
| * SetStdHandle(STD_ERROR_HANDLE, _get_osfhandle(2)) |
| * is implicit in the dup2() call above |
| */ |
| newhand = (HANDLE)_get_osfhandle(2); |
| } |
| else if ((new_file->flags & APR_STD_FLAGS) == APR_STDOUT_FLAG) { |
| /* For the process flow see the stderr case above */ |
| fflush(stdout); |
| setvbuf(stdout, NULL, _IONBF, 0); |
| _commit(1 /* stdout */); |
| |
| if (!DuplicateHandle(hproc, old_file->filehand, hproc, &newhand, |
| 0, FALSE, DUPLICATE_SAME_ACCESS)) { |
| return apr_get_os_error(); |
| } |
| fd = _open_osfhandle((INT_PTR)newhand, _O_WRONLY | _O_BINARY); |
| _dup2(fd, 1); |
| _close(fd); |
| _setmode(1, _O_BINARY); |
| newhand = (HANDLE)_get_osfhandle(1); |
| } |
| else if ((new_file->flags & APR_STD_FLAGS) == APR_STDIN_FLAG) { |
| /* For the process flow see the stderr case above */ |
| fflush(stdin); |
| setvbuf(stdin, NULL, _IONBF, 0); |
| _commit(0 /* stdin */); |
| |
| if (!DuplicateHandle(hproc, old_file->filehand, hproc, &newhand, |
| 0, FALSE, DUPLICATE_SAME_ACCESS)) { |
| return apr_get_os_error(); |
| } |
| fd = _open_osfhandle((INT_PTR)newhand, _O_RDONLY | _O_BINARY); |
| _dup2(fd, 0); |
| _close(fd); |
| _setmode(0, _O_BINARY); |
| newhand = (HANDLE)_get_osfhandle(0); |
| } |
| newflags = (new_file->flags & APR_STD_FLAGS) |
| | (old_file->flags & ~APR_STD_FLAGS) | APR_INHERIT; |
| |
| /* No need to close the old file, _dup2() above did that for us */ |
| } |
| else { |
| if (!DuplicateHandle(hproc, old_file->filehand, |
| hproc, &newhand, 0, |
| FALSE, DUPLICATE_SAME_ACCESS)) { |
| return apr_get_os_error(); |
| } |
| newflags = old_file->flags & ~(APR_STD_FLAGS | APR_INHERIT); |
| |
| if (new_file->filehand |
| && (new_file->filehand != INVALID_HANDLE_VALUE)) { |
| CloseHandle(new_file->filehand); |
| } |
| } |
| |
| new_file->flags = newflags; |
| new_file->filehand = newhand; |
| new_file->fname = apr_pstrdup(new_file->pool, old_file->fname); |
| new_file->append = old_file->append; |
| new_file->buffered = FALSE; |
| new_file->ungetchar = old_file->ungetchar; |
| |
| #if APR_HAS_THREADS |
| if (old_file->mutex) { |
| apr_thread_mutex_create(&(new_file->mutex), |
| APR_THREAD_MUTEX_DEFAULT, p); |
| } |
| #endif |
| |
| return APR_SUCCESS; |
| #endif /* !defined(_WIN32_WCE) */ |
| } |
| |
| APR_DECLARE(apr_status_t) apr_file_setaside(apr_file_t **new_file, |
| apr_file_t *old_file, |
| apr_pool_t *p) |
| { |
| *new_file = (apr_file_t *)apr_palloc(p, sizeof(apr_file_t)); |
| memcpy(*new_file, old_file, sizeof(apr_file_t)); |
| (*new_file)->pool = p; |
| if (old_file->buffered) { |
| (*new_file)->buffer = apr_palloc(p, APR_FILE_BUFSIZE); |
| if (old_file->direction == 1) { |
| memcpy((*new_file)->buffer, old_file->buffer, old_file->bufpos); |
| } |
| else { |
| memcpy((*new_file)->buffer, old_file->buffer, old_file->dataRead); |
| } |
| } |
| if (old_file->mutex) { |
| apr_thread_mutex_create(&((*new_file)->mutex), |
| APR_THREAD_MUTEX_DEFAULT, p); |
| apr_thread_mutex_destroy(old_file->mutex); |
| } |
| if (old_file->fname) { |
| (*new_file)->fname = apr_pstrdup(p, old_file->fname); |
| } |
| if (!(old_file->flags & APR_FILE_NOCLEANUP)) { |
| apr_pool_cleanup_register(p, (void *)(*new_file), |
| file_cleanup, |
| file_cleanup); |
| } |
| |
| old_file->filehand = INVALID_HANDLE_VALUE; |
| apr_pool_cleanup_kill(old_file->pool, (void *)old_file, |
| file_cleanup); |
| |
| /* Create a pollset with room for one descriptor. */ |
| /* ### check return codes */ |
| (void) apr_pollset_create(&(*new_file)->pollset, 1, p, 0); |
| |
| return APR_SUCCESS; |
| } |
