mirror of
https://github.com/xcat2/xNBA.git
synced 2024-12-16 16:21:32 +00:00
c42a384709
Make expand_buffer() a non-inline function, so that we can add debug messages. Expandable buffers now don't *always* round up to the next power of two. This allows e.g. loading a 137MB ISO image on a machine with only 256MB RAM...
81 lines
2.2 KiB
C
81 lines
2.2 KiB
C
/*
|
|
* Copyright (C) 2007 Michael Brown <mbrown@fensystems.co.uk>.
|
|
*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License as
|
|
* published by the Free Software Foundation; either version 2 of the
|
|
* License, or any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful, but
|
|
* WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
|
|
*/
|
|
|
|
/**
|
|
* @file
|
|
*
|
|
* Automatically expanding buffers
|
|
*
|
|
*/
|
|
|
|
#include <errno.h>
|
|
#include <gpxe/buffer.h>
|
|
#include <gpxe/umalloc.h>
|
|
#include <gpxe/ebuffer.h>
|
|
|
|
/**
|
|
* Expand expandable buffer
|
|
*
|
|
* @v buffer Buffer descriptor
|
|
* @v new_len Required new size
|
|
* @ret rc Return status code
|
|
*/
|
|
static int ebuffer_expand ( struct buffer *buffer, size_t new_len ) {
|
|
size_t actual_len = 1;
|
|
userptr_t new_addr;
|
|
|
|
/* Round new_len up to the nearest power of two, to reduce
|
|
* total number of reallocations required. Don't do this for
|
|
* the first expansion; this allows for protocols that do
|
|
* actually know the exact length in advance.
|
|
*/
|
|
if ( buffer->len ) {
|
|
while ( actual_len < new_len )
|
|
actual_len <<= 1;
|
|
} else {
|
|
actual_len = new_len;
|
|
}
|
|
|
|
/* Reallocate buffer */
|
|
#warning "urealloc() has issues with length zero"
|
|
new_addr = urealloc ( buffer->addr, // actual_len );
|
|
actual_len ? actual_len : 1 );
|
|
if ( ! new_addr )
|
|
return -ENOMEM;
|
|
|
|
buffer->addr = new_addr;
|
|
buffer->len = actual_len;
|
|
return 0;
|
|
}
|
|
|
|
/**
|
|
* Allocate expandable buffer
|
|
*
|
|
* @v buffer Buffer descriptor
|
|
* @v len Initial length (may be zero)
|
|
* @ret rc Return status code
|
|
*
|
|
* Allocates space for the buffer and stores it in @c buffer->addr.
|
|
* The space must eventually be freed by calling ufree(buffer->addr).
|
|
*/
|
|
int ebuffer_alloc ( struct buffer *buffer, size_t len ) {
|
|
memset ( buffer, 0, sizeof ( *buffer ) );
|
|
buffer->expand = ebuffer_expand;
|
|
return ebuffer_expand ( buffer, len );
|
|
}
|