|
Packit |
eace71 |
/*
|
|
Packit |
eace71 |
* Copyright (C) 2002 Cisco Systems, Inc.
|
|
Packit |
eace71 |
* maintained by linux-iscsi-devel@lists.sourceforge.net
|
|
Packit |
eace71 |
*
|
|
Packit |
eace71 |
* This program is free software; you can redistribute it and/or modify
|
|
Packit |
eace71 |
* it under the terms of the GNU General Public License as published
|
|
Packit |
eace71 |
* by the Free Software Foundation; either version 2 of the License, or
|
|
Packit |
eace71 |
* (at your option) any later version.
|
|
Packit |
eace71 |
*
|
|
Packit |
eace71 |
* This program is distributed in the hope that it will be useful, but
|
|
Packit |
eace71 |
* WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
Packit |
eace71 |
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
Packit |
eace71 |
* General Public License for more details.
|
|
Packit |
eace71 |
*
|
|
Packit |
eace71 |
* See the file COPYING included with this distribution for more details.
|
|
Packit |
eace71 |
*/
|
|
Packit |
eace71 |
|
|
Packit |
eace71 |
#include <stdio.h>
|
|
Packit |
eace71 |
#include <stdlib.h>
|
|
Packit |
eace71 |
#include <stdarg.h>
|
|
Packit |
eace71 |
#include <string.h>
|
|
Packit |
eace71 |
#include <unistd.h>
|
|
Packit |
eace71 |
#include <errno.h>
|
|
Packit |
eace71 |
#include <sys/param.h>
|
|
Packit |
eace71 |
|
|
Packit |
eace71 |
#include "local_strings.h"
|
|
Packit |
eace71 |
#include "log.h"
|
|
Packit |
eace71 |
|
|
Packit |
eace71 |
int str_init_buffer(struct str_buffer *s, size_t initial_allocation)
|
|
Packit |
eace71 |
{
|
|
Packit |
eace71 |
if (s) {
|
|
Packit |
eace71 |
memset(s, 0, sizeof (*s));
|
|
Packit |
eace71 |
s->buffer = NULL;
|
|
Packit |
eace71 |
if (initial_allocation) {
|
|
Packit |
eace71 |
s->buffer = malloc(initial_allocation);
|
|
Packit |
eace71 |
if (s->buffer) {
|
|
Packit |
eace71 |
s->allocated_length = initial_allocation;
|
|
Packit |
eace71 |
memset(s->buffer, 0, initial_allocation);
|
|
Packit |
eace71 |
}
|
|
Packit |
eace71 |
}
|
|
Packit |
eace71 |
s->data_length = 0;
|
|
Packit |
eace71 |
return 1;
|
|
Packit |
eace71 |
}
|
|
Packit |
eace71 |
|
|
Packit |
eace71 |
return 0;
|
|
Packit |
eace71 |
}
|
|
Packit |
eace71 |
|
|
Packit |
eace71 |
struct str_buffer *str_alloc_buffer(size_t initial_allocation)
|
|
Packit |
eace71 |
{
|
|
Packit |
eace71 |
struct str_buffer *s = calloc(1, sizeof (*s));
|
|
Packit |
eace71 |
|
|
Packit |
eace71 |
if (s)
|
|
Packit |
eace71 |
str_init_buffer(s, initial_allocation);
|
|
Packit |
eace71 |
|
|
Packit |
eace71 |
return s;
|
|
Packit |
eace71 |
}
|
|
Packit |
eace71 |
|
|
Packit |
eace71 |
void str_free_buffer(struct str_buffer *s)
|
|
Packit |
eace71 |
{
|
|
Packit |
eace71 |
if (s) {
|
|
Packit |
eace71 |
if (s->buffer) {
|
|
Packit |
eace71 |
free(s->buffer);
|
|
Packit |
eace71 |
s->buffer = NULL;
|
|
Packit |
eace71 |
}
|
|
Packit |
eace71 |
s->allocated_length = 0;
|
|
Packit |
eace71 |
s->data_length = 0;
|
|
Packit |
eace71 |
}
|
|
Packit |
eace71 |
}
|
|
Packit |
eace71 |
|
|
Packit |
eace71 |
int str_enlarge_data(struct str_buffer *s, int length)
|
|
Packit |
eace71 |
{
|
|
Packit |
eace71 |
void *new_buf;
|
|
Packit |
eace71 |
|
|
Packit |
eace71 |
if (s) {
|
|
Packit |
eace71 |
s->data_length += length;
|
|
Packit |
eace71 |
if (s->data_length > s->allocated_length) {
|
|
Packit |
eace71 |
log_debug(7, "enlarge buffer from %lu to %lu",
|
|
Packit |
eace71 |
s->allocated_length, s->data_length);
|
|
Packit |
eace71 |
new_buf = realloc(s->buffer, s->data_length);
|
|
Packit |
eace71 |
if (!new_buf) {
|
|
Packit |
eace71 |
/* too big */
|
|
Packit |
eace71 |
log_error("enlarged buffer %p to %d data "
|
|
Packit |
eace71 |
"bytes, with only %d bytes of buffer "
|
|
Packit |
eace71 |
"space", s, (int)s->data_length,
|
|
Packit |
eace71 |
(int)s->allocated_length);
|
|
Packit |
eace71 |
return ENOMEM;
|
|
Packit |
eace71 |
}
|
|
Packit |
eace71 |
s->buffer = new_buf;
|
|
Packit |
eace71 |
memset(s->buffer + s->allocated_length, 0,
|
|
Packit |
eace71 |
s->data_length - s->allocated_length);
|
|
Packit |
eace71 |
s->allocated_length = s->data_length;
|
|
Packit |
eace71 |
}
|
|
Packit |
eace71 |
}
|
|
Packit |
eace71 |
|
|
Packit |
eace71 |
return 0;
|
|
Packit |
eace71 |
}
|
|
Packit |
eace71 |
|
|
Packit |
eace71 |
void str_remove_initial(struct str_buffer *s, int length)
|
|
Packit |
eace71 |
{
|
|
Packit |
eace71 |
char *remaining;
|
|
Packit |
eace71 |
int amount;
|
|
Packit |
eace71 |
|
|
Packit |
eace71 |
if (s && length) {
|
|
Packit |
eace71 |
remaining = s->buffer + length;
|
|
Packit |
eace71 |
amount = s->data_length - length;
|
|
Packit |
eace71 |
|
|
Packit |
eace71 |
if (amount < 0)
|
|
Packit |
eace71 |
amount = 0;
|
|
Packit |
eace71 |
if (amount)
|
|
Packit |
eace71 |
memmove(s->buffer, remaining, amount);
|
|
Packit |
eace71 |
s->data_length = amount;
|
|
Packit |
eace71 |
s->buffer[amount] = '\0';
|
|
Packit |
eace71 |
}
|
|
Packit |
eace71 |
}
|
|
Packit |
eace71 |
|
|
Packit |
eace71 |
/* truncate the data length down */
|
|
Packit |
eace71 |
void str_truncate_buffer(struct str_buffer *s, size_t length)
|
|
Packit |
eace71 |
{
|
|
Packit |
eace71 |
if (s) {
|
|
Packit |
eace71 |
if (!s->data_length)
|
|
Packit |
eace71 |
return;
|
|
Packit |
eace71 |
if (length <= s->data_length) {
|
|
Packit |
eace71 |
s->data_length = length;
|
|
Packit |
eace71 |
s->buffer[s->data_length] = '\0';
|
|
Packit |
eace71 |
} else if (length <= s->allocated_length) {
|
|
Packit |
eace71 |
/* clear the data, and declare the
|
|
Packit |
eace71 |
* data length to be larger
|
|
Packit |
eace71 |
*/
|
|
Packit |
eace71 |
memset(s->buffer + s->data_length, 0,
|
|
Packit |
eace71 |
length - s->data_length);
|
|
Packit |
eace71 |
s->data_length = length;
|
|
Packit |
eace71 |
} else {
|
|
Packit |
eace71 |
log_error(
|
|
Packit |
eace71 |
"couldn't truncate data buffer to length %d, "
|
|
Packit |
eace71 |
"only allocated %d",
|
|
Packit |
eace71 |
(int)length, (int)s->allocated_length);
|
|
Packit |
eace71 |
}
|
|
Packit |
eace71 |
}
|
|
Packit |
eace71 |
}
|
|
Packit |
eace71 |
|
|
Packit |
eace71 |
char *str_buffer_data(struct str_buffer *s)
|
|
Packit |
eace71 |
{
|
|
Packit |
eace71 |
if (s)
|
|
Packit |
eace71 |
return s->buffer;
|
|
Packit |
eace71 |
else
|
|
Packit |
eace71 |
return NULL;
|
|
Packit |
eace71 |
}
|
|
Packit |
eace71 |
|
|
Packit |
eace71 |
size_t str_data_length(struct str_buffer * s)
|
|
Packit |
eace71 |
{
|
|
Packit |
eace71 |
if (s)
|
|
Packit |
eace71 |
return s->data_length;
|
|
Packit |
eace71 |
else
|
|
Packit |
eace71 |
return 0;
|
|
Packit |
eace71 |
}
|
|
Packit |
eace71 |
|
|
Packit |
eace71 |
size_t str_unused_length(struct str_buffer * s)
|
|
Packit |
eace71 |
{
|
|
Packit |
eace71 |
if (s)
|
|
Packit |
eace71 |
return s->allocated_length - s->data_length;
|
|
Packit |
eace71 |
else
|
|
Packit |
eace71 |
return 0;
|
|
Packit |
eace71 |
}
|