1 /* 2 * Copyright (c) 2007-2012 Niels Provos and Nick Mathewson 3 * 4 * Redistribution and use in source and binary forms, with or without 5 * modification, are permitted provided that the following conditions 6 * are met: 7 * 1. Redistributions of source code must retain the above copyright 8 * notice, this list of conditions and the following disclaimer. 9 * 2. Redistributions in binary form must reproduce the above copyright 10 * notice, this list of conditions and the following disclaimer in the 11 * documentation and/or other materials provided with the distribution. 12 * 3. The name of the author may not be used to endorse or promote products 13 * derived from this software without specific prior written permission. 14 * 15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 16 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 17 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 18 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 19 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 20 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 21 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 22 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 23 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 24 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 25 */ 26 #ifndef EVENT2_BUFFER_H_INCLUDED_ 27 #define EVENT2_BUFFER_H_INCLUDED_ 28 29 /** @file event2/buffer.h 30 31 Functions for buffering data for network sending or receiving. 32 33 An evbuffer can be used for preparing data before sending it to 34 the network or conversely for reading data from the network. 35 Evbuffers try to avoid memory copies as much as possible. As a 36 result, evbuffers can be used to pass data around without actually 37 incurring the overhead of copying the data. 38 39 A new evbuffer can be allocated with evbuffer_new(), and can be 40 freed with evbuffer_free(). Most users will be using evbuffers via 41 the bufferevent interface. To access a bufferevent's evbuffers, use 42 bufferevent_get_input() and bufferevent_get_output(). 43 44 There are several guidelines for using evbuffers. 45 46 - if you already know how much data you are going to add as a result 47 of calling evbuffer_add() multiple times, it makes sense to use 48 evbuffer_expand() first to make sure that enough memory is allocated 49 before hand. 50 51 - evbuffer_add_buffer() adds the contents of one buffer to the other 52 without incurring any unnecessary memory copies. 53 54 - evbuffer_add() and evbuffer_add_buffer() do not mix very well: 55 if you use them, you will wind up with fragmented memory in your 56 buffer. 57 58 - For high-performance code, you may want to avoid copying data into and out 59 of buffers. You can skip the copy step by using 60 evbuffer_reserve_space()/evbuffer_commit_space() when writing into a 61 buffer, and evbuffer_peek() when reading. 62 63 In Libevent 2.0 and later, evbuffers are represented using a linked 64 list of memory chunks, with pointers to the first and last chunk in 65 the chain. 66 67 As the contents of an evbuffer can be stored in multiple different 68 memory blocks, it cannot be accessed directly. Instead, evbuffer_pullup() 69 can be used to force a specified number of bytes to be contiguous. This 70 will cause memory reallocation and memory copies if the data is split 71 across multiple blocks. It is more efficient, however, to use 72 evbuffer_peek() if you don't require that the memory to be contiguous. 73 */ 74 75 #ifdef __cplusplus 76 extern "C" { 77 #endif 78 79 #include <event2/event-config.h> 80 #include <stdarg.h> 81 #ifdef EVENT__HAVE_SYS_TYPES_H 82 #include <sys/types.h> 83 #endif 84 #ifdef EVENT__HAVE_SYS_UIO_H 85 #include <sys/uio.h> 86 #endif 87 #include <event2/util.h> 88 89 /** 90 An evbuffer is an opaque data type for efficiently buffering data to be 91 sent or received on the network. 92 93 @see event2/event.h for more information 94 */ 95 struct evbuffer 96 #ifdef EVENT_IN_DOXYGEN_ 97 {} 98 #endif 99 ; 100 101 /** 102 Pointer to a position within an evbuffer. 103 104 Used when repeatedly searching through a buffer. Calling any function 105 that modifies or re-packs the buffer contents may invalidate all 106 evbuffer_ptrs for that buffer. Do not modify these values except with 107 evbuffer_ptr_set. 108 109 Used when repeatedly searching through a buffer. Calls to any function 110 that modifies or re-packs the buffer contents may invalidate all 111 evbuffer_ptrs for that buffer. Do not modify these values except with 112 evbuffer_ptr_set. 113 114 An evbuffer_ptr can represent any position from the start of a buffer up 115 to a position immediately after the end of a buffer. 116 117 @see evbuffer_ptr_set() 118 */ 119 struct evbuffer_ptr { 120 ev_ssize_t pos; 121 122 /* Do not alter or rely on the values of fields: they are for internal 123 * use */ 124 struct { 125 void *chain; 126 size_t pos_in_chain; 127 } internal_; 128 }; 129 130 /** Describes a single extent of memory inside an evbuffer. Used for 131 direct-access functions. 132 133 @see evbuffer_reserve_space, evbuffer_commit_space, evbuffer_peek 134 */ 135 #ifdef EVENT__HAVE_SYS_UIO_H 136 #define evbuffer_iovec iovec 137 /* Internal use -- defined only if we are using the native struct iovec */ 138 #define EVBUFFER_IOVEC_IS_NATIVE_ 139 #else 140 struct evbuffer_iovec { 141 /** The start of the extent of memory. */ 142 void *iov_base; 143 /** The length of the extent of memory. */ 144 size_t iov_len; 145 }; 146 #endif 147 148 /** 149 Allocate storage for a new evbuffer. 150 151 @return a pointer to a newly allocated evbuffer struct, or NULL if an error 152 occurred 153 */ 154 struct evbuffer *evbuffer_new(void); 155 /** 156 Deallocate storage for an evbuffer. 157 158 @param buf pointer to the evbuffer to be freed 159 */ 160 void evbuffer_free(struct evbuffer *buf); 161 162 /** 163 Enable locking on an evbuffer so that it can safely be used by multiple 164 threads at the same time. 165 166 NOTE: when locking is enabled, the lock will be held when callbacks are 167 invoked. This could result in deadlock if you aren't careful. Plan 168 accordingly! 169 170 @param buf An evbuffer to make lockable. 171 @param lock A lock object, or NULL if we should allocate our own. 172 @return 0 on success, -1 on failure. 173 */ 174 int evbuffer_enable_locking(struct evbuffer *buf, void *lock); 175 176 /** 177 Acquire the lock on an evbuffer. Has no effect if locking was not enabled 178 with evbuffer_enable_locking. 179 */ 180 void evbuffer_lock(struct evbuffer *buf); 181 182 /** 183 Release the lock on an evbuffer. Has no effect if locking was not enabled 184 with evbuffer_enable_locking. 185 */ 186 void evbuffer_unlock(struct evbuffer *buf); 187 188 189 /** If this flag is set, then we will not use evbuffer_peek(), 190 * evbuffer_remove(), evbuffer_remove_buffer(), and so on to read bytes 191 * from this buffer: we'll only take bytes out of this buffer by 192 * writing them to the network (as with evbuffer_write_atmost), by 193 * removing them without observing them (as with evbuffer_drain), 194 * or by copying them all out at once (as with evbuffer_add_buffer). 195 * 196 * Using this option allows the implementation to use sendfile-based 197 * operations for evbuffer_add_file(); see that function for more 198 * information. 199 * 200 * This flag is on by default for bufferevents that can take advantage 201 * of it; you should never actually need to set it on a bufferevent's 202 * output buffer. 203 */ 204 #define EVBUFFER_FLAG_DRAINS_TO_FD 1 205 206 /** Change the flags that are set for an evbuffer by adding more. 207 * 208 * @param buffer the evbuffer that the callback is watching. 209 * @param cb the callback whose status we want to change. 210 * @param flags One or more EVBUFFER_FLAG_* options 211 * @return 0 on success, -1 on failure. 212 */ 213 int evbuffer_set_flags(struct evbuffer *buf, ev_uint64_t flags); 214 /** Change the flags that are set for an evbuffer by removing some. 215 * 216 * @param buffer the evbuffer that the callback is watching. 217 * @param cb the callback whose status we want to change. 218 * @param flags One or more EVBUFFER_FLAG_* options 219 * @return 0 on success, -1 on failure. 220 */ 221 int evbuffer_clear_flags(struct evbuffer *buf, ev_uint64_t flags); 222 223 /** 224 Returns the total number of bytes stored in the evbuffer 225 226 @param buf pointer to the evbuffer 227 @return the number of bytes stored in the evbuffer 228 */ 229 size_t evbuffer_get_length(const struct evbuffer *buf); 230 231 /** 232 Returns the number of contiguous available bytes in the first buffer chain. 233 234 This is useful when processing data that might be split into multiple 235 chains, or that might all be in the first chain. Calls to 236 evbuffer_pullup() that cause reallocation and copying of data can thus be 237 avoided. 238 239 @param buf pointer to the evbuffer 240 @return 0 if no data is available, otherwise the number of available bytes 241 in the first buffer chain. 242 */ 243 size_t evbuffer_get_contiguous_space(const struct evbuffer *buf); 244 245 /** 246 Expands the available space in an evbuffer. 247 248 Expands the available space in the evbuffer to at least datlen, so that 249 appending datlen additional bytes will not require any new allocations. 250 251 @param buf the evbuffer to be expanded 252 @param datlen the new minimum length requirement 253 @return 0 if successful, or -1 if an error occurred 254 */ 255 int evbuffer_expand(struct evbuffer *buf, size_t datlen); 256 257 /** 258 Reserves space in the last chain or chains of an evbuffer. 259 260 Makes space available in the last chain or chains of an evbuffer that can 261 be arbitrarily written to by a user. The space does not become 262 available for reading until it has been committed with 263 evbuffer_commit_space(). 264 265 The space is made available as one or more extents, represented by 266 an initial pointer and a length. You can force the memory to be 267 available as only one extent. Allowing more extents, however, makes the 268 function more efficient. 269 270 Multiple subsequent calls to this function will make the same space 271 available until evbuffer_commit_space() has been called. 272 273 It is an error to do anything that moves around the buffer's internal 274 memory structures before committing the space. 275 276 NOTE: The code currently does not ever use more than two extents. 277 This may change in future versions. 278 279 @param buf the evbuffer in which to reserve space. 280 @param size how much space to make available, at minimum. The 281 total length of the extents may be greater than the requested 282 length. 283 @param vec an array of one or more evbuffer_iovec structures to 284 hold pointers to the reserved extents of memory. 285 @param n_vec The length of the vec array. Must be at least 1; 286 2 is more efficient. 287 @return the number of provided extents, or -1 on error. 288 @see evbuffer_commit_space() 289 */ 290 int 291 evbuffer_reserve_space(struct evbuffer *buf, ev_ssize_t size, 292 struct evbuffer_iovec *vec, int n_vec); 293 294 /** 295 Commits previously reserved space. 296 297 Commits some of the space previously reserved with 298 evbuffer_reserve_space(). It then becomes available for reading. 299 300 This function may return an error if the pointer in the extents do 301 not match those returned from evbuffer_reserve_space, or if data 302 has been added to the buffer since the space was reserved. 303 304 If you want to commit less data than you got reserved space for, 305 modify the iov_len pointer of the appropriate extent to a smaller 306 value. Note that you may have received more space than you 307 requested if it was available! 308 309 @param buf the evbuffer in which to reserve space. 310 @param vec one or two extents returned by evbuffer_reserve_space. 311 @param n_vecs the number of extents. 312 @return 0 on success, -1 on error 313 @see evbuffer_reserve_space() 314 */ 315 int evbuffer_commit_space(struct evbuffer *buf, 316 struct evbuffer_iovec *vec, int n_vecs); 317 318 /** 319 Append data to the end of an evbuffer. 320 321 @param buf the evbuffer to be appended to 322 @param data pointer to the beginning of the data buffer 323 @param datlen the number of bytes to be copied from the data buffer 324 @return 0 on success, -1 on failure. 325 */ 326 int evbuffer_add(struct evbuffer *buf, const void *data, size_t datlen); 327 328 329 /** 330 Read data from an evbuffer and drain the bytes read. 331 332 If more bytes are requested than are available in the evbuffer, we 333 only extract as many bytes as were available. 334 335 @param buf the evbuffer to be read from 336 @param data the destination buffer to store the result 337 @param datlen the maximum size of the destination buffer 338 @return the number of bytes read, or -1 if we can't drain the buffer. 339 */ 340 int evbuffer_remove(struct evbuffer *buf, void *data, size_t datlen); 341 342 /** 343 Read data from an evbuffer, and leave the buffer unchanged. 344 345 If more bytes are requested than are available in the evbuffer, we 346 only extract as many bytes as were available. 347 348 @param buf the evbuffer to be read from 349 @param data_out the destination buffer to store the result 350 @param datlen the maximum size of the destination buffer 351 @return the number of bytes read, or -1 if we can't drain the buffer. 352 */ 353 ev_ssize_t evbuffer_copyout(struct evbuffer *buf, void *data_out, size_t datlen); 354 355 /** 356 Read data from the middle of an evbuffer, and leave the buffer unchanged. 357 358 If more bytes are requested than are available in the evbuffer, we 359 only extract as many bytes as were available. 360 361 @param buf the evbuffer to be read from 362 @param pos the position to start reading from 363 @param data_out the destination buffer to store the result 364 @param datlen the maximum size of the destination buffer 365 @return the number of bytes read, or -1 if we can't drain the buffer. 366 */ 367 ev_ssize_t evbuffer_copyout_from(struct evbuffer *buf, const struct evbuffer_ptr *pos, void *data_out, size_t datlen); 368 369 /** 370 Read data from an evbuffer into another evbuffer, draining 371 the bytes from the source buffer. This function avoids copy 372 operations to the extent possible. 373 374 If more bytes are requested than are available in src, the src 375 buffer is drained completely. 376 377 @param src the evbuffer to be read from 378 @param dst the destination evbuffer to store the result into 379 @param datlen the maximum numbers of bytes to transfer 380 @return the number of bytes read 381 */ 382 int evbuffer_remove_buffer(struct evbuffer *src, struct evbuffer *dst, 383 size_t datlen); 384 385 /** Used to tell evbuffer_readln what kind of line-ending to look for. 386 */ 387 enum evbuffer_eol_style { 388 /** Any sequence of CR and LF characters is acceptable as an 389 * EOL. 390 * 391 * Note that this style can produce ambiguous results: the 392 * sequence "CRLF" will be treated as a single EOL if it is 393 * all in the buffer at once, but if you first read a CR from 394 * the network and later read an LF from the network, it will 395 * be treated as two EOLs. 396 */ 397 EVBUFFER_EOL_ANY, 398 /** An EOL is an LF, optionally preceded by a CR. This style is 399 * most useful for implementing text-based internet protocols. */ 400 EVBUFFER_EOL_CRLF, 401 /** An EOL is a CR followed by an LF. */ 402 EVBUFFER_EOL_CRLF_STRICT, 403 /** An EOL is a LF. */ 404 EVBUFFER_EOL_LF, 405 /** An EOL is a NUL character (that is, a single byte with value 0) */ 406 EVBUFFER_EOL_NUL 407 }; 408 409 /** 410 * Read a single line from an evbuffer. 411 * 412 * Reads a line terminated by an EOL as determined by the evbuffer_eol_style 413 * argument. Returns a newly allocated nul-terminated string; the caller must 414 * free the returned value. The EOL is not included in the returned string. 415 * 416 * @param buffer the evbuffer to read from 417 * @param n_read_out if non-NULL, points to a size_t that is set to the 418 * number of characters in the returned string. This is useful for 419 * strings that can contain NUL characters. 420 * @param eol_style the style of line-ending to use. 421 * @return pointer to a single line, or NULL if an error occurred 422 */ 423 char *evbuffer_readln(struct evbuffer *buffer, size_t *n_read_out, 424 enum evbuffer_eol_style eol_style); 425 426 /** 427 Move all data from one evbuffer into another evbuffer. 428 429 This is a destructive add. The data from one buffer moves into 430 the other buffer. However, no unnecessary memory copies occur. 431 432 @param outbuf the output buffer 433 @param inbuf the input buffer 434 @return 0 if successful, or -1 if an error occurred 435 436 @see evbuffer_remove_buffer() 437 */ 438 int evbuffer_add_buffer(struct evbuffer *outbuf, struct evbuffer *inbuf); 439 440 /** 441 Copy data from one evbuffer into another evbuffer. 442 443 This is a non-destructive add. The data from one buffer is copied 444 into the other buffer. However, no unnecessary memory copies occur. 445 446 Note that buffers already containing buffer references can't be added 447 to other buffers. 448 449 @param outbuf the output buffer 450 @param inbuf the input buffer 451 @return 0 if successful, or -1 if an error occurred 452 */ 453 int evbuffer_add_buffer_reference(struct evbuffer *outbuf, 454 struct evbuffer *inbuf); 455 456 /** 457 A cleanup function for a piece of memory added to an evbuffer by 458 reference. 459 460 @see evbuffer_add_reference() 461 */ 462 typedef void (*evbuffer_ref_cleanup_cb)(const void *data, 463 size_t datalen, void *extra); 464 465 /** 466 Reference memory into an evbuffer without copying. 467 468 The memory needs to remain valid until all the added data has been 469 read. This function keeps just a reference to the memory without 470 actually incurring the overhead of a copy. 471 472 @param outbuf the output buffer 473 @param data the memory to reference 474 @param datlen how memory to reference 475 @param cleanupfn callback to be invoked when the memory is no longer 476 referenced by this evbuffer. 477 @param cleanupfn_arg optional argument to the cleanup callback 478 @return 0 if successful, or -1 if an error occurred 479 */ 480 int evbuffer_add_reference(struct evbuffer *outbuf, 481 const void *data, size_t datlen, 482 evbuffer_ref_cleanup_cb cleanupfn, void *cleanupfn_arg); 483 484 /** 485 Copy data from a file into the evbuffer for writing to a socket. 486 487 This function avoids unnecessary data copies between userland and 488 kernel. If sendfile is available and the EVBUFFER_FLAG_DRAINS_TO_FD 489 flag is set, it uses those functions. Otherwise, it tries to use 490 mmap (or CreateFileMapping on Windows). 491 492 The function owns the resulting file descriptor and will close it 493 when finished transferring data. 494 495 The results of using evbuffer_remove() or evbuffer_pullup() on 496 evbuffers whose data was added using this function are undefined. 497 498 For more fine-grained control, use evbuffer_add_file_segment. 499 500 @param outbuf the output buffer 501 @param fd the file descriptor 502 @param offset the offset from which to read data 503 @param length how much data to read, or -1 to read as much as possible. 504 (-1 requires that 'fd' support fstat.) 505 @return 0 if successful, or -1 if an error occurred 506 */ 507 508 int evbuffer_add_file(struct evbuffer *outbuf, int fd, ev_off_t offset, 509 ev_off_t length); 510 511 /** 512 An evbuffer_file_segment holds a reference to a range of a file -- 513 possibly the whole file! -- for use in writing from an evbuffer to a 514 socket. It could be implemented with mmap, sendfile, splice, or (if all 515 else fails) by just pulling all the data into RAM. A single 516 evbuffer_file_segment can be added more than once, and to more than one 517 evbuffer. 518 */ 519 struct evbuffer_file_segment; 520 521 /** 522 Flag for creating evbuffer_file_segment: If this flag is set, then when 523 the evbuffer_file_segment is freed and no longer in use by any 524 evbuffer, the underlying fd is closed. 525 */ 526 #define EVBUF_FS_CLOSE_ON_FREE 0x01 527 /** 528 Flag for creating evbuffer_file_segment: Disable memory-map based 529 implementations. 530 */ 531 #define EVBUF_FS_DISABLE_MMAP 0x02 532 /** 533 Flag for creating evbuffer_file_segment: Disable direct fd-to-fd 534 implementations (including sendfile and splice). 535 536 You might want to use this option if data needs to be taken from the 537 evbuffer by any means other than writing it to the network: the sendfile 538 backend is fast, but it only works for sending files directly to the 539 network. 540 */ 541 #define EVBUF_FS_DISABLE_SENDFILE 0x04 542 /** 543 Flag for creating evbuffer_file_segment: Do not allocate a lock for this 544 segment. If this option is set, then neither the segment nor any 545 evbuffer it is added to may ever be accessed from more than one thread 546 at a time. 547 */ 548 #define EVBUF_FS_DISABLE_LOCKING 0x08 549 550 /** 551 A cleanup function for a evbuffer_file_segment added to an evbuffer 552 for reference. 553 */ 554 typedef void (*evbuffer_file_segment_cleanup_cb)( 555 struct evbuffer_file_segment const* seg, int flags, void* arg); 556 557 /** 558 Create and return a new evbuffer_file_segment for reading data from a 559 file and sending it out via an evbuffer. 560 561 This function avoids unnecessary data copies between userland and 562 kernel. Where available, it uses sendfile or splice. 563 564 The file descriptor must not be closed so long as any evbuffer is using 565 this segment. 566 567 The results of using evbuffer_remove() or evbuffer_pullup() or any other 568 function that reads bytes from an evbuffer on any evbuffer containing 569 the newly returned segment are undefined, unless you pass the 570 EVBUF_FS_DISABLE_SENDFILE flag to this function. 571 572 @param fd an open file to read from. 573 @param offset an index within the file at which to start reading 574 @param length how much data to read, or -1 to read as much as possible. 575 (-1 requires that 'fd' support fstat.) 576 @param flags any number of the EVBUF_FS_* flags 577 @return a new evbuffer_file_segment, or NULL on failure. 578 **/ 579 struct evbuffer_file_segment *evbuffer_file_segment_new( 580 int fd, ev_off_t offset, ev_off_t length, unsigned flags); 581 582 /** 583 Free an evbuffer_file_segment 584 585 It is safe to call this function even if the segment has been added to 586 one or more evbuffers. The evbuffer_file_segment will not be freed 587 until no more references to it exist. 588 */ 589 void evbuffer_file_segment_free(struct evbuffer_file_segment *seg); 590 591 /** 592 Add cleanup callback and argument for the callback to an 593 evbuffer_file_segment. 594 595 The cleanup callback will be invoked when no more references to the 596 evbuffer_file_segment exist. 597 **/ 598 void evbuffer_file_segment_add_cleanup_cb(struct evbuffer_file_segment *seg, 599 evbuffer_file_segment_cleanup_cb cb, void* arg); 600 601 /** 602 Insert some or all of an evbuffer_file_segment at the end of an evbuffer 603 604 Note that the offset and length parameters of this function have a 605 different meaning from those provided to evbuffer_file_segment_new: When 606 you create the segment, the offset is the offset _within the file_, and 607 the length is the length _of the segment_, whereas when you add a 608 segment to an evbuffer, the offset is _within the segment_ and the 609 length is the length of the _part of the segment you want to use. 610 611 In other words, if you have a 10 KiB file, and you create an 612 evbuffer_file_segment for it with offset 20 and length 1000, it will 613 refer to bytes 20..1019 inclusive. If you then pass this segment to 614 evbuffer_add_file_segment and specify an offset of 20 and a length of 615 50, you will be adding bytes 40..99 inclusive. 616 617 @param buf the evbuffer to append to 618 @param seg the segment to add 619 @param offset the offset within the segment to start from 620 @param length the amount of data to add, or -1 to add it all. 621 @return 0 on success, -1 on failure. 622 */ 623 int evbuffer_add_file_segment(struct evbuffer *buf, 624 struct evbuffer_file_segment *seg, ev_off_t offset, ev_off_t length); 625 626 /** 627 Append a formatted string to the end of an evbuffer. 628 629 The string is formated as printf. 630 631 @param buf the evbuffer that will be appended to 632 @param fmt a format string 633 @param ... arguments that will be passed to printf(3) 634 @return The number of bytes added if successful, or -1 if an error occurred. 635 636 @see evutil_printf(), evbuffer_add_vprintf() 637 */ 638 int evbuffer_add_printf(struct evbuffer *buf, const char *fmt, ...) 639 #ifdef __GNUC__ 640 __attribute__((format(printf, 2, 3))) 641 #endif 642 ; 643 644 /** 645 Append a va_list formatted string to the end of an evbuffer. 646 647 @param buf the evbuffer that will be appended to 648 @param fmt a format string 649 @param ap a varargs va_list argument array that will be passed to vprintf(3) 650 @return The number of bytes added if successful, or -1 if an error occurred. 651 */ 652 int evbuffer_add_vprintf(struct evbuffer *buf, const char *fmt, va_list ap) 653 #ifdef __GNUC__ 654 __attribute__((format(printf, 2, 0))) 655 #endif 656 ; 657 658 659 /** 660 Remove a specified number of bytes data from the beginning of an evbuffer. 661 662 @param buf the evbuffer to be drained 663 @param len the number of bytes to drain from the beginning of the buffer 664 @return 0 on success, -1 on failure. 665 */ 666 int evbuffer_drain(struct evbuffer *buf, size_t len); 667 668 669 /** 670 Write the contents of an evbuffer to a file descriptor. 671 672 The evbuffer will be drained after the bytes have been successfully written. 673 674 @param buffer the evbuffer to be written and drained 675 @param fd the file descriptor to be written to 676 @return the number of bytes written, or -1 if an error occurred 677 @see evbuffer_read() 678 */ 679 int evbuffer_write(struct evbuffer *buffer, evutil_socket_t fd); 680 681 /** 682 Write some of the contents of an evbuffer to a file descriptor. 683 684 The evbuffer will be drained after the bytes have been successfully written. 685 686 @param buffer the evbuffer to be written and drained 687 @param fd the file descriptor to be written to 688 @param howmuch the largest allowable number of bytes to write, or -1 689 to write as many bytes as we can. 690 @return the number of bytes written, or -1 if an error occurred 691 @see evbuffer_read() 692 */ 693 int evbuffer_write_atmost(struct evbuffer *buffer, evutil_socket_t fd, 694 ev_ssize_t howmuch); 695 696 /** 697 Read from a file descriptor and store the result in an evbuffer. 698 699 @param buffer the evbuffer to store the result 700 @param fd the file descriptor to read from 701 @param howmuch the number of bytes to be read 702 @return the number of bytes read, or -1 if an error occurred 703 @see evbuffer_write() 704 */ 705 int evbuffer_read(struct evbuffer *buffer, evutil_socket_t fd, int howmuch); 706 707 /** 708 Search for a string within an evbuffer. 709 710 @param buffer the evbuffer to be searched 711 @param what the string to be searched for 712 @param len the length of the search string 713 @param start NULL or a pointer to a valid struct evbuffer_ptr. 714 @return a struct evbuffer_ptr whose 'pos' field has the offset of the 715 first occurrence of the string in the buffer after 'start'. The 'pos' 716 field of the result is -1 if the string was not found. 717 */ 718 struct evbuffer_ptr evbuffer_search(struct evbuffer *buffer, const char *what, size_t len, const struct evbuffer_ptr *start); 719 720 /** 721 Search for a string within part of an evbuffer. 722 723 @param buffer the evbuffer to be searched 724 @param what the string to be searched for 725 @param len the length of the search string 726 @param start NULL or a pointer to a valid struct evbuffer_ptr that 727 indicates where we should start searching. 728 @param end NULL or a pointer to a valid struct evbuffer_ptr that 729 indicates where we should stop searching. 730 @return a struct evbuffer_ptr whose 'pos' field has the offset of the 731 first occurrence of the string in the buffer after 'start'. The 'pos' 732 field of the result is -1 if the string was not found. 733 */ 734 struct evbuffer_ptr evbuffer_search_range(struct evbuffer *buffer, const char *what, size_t len, const struct evbuffer_ptr *start, const struct evbuffer_ptr *end); 735 736 /** 737 Defines how to adjust an evbuffer_ptr by evbuffer_ptr_set() 738 739 @see evbuffer_ptr_set() */ 740 enum evbuffer_ptr_how { 741 /** Sets the pointer to the position; can be called on with an 742 uninitialized evbuffer_ptr. */ 743 EVBUFFER_PTR_SET, 744 /** Advances the pointer by adding to the current position. */ 745 EVBUFFER_PTR_ADD 746 }; 747 748 /** 749 Sets the search pointer in the buffer to position. 750 751 There are two ways to use this function: you can call 752 evbuffer_ptr_set(buf, &pos, N, EVBUFFER_PTR_SET) 753 to move 'pos' to a position 'N' bytes after the start of the buffer, or 754 evbuffer_ptr_set(buf, &pos, N, EVBUFFER_PTR_SET) 755 to move 'pos' forward by 'N' bytes. 756 757 If evbuffer_ptr is not initialized, this function can only be called 758 with EVBUFFER_PTR_SET. 759 760 An evbuffer_ptr can represent any position from the start of the buffer to 761 a position immediately after the end of the buffer. 762 763 @param buffer the evbuffer to be search 764 @param ptr a pointer to a struct evbuffer_ptr 765 @param position the position at which to start the next search 766 @param how determines how the pointer should be manipulated. 767 @returns 0 on success or -1 otherwise 768 */ 769 int 770 evbuffer_ptr_set(struct evbuffer *buffer, struct evbuffer_ptr *ptr, 771 size_t position, enum evbuffer_ptr_how how); 772 773 /** 774 Search for an end-of-line string within an evbuffer. 775 776 @param buffer the evbuffer to be searched 777 @param start NULL or a pointer to a valid struct evbuffer_ptr to start 778 searching at. 779 @param eol_len_out If non-NULL, the pointed-to value will be set to 780 the length of the end-of-line string. 781 @param eol_style The kind of EOL to look for; see evbuffer_readln() for 782 more information 783 @return a struct evbuffer_ptr whose 'pos' field has the offset of the 784 first occurrence EOL in the buffer after 'start'. The 'pos' 785 field of the result is -1 if the string was not found. 786 */ 787 struct evbuffer_ptr evbuffer_search_eol(struct evbuffer *buffer, 788 struct evbuffer_ptr *start, size_t *eol_len_out, 789 enum evbuffer_eol_style eol_style); 790 791 /** Function to peek at data inside an evbuffer without removing it or 792 copying it out. 793 794 Pointers to the data are returned by filling the 'vec_out' array 795 with pointers to one or more extents of data inside the buffer. 796 797 The total data in the extents that you get back may be more than 798 you requested (if there is more data last extent than you asked 799 for), or less (if you do not provide enough evbuffer_iovecs, or if 800 the buffer does not have as much data as you asked to see). 801 802 @param buffer the evbuffer to peek into, 803 @param len the number of bytes to try to peek. If len is negative, we 804 will try to fill as much of vec_out as we can. If len is negative 805 and vec_out is not provided, we return the number of evbuffer_iovecs 806 that would be needed to get all the data in the buffer. 807 @param start_at an evbuffer_ptr indicating the point at which we 808 should start looking for data. NULL means, "At the start of the 809 buffer." 810 @param vec_out an array of evbuffer_iovec 811 @param n_vec the length of vec_out. If 0, we only count how many 812 extents would be necessary to point to the requested amount of 813 data. 814 @return The number of extents needed. This may be less than n_vec 815 if we didn't need all the evbuffer_iovecs we were given, or more 816 than n_vec if we would need more to return all the data that was 817 requested. 818 */ 819 int evbuffer_peek(struct evbuffer *buffer, ev_ssize_t len, 820 struct evbuffer_ptr *start_at, 821 struct evbuffer_iovec *vec_out, int n_vec); 822 823 824 /** Structure passed to an evbuffer_cb_func evbuffer callback 825 826 @see evbuffer_cb_func, evbuffer_add_cb() 827 */ 828 struct evbuffer_cb_info { 829 /** The number of bytes in this evbuffer when callbacks were last 830 * invoked. */ 831 size_t orig_size; 832 /** The number of bytes added since callbacks were last invoked. */ 833 size_t n_added; 834 /** The number of bytes removed since callbacks were last invoked. */ 835 size_t n_deleted; 836 }; 837 838 /** Type definition for a callback that is invoked whenever data is added or 839 removed from an evbuffer. 840 841 An evbuffer may have one or more callbacks set at a time. The order 842 in which they are executed is undefined. 843 844 A callback function may add more callbacks, or remove itself from the 845 list of callbacks, or add or remove data from the buffer. It may not 846 remove another callback from the list. 847 848 If a callback adds or removes data from the buffer or from another 849 buffer, this can cause a recursive invocation of your callback or 850 other callbacks. If you ask for an infinite loop, you might just get 851 one: watch out! 852 853 @param buffer the buffer whose size has changed 854 @param info a structure describing how the buffer changed. 855 @param arg a pointer to user data 856 */ 857 typedef void (*evbuffer_cb_func)(struct evbuffer *buffer, const struct evbuffer_cb_info *info, void *arg); 858 859 struct evbuffer_cb_entry; 860 /** Add a new callback to an evbuffer. 861 862 Subsequent calls to evbuffer_add_cb() add new callbacks. To remove this 863 callback, call evbuffer_remove_cb or evbuffer_remove_cb_entry. 864 865 @param buffer the evbuffer to be monitored 866 @param cb the callback function to invoke when the evbuffer is modified, 867 or NULL to remove all callbacks. 868 @param cbarg an argument to be provided to the callback function 869 @return a handle to the callback on success, or NULL on failure. 870 */ 871 struct evbuffer_cb_entry *evbuffer_add_cb(struct evbuffer *buffer, evbuffer_cb_func cb, void *cbarg); 872 873 /** Remove a callback from an evbuffer, given a handle returned from 874 evbuffer_add_cb. 875 876 Calling this function invalidates the handle. 877 878 @return 0 if a callback was removed, or -1 if no matching callback was 879 found. 880 */ 881 int evbuffer_remove_cb_entry(struct evbuffer *buffer, 882 struct evbuffer_cb_entry *ent); 883 884 /** Remove a callback from an evbuffer, given the function and argument 885 used to add it. 886 887 @return 0 if a callback was removed, or -1 if no matching callback was 888 found. 889 */ 890 int evbuffer_remove_cb(struct evbuffer *buffer, evbuffer_cb_func cb, void *cbarg); 891 892 /** If this flag is not set, then a callback is temporarily disabled, and 893 * should not be invoked. 894 * 895 * @see evbuffer_cb_set_flags(), evbuffer_cb_clear_flags() 896 */ 897 #define EVBUFFER_CB_ENABLED 1 898 899 /** Change the flags that are set for a callback on a buffer by adding more. 900 901 @param buffer the evbuffer that the callback is watching. 902 @param cb the callback whose status we want to change. 903 @param flags EVBUFFER_CB_ENABLED to re-enable the callback. 904 @return 0 on success, -1 on failure. 905 */ 906 int evbuffer_cb_set_flags(struct evbuffer *buffer, 907 struct evbuffer_cb_entry *cb, ev_uint32_t flags); 908 909 /** Change the flags that are set for a callback on a buffer by removing some 910 911 @param buffer the evbuffer that the callback is watching. 912 @param cb the callback whose status we want to change. 913 @param flags EVBUFFER_CB_ENABLED to disable the callback. 914 @return 0 on success, -1 on failure. 915 */ 916 int evbuffer_cb_clear_flags(struct evbuffer *buffer, 917 struct evbuffer_cb_entry *cb, ev_uint32_t flags); 918 919 #if 0 920 /** Postpone calling a given callback until unsuspend is called later. 921 922 This is different from disabling the callback, since the callback will get 923 invoked later if the buffer size changes between now and when we unsuspend 924 it. 925 926 @param the buffer that the callback is watching. 927 @param cb the callback we want to suspend. 928 */ 929 void evbuffer_cb_suspend(struct evbuffer *buffer, struct evbuffer_cb_entry *cb); 930 /** Stop postponing a callback that we postponed with evbuffer_cb_suspend. 931 932 If data was added to or removed from the buffer while the callback was 933 suspended, the callback will get called once now. 934 935 @param the buffer that the callback is watching. 936 @param cb the callback we want to stop suspending. 937 */ 938 void evbuffer_cb_unsuspend(struct evbuffer *buffer, struct evbuffer_cb_entry *cb); 939 #endif 940 941 /** 942 Makes the data at the begging of an evbuffer contiguous. 943 944 @param buf the evbuffer to make contiguous 945 @param size the number of bytes to make contiguous, or -1 to make the 946 entire buffer contiguous. 947 @return a pointer to the contiguous memory array 948 */ 949 950 unsigned char *evbuffer_pullup(struct evbuffer *buf, ev_ssize_t size); 951 952 /** 953 Prepends data to the beginning of the evbuffer 954 955 @param buf the evbuffer to which to prepend data 956 @param data a pointer to the memory to prepend 957 @param size the number of bytes to prepend 958 @return 0 if successful, or -1 otherwise 959 */ 960 961 int evbuffer_prepend(struct evbuffer *buf, const void *data, size_t size); 962 963 /** 964 Prepends all data from the src evbuffer to the beginning of the dst 965 evbuffer. 966 967 @param dst the evbuffer to which to prepend data 968 @param src the evbuffer to prepend; it will be emptied as a result 969 @return 0 if successful, or -1 otherwise 970 */ 971 int evbuffer_prepend_buffer(struct evbuffer *dst, struct evbuffer* src); 972 973 /** 974 Prevent calls that modify an evbuffer from succeeding. A buffer may 975 frozen at the front, at the back, or at both the front and the back. 976 977 If the front of a buffer is frozen, operations that drain data from 978 the front of the buffer, or that prepend data to the buffer, will 979 fail until it is unfrozen. If the back a buffer is frozen, operations 980 that append data from the buffer will fail until it is unfrozen. 981 982 @param buf The buffer to freeze 983 @param at_front If true, we freeze the front of the buffer. If false, 984 we freeze the back. 985 @return 0 on success, -1 on failure. 986 */ 987 int evbuffer_freeze(struct evbuffer *buf, int at_front); 988 /** 989 Re-enable calls that modify an evbuffer. 990 991 @param buf The buffer to un-freeze 992 @param at_front If true, we unfreeze the front of the buffer. If false, 993 we unfreeze the back. 994 @return 0 on success, -1 on failure. 995 */ 996 int evbuffer_unfreeze(struct evbuffer *buf, int at_front); 997 998 struct event_base; 999 /** 1000 Force all the callbacks on an evbuffer to be run, not immediately after 1001 the evbuffer is altered, but instead from inside the event loop. 1002 1003 This can be used to serialize all the callbacks to a single thread 1004 of execution. 1005 */ 1006 int evbuffer_defer_callbacks(struct evbuffer *buffer, struct event_base *base); 1007 1008 /** 1009 Append data from 1 or more iovec's to an evbuffer 1010 1011 Calculates the number of bytes needed for an iovec structure and guarantees 1012 all data will fit into a single chain. Can be used in lieu of functionality 1013 which calls evbuffer_add() constantly before being used to increase 1014 performance. 1015 1016 @param buffer the destination buffer 1017 @param vec the source iovec 1018 @param n_vec the number of iovec structures. 1019 @return the number of bytes successfully written to the output buffer. 1020 */ 1021 size_t evbuffer_add_iovec(struct evbuffer * buffer, struct evbuffer_iovec * vec, int n_vec); 1022 1023 #ifdef __cplusplus 1024 } 1025 #endif 1026 1027 #endif /* EVENT2_BUFFER_H_INCLUDED_ */ 1028