3 memcached_return
memcached_string_check(memcached_string_st
*string
, size_t need
)
5 if (need
&& need
> (size_t)(string
->current_size
- (size_t)(string
->end
- string
->string
)))
7 size_t current_offset
= string
->end
- string
->string
;
12 /* This is the block multiplier. To keep it larger and surive division errors we must round it up */
13 adjust
= (need
- (size_t)(string
->current_size
- (size_t)(string
->end
- string
->string
))) / string
->block_size
;
16 new_size
= sizeof(char) * (size_t)((adjust
* string
->block_size
) + string
->current_size
);
17 /* Test for overflow */
19 return MEMCACHED_MEMORY_ALLOCATION_FAILURE
;
21 if (string
->root
->call_realloc
)
22 new_value
= (char *)string
->root
->call_realloc(string
->root
, string
->string
, new_size
);
24 new_value
= (char *)realloc(string
->string
, new_size
);
26 if (new_value
== NULL
)
27 return MEMCACHED_MEMORY_ALLOCATION_FAILURE
;
29 string
->string
= new_value
;
30 string
->end
= string
->string
+ current_offset
;
32 string
->current_size
+= (string
->block_size
* adjust
);
35 return MEMCACHED_SUCCESS
;
38 memcached_string_st
*memcached_string_create(memcached_st
*ptr
, memcached_string_st
*string
, size_t initial_size
)
42 /* Saving malloc calls :) */
44 memset(string
, 0, sizeof(memcached_string_st
));
48 string
= (memcached_string_st
*)ptr
->call_malloc(ptr
, sizeof(memcached_string_st
));
50 string
= (memcached_string_st
*)calloc(1, sizeof(memcached_string_st
));
54 string
->is_allocated
= true;
56 string
->block_size
= MEMCACHED_BLOCK_SIZE
;
59 rc
= memcached_string_check(string
, initial_size
);
60 if (rc
!= MEMCACHED_SUCCESS
)
63 ptr
->call_free(ptr
, string
);
70 WATCHPOINT_ASSERT(string
->string
== string
->end
);
75 memcached_return
memcached_string_append_character(memcached_string_st
*string
,
80 rc
= memcached_string_check(string
, 1);
82 if (rc
!= MEMCACHED_SUCCESS
)
85 *string
->end
= character
;
88 return MEMCACHED_SUCCESS
;
91 memcached_return
memcached_string_append(memcached_string_st
*string
,
92 char *value
, size_t length
)
96 rc
= memcached_string_check(string
, length
);
98 if (rc
!= MEMCACHED_SUCCESS
)
101 WATCHPOINT_ASSERT(length
<= string
->current_size
);
102 WATCHPOINT_ASSERT(string
->string
);
103 WATCHPOINT_ASSERT(string
->end
>= string
->string
);
105 memcpy(string
->end
, value
, length
);
106 string
->end
+= length
;
108 return MEMCACHED_SUCCESS
;
111 char *memcached_string_c_copy(memcached_string_st
*string
)
115 if (memcached_string_length(string
) == 0)
118 if (string
->root
->call_malloc
)
119 c_ptr
= (char *)string
->root
->call_malloc(string
->root
, (memcached_string_length(string
)+1) * sizeof(char));
121 c_ptr
= (char *)malloc((memcached_string_length(string
)+1) * sizeof(char));
126 memcpy(c_ptr
, memcached_string_value(string
), memcached_string_length(string
));
127 c_ptr
[memcached_string_length(string
)]= 0;
132 memcached_return
memcached_string_reset(memcached_string_st
*string
)
134 string
->end
= string
->string
;
136 return MEMCACHED_SUCCESS
;
139 void memcached_string_free(memcached_string_st
*ptr
)
146 if (ptr
->root
->call_free
)
147 ptr
->root
->call_free(ptr
->root
, ptr
->string
);
152 if (ptr
->is_allocated
)
154 if (ptr
->root
->call_free
)
155 ptr
->root
->call_free(ptr
->root
, ptr
);
160 memset(ptr
, 0, sizeof(memcached_string_st
));