/* 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 "apr_arch_file_io.h"
#include "apr_thread_mutex.h"

APR_DECLARE(apr_status_t) apr_file_buffer_set(apr_file_t *file, 
                                              char * buffer,
                                              apr_size_t bufsize)
{
    apr_status_t rv = APR_SUCCESS;
    int do_locking = file->mutex != NULL && file->buffered;

    if (do_locking) {
        apr_thread_mutex_lock(file->mutex);
    }
 
    if (file->buffered) {
        /* Flush the existing buffer */
        rv = apr_file_flush(file);
        if (rv != APR_SUCCESS) {
            if (do_locking) {
                apr_thread_mutex_unlock(file->mutex);
            }

            return rv;
        }
    }
        
    file->buffer = buffer;
    file->bufsize = bufsize;
    file->bufpos = 0;
    file->direction = 0;
    file->dataRead = 0;

    if (bufsize > 0 && file->mutex == NULL && (file->flags & APR_FOPEN_XTHREAD)) {
        /* buffering is being turned on but we have no mutex, make one */
        rv = apr_thread_mutex_create(&file->mutex, 0, file->pool);
    }

    /* Setting the buffer size to zero is equivalent to turning 
     * buffering off. 
     */
    file->buffered = file->bufsize > 0;
    
    if (do_locking) {
        apr_thread_mutex_unlock(file->mutex);
    }

    return rv;
}

APR_DECLARE(apr_size_t) apr_file_buffer_size_get(apr_file_t *file)
{
    return file->bufsize;
}