lib/tsocket/tsocket.h

   1 /*
   2    Unix SMB/CIFS implementation.
   3
   4    Copyright (C) Stefan Metzmacher 2009
   5
   6      ** NOTE! The following LGPL license applies to the tsocket
   7      ** library. This does NOT imply that all of Samba is released
   8      ** under the LGPL
   9
  10    This library is free software; you can redistribute it and/or
  11    modify it under the terms of the GNU Lesser General Public
  12    License as published by the Free Software Foundation; either
  13    version 3 of the License, or (at your option) any later version.
  14
  15    This library is distributed in the hope that it will be useful,
  16    but WITHOUT ANY WARRANTY; without even the implied warranty of
  17    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  18    Lesser General Public License for more details.
  19
  20    You should have received a copy of the GNU Lesser General Public
  21    License along with this library; if not, see <http://www.gnu.org/licenses/>.
  22 */
  23
  24 #ifndef _TSOCKET_H
  25 #define _TSOCKET_H
  26
  27 #include <talloc.h>
  28 #include <tevent.h>
  29
  30 struct tsocket_address;
  31 struct tdgram_context;
  32 struct tstream_context;
  33 struct iovec;
  34
  35 /**
  36  * @mainpage
  37  *
  38  * The tsocket abstraction is an API ...
  39  */
  40
  41 /**
  42  * @defgroup tsocket The tsocket API
  43  *
  44  * The tsocket abstraction is split into two different kinds of
  45  * communication interfaces.
  46  *
  47  * There's the "tstream_context" interface with abstracts the communication
  48  * through a bidirectional byte stream between two endpoints.
  49  *
  50  * And there's the "tdgram_context" interface with abstracts datagram based
  51  * communication between any number of endpoints.
  52  *
  53  * Both interfaces share the "tsocket_address" abstraction for endpoint
  54  * addresses.
  55  *
  56  * The whole library is based on the talloc(3) and 'tevent' libraries and
  57  * provides "tevent_req" based "foo_send()"/"foo_recv()" functions pairs for
  58  * all abstracted methods that need to be async.
  59  *
  60  * @section vsock Virtual Sockets
  61  *
  62  * The abstracted layout of tdgram_context and tstream_context allow
  63  * implementations around virtual sockets for encrypted tunnels (like TLS,
  64  * SASL or GSSAPI) or named pipes over smb.
  65  *
  66  * @section npa Named Pipe Auth (NPA) Sockets
  67  *
  68  * Samba has an implementation to abstract named pipes over smb (within the
  69  * server side). See libcli/named_pipe_auth/npa_tstream.[ch] for the core code.
  70  * The current callers are located in source4/ntvfs/ipc/vfs_ipc.c and
  71  * source4/rpc_server/service_rpc.c for the users.
  72  */
  73
  74 /**
  75  * @defgroup tsocket_address The tsocket_address abstraction
  76  * @ingroup tsocket
  77  *
  78  * The tsocket_address represents an socket endpoint genericly.
  79  * As it's like an abstract class it has no specific constructor.
  80  * The specific constructors are descripted in later sections.
  81  *
  82  * @{
  83  */
  84
  85 /**
  86  * @brief Get a string representaion of the endpoint.
  87  *
  88  * This function creates a string representation of the endpoint for debugging.
  89  * The output will look as followed:
  90  *      prefix:address:port
  91  *
  92  * e.g.
  93  *      ipv4:192.168.1.1:143
  94  *
  95  * Callers should not try to parse the string! The should use additional methods
  96  * of the specific tsocket_address implemention to get more details.
  97  *
  98  * @param[in]  addr     The address to convert.
  99  *
 100  * @param[in]  mem_ctx  The talloc memory context to allocate the memory.
 101  *
 102  * @return              The address as a string representation, NULL on error.
 103  *
 104  * @see tsocket_address_is_inet()
 105  * @see tsocket_address_inet_addr_string()
 106  * @see tsocket_address_inet_port()
 107  */
 108 char *tsocket_address_string(const struct tsocket_address *addr,
 109                              TALLOC_CTX *mem_ctx);
 110
 111 #ifdef DOXYGEN
 112 /**
 113  * @brief This creates a copy of a tsocket_address.
 114  *
 115  * This is useful when before doing modifications to a socket via additional
 116  * methods of the specific tsocket_address implementation.
 117  *
 118  * @param[in]  addr     The address to create the copy from.
 119  *
 120  * @param[in]  mem_ctx  The talloc memory context to use.
 121  *
 122  * @return              A newly allocated copy of addr (tsocket_address *), NULL
 123  *                      on error.
 124  */
 125 struct tsocket_address *tsocket_address_copy(const struct tsocket_address *addr,
 126                 TALLOC_CTX *mem_ctx);
 127 #else
 128 struct tsocket_address *_tsocket_address_copy(const struct tsocket_address *addr,
 129                                               TALLOC_CTX *mem_ctx,
 130                                               const char *location);
 131
 132 #define tsocket_address_copy(addr, mem_ctx) \
 133         _tsocket_address_copy(addr, mem_ctx, __location__)
 134 #endif
 135
 136 /**
 137  * @}
 138  */
 139
 140 /**
 141  * @defgroup tdgram_context The tdgram_context abstraction
 142  * @ingroup tsocket
 143  *
 144  * The tdgram_context is like an abstract class for datagram based sockets. The
 145  * interface provides async 'tevent_req' based functions on top functionality
 146  * is similar to the recvfrom(2)/sendto(2)/close(2) syscalls.
 147  *
 148  * @note You can always use talloc_free(tdgram) to cleanup the resources
 149  * of the tdgram_context on a fatal error.
 150  * @{
 151  */
 152
 153 /**
 154  * @brief Ask for next available datagram on the abstracted tdgram_context.
 155  *
 156  * It returns a 'tevent_req' handle, where the caller can register
 157  * a callback with tevent_req_set_callback(). The callback is triggered
 158  * when a datagram is available or an error happened.
 159  *
 160  * @param[in]  mem_ctx  The talloc memory context to use.
 161  *
 162  * @param[in]  ev       The tevent_context to run on.
 163  *
 164  * @param[in]  dgram    The dgram context to work on.
 165  *
 166  * @return              Returns a 'tevent_req' handle, where the caller can
 167  *                      register a callback with tevent_req_set_callback().
 168  *                      NULL on fatal error.
 169  *
 170  * @see tdgram_inet_udp_socket()
 171  * @see tdgram_unix_socket()
 172  */
 173 struct tevent_req *tdgram_recvfrom_send(TALLOC_CTX *mem_ctx,
 174                                         struct tevent_context *ev,
 175                                         struct tdgram_context *dgram);
 176
 177 /**
 178  * @brief Receive the next available datagram on the abstracted tdgram_context.
 179  *
 180  * This function should be called by the callback when a datagram is available
 181  * or an error happened.
 182  *
 183  * The caller can only have one outstanding tdgram_recvfrom_send() at a time
 184  * otherwise the caller will get '*perrno = EBUSY'.
 185  *
 186  * @param[in]  req      The tevent request from tdgram_recvfrom_send().
 187  *
 188  * @param[out] perrno   The error number, set if an error occurred.
 189  *
 190  * @param[in]  mem_ctx  The memory context to use.
 191  *
 192  * @param[out] buf      This will hold the buffer of the datagram.
 193  *
 194  * @param[out] src      The abstracted tsocket_address of the sender of the
 195  *                      received datagram.
 196  *
 197  * @return              The length of the datagram (0 is never returned!),
 198  *                      -1 on error with perrno set to the actual errno.
 199  *
 200  * @see tdgram_recvfrom_send()
 201  */
 202 ssize_t tdgram_recvfrom_recv(struct tevent_req *req,
 203                              int *perrno,
 204                              TALLOC_CTX *mem_ctx,
 205                              uint8_t **buf,
 206                              struct tsocket_address **src);
 207
 208 /**
 209  * @brief Send a datagram to a destination endpoint.
 210  *
 211  * The function can be called to send a datagram (specified by a buf/len) to a
 212  * destination endpoint (specified by dst). It's not allowed for len to be 0.
 213  *
 214  * It returns a 'tevent_req' handle, where the caller can register a callback
 215  * with tevent_req_set_callback(). The callback is triggered when the specific
 216  * implementation (assumes it) has delivered the datagram to the "wire".
 217  *
 218  * The callback is then supposed to get the result by calling
 219  * tdgram_sendto_recv() on the 'tevent_req'.
 220  *
 221  * @param[in]  mem_ctx  The talloc memory context to use.
 222  *
 223  * @param[in]  ev       The tevent_context to run on.
 224  *
 225  * @param[in]  dgram    The dgram context to work on.
 226  *
 227  * @param[in]  buf      The buffer to send.
 228  *
 229  * @param[in]  len      The length of the buffer to send. It has to be bigger
 230  *                      than 0.
 231  *
 232  * @param[in]  dst      The destination to send the datagram to in form of a
 233  *                      tsocket_address.
 234  *
 235  * @return              Returns a 'tevent_req' handle, where the caller can
 236  *                      register a callback with tevent_req_set_callback().
 237  *                      NULL on fatal error.
 238  *
 239  * @see tdgram_inet_udp_socket()
 240  * @see tdgram_unix_socket()
 241  * @see tdgram_sendto_recv()
 242  */
 243 struct tevent_req *tdgram_sendto_send(TALLOC_CTX *mem_ctx,
 244                                       struct tevent_context *ev,
 245                                       struct tdgram_context *dgram,
 246                                       const uint8_t *buf, size_t len,
 247                                       const struct tsocket_address *dst);
 248
 249 /**
 250  * @brief Receive the result of the sent datagram.
 251  *
 252  * The caller can only have one outstanding tdgram_sendto_send() at a time
 253  * otherwise the caller will get '*perrno = EBUSY'.
 254  *
 255  * @param[in]  req      The tevent request from tdgram_sendto_send().
 256  *
 257  * @param[out] perrno   The error number, set if an error occurred.
 258  *
 259  * @return              The length of the datagram (0 is never returned!), -1 on
 260  *                      error with perrno set to the actual errno.
 261  *
 262  * @see tdgram_sendto_send()
 263  */
 264 ssize_t tdgram_sendto_recv(struct tevent_req *req,
 265                            int *perrno);
 266
 267 /**
 268  * @brief Shutdown/close an abstracted socket.
 269  *
 270  * It returns a 'tevent_req' handle, where the caller can register a callback
 271  * with tevent_req_set_callback(). The callback is triggered when the specific
 272  * implementation (assumes it) has delivered the datagram to the "wire".
 273  *
 274  * The callback is then supposed to get the result by calling
 275  * tdgram_sendto_recv() on the 'tevent_req'.
 276  *
 277  * @param[in]  mem_ctx  The talloc memory context to use.
 278  *
 279  * @param[in]  ev       The tevent_context to run on.
 280  *
 281  * @param[in]  dgram    The dgram context diconnect from.
 282  *
 283  * @return              Returns a 'tevent_req' handle, where the caller can
 284  *                      register a callback with tevent_req_set_callback().
 285  *                      NULL on fatal error.
 286  *
 287  * @see tdgram_disconnect_recv()
 288  */
 289 struct tevent_req *tdgram_disconnect_send(TALLOC_CTX *mem_ctx,
 290                                           struct tevent_context *ev,
 291                                           struct tdgram_context *dgram);
 292
 293 /**
 294  * @brief Receive the result from a tdgram_disconnect_send() request.
 295  *
 296  * The caller should make sure there're no outstanding tdgram_recvfrom_send()
 297  * and tdgram_sendto_send() calls otherwise the caller will get
 298  * '*perrno = EBUSY'.
 299  *
 300  * @param[in]  req      The tevent request from tdgram_disconnect_send().
 301  *
 302  * @param[out] perrno   The error number, set if an error occurred.
 303  *
 304  * @return              The length of the datagram (0 is never returned!), -1 on
 305  *                      error with perrno set to the actual errno.
 306  *
 307  * @see tdgram_disconnect_send()
 308  */
 309 int tdgram_disconnect_recv(struct tevent_req *req,
 310                            int *perrno);
 311
 312 /**
 313  * @}
 314  */
 315
 316 /**
 317  * @defgroup tstream_context The tstream_context abstraction
 318  * @ingroup tsocket
 319  *
 320  * The tstream_context is like an abstract class for stream based sockets. The
 321  * interface provides async 'tevent_req' based functions on top functionality
 322  * is similar to the readv(2)/writev(2)/close(2) syscalls.
 323  *
 324  * @note You can always use talloc_free(tstream) to cleanup the resources
 325  * of the tstream_context on a fatal error.
 326  *
 327  * @{
 328  */
 329
 330 /**
 331  * @brief Report the number of bytes received but not consumed yet.
 332  *
 333  * The tstream_pending_bytes() function reports how much bytes of the incoming
 334  * stream have been received but not consumed yet.
 335  *
 336  * @param[in]  stream   The tstream_context to check for pending bytes.
 337  *
 338  * @return              The number of bytes received, -1 on error with errno
 339  *                      set.
 340  */
 341 ssize_t tstream_pending_bytes(struct tstream_context *stream);
 342
 343 /**
 344  * @brief Read a specific amount of bytes from a stream socket.
 345  *
 346  * The function can be called to read for a specific amount of bytes from the
 347  * stream into given buffers. The caller has to preallocate the buffers.
 348  *
 349  * The caller might need to use tstream_pending_bytes() if the protocol doesn't
 350  * have a fixed pdu header containing the pdu size.
 351  *
 352  * @param[in]  mem_ctx  The talloc memory context to use.
 353  *
 354  * @param[in]  ev       The tevent_context to run on.
 355  *
 356  * @param[in]  stream   The tstream context to work on.
 357  *
 358  * @param[out] vector   A preallocated iovec to store the data to read.
 359  *
 360  * @param[in]  count    The number of buffers in the vector allocated.
 361  *
 362  * @return              A 'tevent_req' handle, where the caller can register
 363  *                      a callback with tevent_req_set_callback(). NULL on
 364  *                      fatal error.
 365  *
 366  * @see tstream_unix_connect_send()
 367  * @see tstream_inet_tcp_connect_send()
 368  */
 369 struct tevent_req *tstream_readv_send(TALLOC_CTX *mem_ctx,
 370                                       struct tevent_context *ev,
 371                                       struct tstream_context *stream,
 372                                       struct iovec *vector,
 373                                       size_t count);
 374
 375 /**
 376  * @brief Get the result of a tstream_readv_send().
 377  *
 378  * The caller can only have one outstanding tstream_readv_send()
 379  * at a time otherwise the caller will get *perrno = EBUSY.
 380  *
 381  * @param[in]  req      The tevent request from tstream_readv_send().
 382  *
 383  * @param[out] perrno   The error number, set if an error occurred.
 384  *
 385  * @return              The length of the stream (0 is never returned!), -1 on
 386  *                      error with perrno set to the actual errno.
 387  */
 388 int tstream_readv_recv(struct tevent_req *req,
 389                        int *perrno);
 390
 391 /**
 392  * @brief Write buffers from a vector into a stream socket.
 393  *
 394  * The function can be called to write buffers from a given vector
 395  * to a stream socket.
 396  *
 397  * You have to ensure that the vector is not empty.
 398  *
 399  * @param[in]  mem_ctx  The talloc memory context to use.
 400  *
 401  * @param[in]  ev       The tevent_context to run on.
 402  *
 403  * @param[in]  stream   The tstream context to work on.
 404  *
 405  * @param[in]  vector   The iovec vector with data to write on a stream socket.
 406  *
 407  * @param[in]  count    The number of buffers in the vector to write.
 408  *
 409  * @return              A 'tevent_req' handle, where the caller can register
 410  *                      a callback with tevent_req_set_callback(). NULL on
 411  *                      fatal error.
 412  */
 413 struct tevent_req *tstream_writev_send(TALLOC_CTX *mem_ctx,
 414                                        struct tevent_context *ev,
 415                                        struct tstream_context *stream,
 416                                        const struct iovec *vector,
 417                                        size_t count);
 418
 419 /**
 420  * @brief Get the result of a tstream_writev_send().
 421  *
 422  * The caller can only have one outstanding tstream_writev_send()
 423  * at a time otherwise the caller will get *perrno = EBUSY.
 424  *
 425  * @param[in]  req      The tevent request from tstream_writev_send().
 426  *
 427  * @param[out] perrno   The error number, set if an error occurred.
 428  *
 429  * @return              The length of the stream (0 is never returned!), -1 on
 430  *                      error with perrno set to the actual errno.
 431  */
 432 int tstream_writev_recv(struct tevent_req *req,
 433                         int *perrno);
 434
 435 /**
 436  * @brief Shutdown/close an abstracted socket.
 437  *
 438  * It returns a 'tevent_req' handle, where the caller can register a callback
 439  * with tevent_req_set_callback(). The callback is triggered when the specific
 440  * implementation (assumes it) has delivered the stream to the "wire".
 441  *
 442  * The callback is then supposed to get the result by calling
 443  * tdgram_sendto_recv() on the 'tevent_req'.
 444  *
 445  * @param[in]  mem_ctx  The talloc memory context to use.
 446  *
 447  * @param[in]  ev       The tevent_context to run on.
 448  *
 449  * @param[in]  stream   The tstream context to work on.
 450  *
 451  * @return              A 'tevent_req' handle, where the caller can register
 452  *                      a callback with tevent_req_set_callback(). NULL on
 453  *                      fatal error.
 454  */
 455 struct tevent_req *tstream_disconnect_send(TALLOC_CTX *mem_ctx,
 456                                            struct tevent_context *ev,
 457                                            struct tstream_context *stream);
 458
 459 /**
 460  * @brief Get the result of a tstream_disconnect_send().
 461  *
 462  * The caller can only have one outstanding tstream_writev_send()
 463  * at a time otherwise the caller will get *perrno = EBUSY.
 464  *
 465  * @param[in]  req      The tevent request from tstream_disconnect_send().
 466  *
 467  * @param[out] perrno   The error number, set if an error occurred.
 468  *
 469  * @return              The length of the stream (0 is never returned!), -1 on
 470  *                      error with perrno set to the actual errno.
 471  */
 472 int tstream_disconnect_recv(struct tevent_req *req,
 473                             int *perrno);
 474
 475 /**
 476  * @}
 477  */
 478
 479
 480 /**
 481  * @defgroup tsocket_bsd  tsocket_bsd - inet, inet6 and unix
 482  * @ingroup tsocket
 483  *
 484  * The main tsocket library comes with implentations for BSD style ipv4, ipv6
 485  * and unix sockets.
 486  *
 487  * @{
 488  */
 489
 490 /**
 491  * @brief Find out if the tsocket_address represents an ipv4 or ipv6 endpoint.
 492  *
 493  * @param[in]  addr     The tsocket_address pointer
 494  *
 495  * @param[in]  fam      The family can be can be "ipv4", "ipv6" or "ip". With
 496  *                      "ip" is autodetects "ipv4" or "ipv6" based on the
 497  *                      addr.
 498  *
 499  * @return              true if addr represents an address of the given family,
 500  *                      otherwise false.
 501  */
 502 bool tsocket_address_is_inet(const struct tsocket_address *addr, const char *fam);
 503
 504 #if DOXYGEN
 505 /**
 506  * @brief Create a tsocket_address for ipv4 and ipv6 endpoint addresses.
 507  *
 508  * @param[in]  mem_ctx  The talloc memory context to use.
 509  *
 510  * @param[in]  fam      The family can be can be "ipv4", "ipv6" or "ip". With
 511  *                      "ip" is autodetects "ipv4" or "ipv6" based on the
 512  *                      addr.
 513  *
 514  * @param[in]  addr     A valid ip address string based on the selected family
 515  *                      (dns names are not allowed!). It's valid to pass NULL,
 516  *                      which gets mapped to "0.0.0.0" or "::".
 517  *
 518  * @param[in]  port     A valid port number.
 519  *
 520  * @param[out] _addr    A tsocket_address pointer to store the information.
 521  *
 522  * @return              0 on success, -1 on error with errno set.
 523  */
 524 int tsocket_address_inet_from_strings(TALLOC_CTX *mem_ctx,
 525                                       const char *fam,
 526                                       const char *addr,
 527                                       uint16_t port,
 528                                       struct tsocket_address **_addr);
 529 #else
 530 int _tsocket_address_inet_from_strings(TALLOC_CTX *mem_ctx,
 531                                        const char *fam,
 532                                        const char *addr,
 533                                        uint16_t port,
 534                                        struct tsocket_address **_addr,
 535                                        const char *location);
 536
 537 #define tsocket_address_inet_from_strings(mem_ctx, fam, addr, port, _addr) \
 538         _tsocket_address_inet_from_strings(mem_ctx, fam, addr, port, _addr, \
 539                                            __location__)
 540 #endif
 541
 542 /**
 543  * @brief Get the address of an 'inet' tsocket_address as a string.
 544  *
 545  * @param[in]  addr     The address to convert to a string.
 546  *
 547  * @param[in]  mem_ctx  The talloc memory context to use.
 548  *
 549  * @return              A newly allocated string of the address, NULL on error
 550  *                      with errno set.
 551  *
 552  * @see tsocket_address_is_inet()
 553  */
 554 char *tsocket_address_inet_addr_string(const struct tsocket_address *addr,
 555                                        TALLOC_CTX *mem_ctx);
 556
 557 /**
 558  * @brief Get the port number as an integer from an 'inet' tsocket_address.
 559  *
 560  * @param[in]  addr     The tsocket address to use.
 561  *
 562  * @return              The port number, 0 on error with errno set.
 563  */
 564 uint16_t tsocket_address_inet_port(const struct tsocket_address *addr);
 565
 566 /**
 567  * @brief Set the port number of an existing 'inet' tsocket_address.
 568  *
 569  * @param[in]  addr     The existing tsocket_address to use.
 570  *
 571  * @param[in]  port     The valid port number to set.
 572  *
 573  * @return              0 on success, -1 on error with errno set.
 574  */
 575 int tsocket_address_inet_set_port(struct tsocket_address *addr,
 576                                   uint16_t port);
 577
 578 /**
 579  * @brief Find out if the tsocket_address represents an unix domain endpoint.
 580  *
 581  * @param[in]  addr     The tsocket_address pointer
 582  *
 583  * @return              true if addr represents an unix domain endpoint,
 584  *                      otherwise false.
 585  */
 586 bool tsocket_address_is_unix(const struct tsocket_address *addr);
 587
 588 #ifdef DOXYGEN
 589 /**
 590  * @brief Create a tsocket_address for a unix domain endpoint addresses.
 591  *
 592  * @param[in]  mem_ctx  The talloc memory context to use.
 593  *
 594  * @param[in]  path     The filesystem path, NULL will map "".
 595  *
 596  * @param[in]  _addr    The tsocket_address pointer to store the information.
 597  *
 598  * @return              0 on success, -1 on error with errno set.
 599  *
 600  * @see tsocket_address_is_unix()
 601  */
 602 int tsocket_address_unix_from_path(TALLOC_CTX *mem_ctx,
 603                                    const char *path,
 604                                    struct tsocket_address **_addr);
 605 #else
 606 int _tsocket_address_unix_from_path(TALLOC_CTX *mem_ctx,
 607                                     const char *path,
 608                                     struct tsocket_address **_addr,
 609                                     const char *location);
 610
 611 #define tsocket_address_unix_from_path(mem_ctx, path, _addr) \
 612         _tsocket_address_unix_from_path(mem_ctx, path, _addr, \
 613                                         __location__)
 614 #endif
 615
 616 /**
 617  * @brief Get the address of an 'unix' tsocket_address.
 618  *
 619  * @param[in]  addr     A valid 'unix' tsocket_address.
 620  *
 621  * @param[in]  mem_ctx  The talloc memory context to use.
 622  *
 623  * @return              The path of the unix domain socket, NULL on error or if
 624  *                      the tsocket_address doesn't represent an unix domain
 625  *                      endpoint path.
 626  */
 627 char *tsocket_address_unix_path(const struct tsocket_address *addr,
 628                                 TALLOC_CTX *mem_ctx);
 629
 630 #ifdef DOXYGEN
 631 /**
 632  * @brief Create a tdgram_context for a ipv4 or ipv6 UDP communication.
 633  *
 634  * @param[in]  local    An 'inet' tsocket_address for the local endpoint.
 635  *
 636  * @param[in]  remote   An 'inet' tsocket_address for the remote endpoint or
 637  *                      NULL (??? to create a listener?).
 638  *
 639  * @param[in]  mem_ctx  The talloc memory context to use.
 640  *
 641  * @param[in]  dgram    The tdgram_context pointer to setup the udp
 642  *                      communication. The function will allocate the memory.
 643  *
 644  * @return              0 on success, -1 on error with errno set.
 645  */
 646 int tdgram_inet_udp_socket(const struct tsocket_address *local,
 647                             const struct tsocket_address *remote,
 648                             TALLOC_CTX *mem_ctx,
 649                             struct tdgram_context **dgram);
 650 #else
 651 int _tdgram_inet_udp_socket(const struct tsocket_address *local,
 652                             const struct tsocket_address *remote,
 653                             TALLOC_CTX *mem_ctx,
 654                             struct tdgram_context **dgram,
 655                             const char *location);
 656 #define tdgram_inet_udp_socket(local, remote, mem_ctx, dgram) \
 657         _tdgram_inet_udp_socket(local, remote, mem_ctx, dgram, __location__)
 658 #endif
 659
 660 #ifdef DOXYGEN
 661 /**
 662  * @brief Create a tdgram_context for unix domain datagram communication.
 663  *
 664  * @param[in]  local    An 'unix' tsocket_address for the local endpoint.
 665  *
 666  * @param[in]  remote   An 'unix' tsocket_address for the remote endpoint or
 667  *                      NULL (??? to create a listener?).
 668  *
 669  * @param[in]  mem_ctx  The talloc memory context to use.
 670  *
 671  * @param[in]  dgram    The tdgram_context pointer to setup the udp
 672  *                      communication. The function will allocate the memory.
 673  *
 674  * @return              0 on success, -1 on error with errno set.
 675  */
 676 int tdgram_unix_socket(const struct tsocket_address *local,
 677                         const struct tsocket_address *remote,
 678                         TALLOC_CTX *mem_ctx,
 679                         struct tdgram_context **dgram);
 680 #else
 681 int _tdgram_unix_socket(const struct tsocket_address *local,
 682                         const struct tsocket_address *remote,
 683                         TALLOC_CTX *mem_ctx,
 684                         struct tdgram_context **dgram,
 685                         const char *location);
 686
 687 #define tdgram_unix_socket(local, remote, mem_ctx, dgram) \
 688         _tdgram_unix_socket(local, remote, mem_ctx, dgram, __location__)
 689 #endif
 690
 691 /**
 692  * @brief Connect async to a TCP endpoint and create a tstream_context for the
 693  * stream based communication.
 694  *
 695  * Use this function to connenct asynchronously to a remote ipv4 or ipv6 TCP
 696  * endpoint and create a tstream_context for the stream based communication.
 697  *
 698  * @param[in]  mem_ctx  The talloc memory context to use.
 699  *
 700  * @param[in]  ev       The tevent_context to run on.
 701  *
 702  * @param[in]  local    An 'inet' tsocket_address for the local endpoint.
 703  *
 704  * @param[in]  remote   An 'inet' tsocket_address for the remote endpoint.
 705  *
 706  * @return              A 'tevent_req' handle, where the caller can register a
 707  *                      callback with tevent_req_set_callback(). NULL on a fatal
 708  *                      error.
 709  *
 710  * @see tstream_inet_tcp_connect_recv()
 711  */
 712 struct tevent_req *tstream_inet_tcp_connect_send(TALLOC_CTX *mem_ctx,
 713                                         struct tevent_context *ev,
 714                                         const struct tsocket_address *local,
 715                                         const struct tsocket_address *remote);
 716
 717 #ifdef DOXYGEN
 718 /**
 719  * @brief Receive the result from a tstream_inet_tcp_connect_send().
 720  *
 721  * @param[in]  req      The tevent request from tstream_inet_tcp_connect_send().
 722  *
 723  * @param[out] perrno   The error number, set if an error occurred.
 724  *
 725  * @param[in]  mem_ctx  The talloc memory context to use.
 726  *
 727  * @param[in]  stream   A tstream_context pointer to setup the tcp communication
 728  *                      on. This function will allocate the memory.
 729  *
 730  * @return              0 on success, -1 on error with perrno set.
 731  */
 732 int tstream_inet_tcp_connect_recv(struct tevent_req *req,
 733                                   int *perrno,
 734                                   TALLOC_CTX *mem_ctx,
 735                                   struct tstream_context **stream);
 736 #else
 737 int _tstream_inet_tcp_connect_recv(struct tevent_req *req,
 738                                    int *perrno,
 739                                    TALLOC_CTX *mem_ctx,
 740                                    struct tstream_context **stream,
 741                                    const char *location);
 742 #define tstream_inet_tcp_connect_recv(req, perrno, mem_ctx, stream) \
 743         _tstream_inet_tcp_connect_recv(req, perrno, mem_ctx, stream, \
 744                                        __location__)
 745 #endif
 746
 747 /**
 748  * @brief Connect async to a unix domain endpoint and create a tstream_context
 749  * for the stream based communication.
 750  *
 751  * Use this function to connenct asynchronously to a unix domainendpoint and
 752  * create a tstream_context for the stream based communication.
 753  *
 754  * The callback is triggered when a socket is connected and ready for IO or an
 755  * error happened.
 756  *
 757  * @param[in]  mem_ctx  The talloc memory context to use.
 758  *
 759  * @param[in]  ev       The tevent_context to run on.
 760  *
 761  * @param[in]  local    An 'unix' tsocket_address for the local endpoint.
 762  *
 763  * @param[in]  remote   An 'unix' tsocket_address for the remote endpoint.
 764  *
 765  * @return              A 'tevent_req' handle, where the caller can register a
 766  *                      callback with tevent_req_set_callback(). NULL on a falal
 767  *                      error.
 768  *
 769  * @see tstream_unix_connect_recv()
 770  */
 771 struct tevent_req * tstream_unix_connect_send(TALLOC_CTX *mem_ctx,
 772                                         struct tevent_context *ev,
 773                                         const struct tsocket_address *local,
 774                                         const struct tsocket_address *remote);
 775
 776 #ifdef DOXYGEN
 777 /**
 778  * @brief Receive the result from a tstream_unix_connect_send().
 779  *
 780  * @param[in]  req      The tevent request from tstream_inet_tcp_connect_send().
 781  *
 782  * @param[out] perrno   The error number, set if an error occurred.
 783  *
 784  * @param[in]  mem_ctx  The talloc memory context to use.
 785  *
 786  * @param[in]  stream   The tstream context to work on.
 787  *
 788  * @return              0 on success, -1 on error with perrno set.
 789  */
 790 int tstream_unix_connect_recv(struct tevent_req *req,
 791                               int *perrno,
 792                               TALLOC_CTX *mem_ctx,
 793                               struct tstream_context **stream);
 794 #else
 795 int _tstream_unix_connect_recv(struct tevent_req *req,
 796                                int *perrno,
 797                                TALLOC_CTX *mem_ctx,
 798                                struct tstream_context **stream,
 799                                const char *location);
 800 #define tstream_unix_connect_recv(req, perrno, mem_ctx, stream) \
 801         _tstream_unix_connect_recv(req, perrno, mem_ctx, stream, \
 802                                           __location__)
 803 #endif
 804
 805 #ifdef DOXYGEN
 806 /**
 807  * @brief Create two connected 'unix' tsocket_contexts for stream based
 808  *        communication.
 809  *
 810  * @param[in]  mem_ctx1 The talloc memory context to use for stream1.
 811  *
 812  * @param[in]  stream1  The first stream to connect.
 813  *
 814  * @param[in]  mem_ctx2 The talloc memory context to use for stream2.
 815  *
 816  * @param[in]  stream2  The second stream to connect.
 817  *
 818  * @return              0 on success, -1 on error with errno set.
 819  */
 820 int tstream_unix_socketpair(TALLOC_CTX *mem_ctx1,
 821                             struct tstream_context **stream1,
 822                             TALLOC_CTX *mem_ctx2,
 823                             struct tstream_context **stream2);
 824 #else
 825 int _tstream_unix_socketpair(TALLOC_CTX *mem_ctx1,
 826                              struct tstream_context **_stream1,
 827                              TALLOC_CTX *mem_ctx2,
 828                              struct tstream_context **_stream2,
 829                              const char *location);
 830
 831 #define tstream_unix_socketpair(mem_ctx1, stream1, mem_ctx2, stream2) \
 832         _tstream_unix_socketpair(mem_ctx1, stream1, mem_ctx2, stream2, \
 833                                  __location__)
 834 #endif
 835
 836 struct sockaddr;
 837
 838 #ifdef DOXYGEN
 839 /**
 840  * @brief Convert a tsocket address to a bsd socket address.
 841  *
 842  * @param[in]  mem_ctx  The talloc memory context to use.
 843  *
 844  * @param[in]  sa       The sockaddr structure to convert.
 845  *
 846  * @param[in]  sa_socklen   The lenth of the sockaddr sturucte.
 847  *
 848  * @param[out] addr     The tsocket pointer to allocate and fill.
 849  *
 850  * @return              0 on success, -1 on error with errno set.
 851  */
 852 int tsocket_address_bsd_from_sockaddr(TALLOC_CTX *mem_ctx,
 853                                       struct sockaddr *sa,
 854                                       size_t sa_socklen,
 855                                       struct tsocket_address **addr);
 856 #else
 857 int _tsocket_address_bsd_from_sockaddr(TALLOC_CTX *mem_ctx,
 858                                        struct sockaddr *sa,
 859                                        size_t sa_socklen,
 860                                        struct tsocket_address **_addr,
 861                                        const char *location);
 862
 863 #define tsocket_address_bsd_from_sockaddr(mem_ctx, sa, sa_socklen, _addr) \
 864         _tsocket_address_bsd_from_sockaddr(mem_ctx, sa, sa_socklen, _addr, \
 865                                            __location__)
 866 #endif
 867
 868 /**
 869  * @brief Fill a bsd sockaddr structure.
 870  *
 871  * @param[in]  addr     The tsocket address structure to use.
 872  *
 873  * @param[in]  sa       The bsd sockaddr structure to fill out.
 874  *
 875  * @param[in]  sa_socklen   The length of the  bsd sockaddr structure to fill out.
 876  *
 877  * @return              The actual size of the sockaddr structure, -1 on error
 878  *                      with errno set. The size could differ from sa_socklen.
 879  *
 880  * @code
 881  *   ssize_t socklen;
 882  *   struct sockaddr_storage ss;
 883  *
 884  *   socklen = tsocket_address_bsd_sockaddr(taddr,
 885  *                    (struct sockaddr *) &ss,
 886  *                    sizeof(struct sockaddr_storage));
 887  *   if (socklen < 0) {
 888  *     return -1;
 889  *   }
 890  * @endcode
 891  */
 892 ssize_t tsocket_address_bsd_sockaddr(const struct tsocket_address *addr,
 893                                      struct sockaddr *sa,
 894                                      size_t sa_socklen);
 895
 896 #ifdef DOXYGEN
 897 /**
 898  * @brief Wrap an existing file descriptors into the tstream abstraction.
 899  *
 900  * You can use this function to wrap an existing file descriptors into the
 901  * tstream abstraction. After that you're not able to use this file descriptor
 902  * for anything else. The file descriptor will be closed when the stream gets
 903  * freed. If you still want to use the fd you have have to create a duplicate.
 904  *
 905  * @param[in]  mem_ctx  The talloc memory context to use.
 906  *
 907  * @param[in]  fd       The non blocking fd to use!
 908  *
 909  * @param[out] stream   A pointer to store an allocated tstream_context.
 910  *
 911  * @return              0 on success, -1 on error.
 912  *
 913  * Example:
 914  * @code
 915  *   fd2 = dup(fd);
 916  *   rc = tstream_bsd_existing_socket(mem_ctx, fd2, &tstream);
 917  *   if (rc < 0) {
 918  *     stream_terminate_connection(conn, "named_pipe_accept: out of memory");
 919  *     return;
 920  *   }
 921  * @endcode
 922  *
 923  * @warning This is an internal function. You should read the code to fully
 924  *          understand it if you plan to use it.
 925  */
 926 int tstream_bsd_existing_socket(TALLOC_CTX *mem_ctx,
 927                                 int fd,
 928                                 struct tstream_context **stream);
 929 #else
 930 int _tstream_bsd_existing_socket(TALLOC_CTX *mem_ctx,
 931                                  int fd,
 932                                  struct tstream_context **_stream,
 933                                  const char *location);
 934 #define tstream_bsd_existing_socket(mem_ctx, fd, stream) \
 935         _tstream_bsd_existing_socket(mem_ctx, fd, stream, \
 936                                      __location__)
 937 #endif
 938
 939 /**
 940  * @}
 941  */
 942
 943 /**
 944  * @defgroup tsocket_helper Queue and PDU helpers
 945  * @ingroup tsocket
 946  *
 947  * In order to make the live easier for callers which want to implement a
 948  * function to receive a full PDU with a single async function pair, there're
 949  * some helper functions.
 950  *
 951  * There're some cases where the caller wants doesn't care about the order of
 952  * doing IO on the abstracted sockets.
 953  *
 954  * @{
 955  */
 956
 957 /**
 958  * @brief Queue a dgram blob for sending through the socket.
 959  *
 960  * This function queues a blob for sending to destination through an existing
 961  * dgram socket. The async callback is triggered when the whole blob is
 962  * delivered to the underlying system socket.
 963  *
 964  * The caller needs to make sure that all non-scalar input parameters hang
 965  * around for the whole lifetime of the request.
 966  *
 967  * @param[in]  mem_ctx  The memory context for the result.
 968  *
 969  * @param[in]  ev       The event context the operation should work on.
 970  *
 971  * @param[in]  dgram    The tdgram_context to send the message buffer.
 972  *
 973  * @param[in]  queue    The existing dgram queue.
 974  *
 975  * @param[in]  buf      The message buffer to send.
 976  *
 977  * @param[in]  len      The message length.
 978  *
 979  * @param[in]  dst      The destination socket address.
 980  *
 981  * @return              The async request handle. NULL on fatal error.
 982  *
 983  * @see tdgram_sendto_queue_recv()
 984  */
 985 struct tevent_req *tdgram_sendto_queue_send(TALLOC_CTX *mem_ctx,
 986                                             struct tevent_context *ev,
 987                                             struct tdgram_context *dgram,
 988                                             struct tevent_queue *queue,
 989                                             const uint8_t *buf,
 990                                             size_t len,
 991                                             struct tsocket_address *dst);
 992
 993 /**
 994  * @brief Receive the result of the sent dgram blob.
 995  *
 996  * @param[in]  req      The tevent request from tdgram_sendto_queue_send().
 997  *
 998  * @param[out] perrno   The error set to the actual errno.
 999  *
1000  * @return              The length of the datagram (0 is never returned!), -1 on
1001  *                      error with perrno set to the actual errno.
1002  */
1003 ssize_t tdgram_sendto_queue_recv(struct tevent_req *req, int *perrno);
1004
1005 typedef int (*tstream_readv_pdu_next_vector_t)(struct tstream_context *stream,
1006                                                void *private_data,
1007                                                TALLOC_CTX *mem_ctx,
1008                                                struct iovec **vector,
1009                                                size_t *count);
1010
1011 struct tevent_req *tstream_readv_pdu_send(TALLOC_CTX *mem_ctx,
1012                                 struct tevent_context *ev,
1013                                 struct tstream_context *stream,
1014                                 tstream_readv_pdu_next_vector_t next_vector_fn,
1015                                 void *next_vector_private);
1016 int tstream_readv_pdu_recv(struct tevent_req *req, int *perrno);
1017
1018 /**
1019  * @brief Queue a read request for a PDU on the socket.
1020  *
1021  * This function queues a read request for a PDU on a stream socket. The async
1022  * callback is triggered when a full PDU has been read from the socket.
1023  *
1024  * The caller needs to make sure that all non-scalar input parameters hang
1025  * around for the whole lifetime of the request.
1026  *
1027  * @param[in]  mem_ctx  The memory context for the result
1028  *
1029  * @param[in]  ev       The tevent_context to run on
1030  *
1031  * @param[in]  stream   The stream to send data through
1032  *
1033  * @param[in]  queue    The existing send queue
1034  *
1035  * @param[in]  next_vector_fn  The next vector function
1036  *
1037  * @param[in]  next_vector_private  The private_data of the next vector function
1038  *
1039  * @return              The async request handle. NULL on fatal error.
1040  *
1041  * @see tstream_readv_pdu_queue_recv()
1042  */
1043 struct tevent_req *tstream_readv_pdu_queue_send(TALLOC_CTX *mem_ctx,
1044                                 struct tevent_context *ev,
1045                                 struct tstream_context *stream,
1046                                 struct tevent_queue *queue,
1047                                 tstream_readv_pdu_next_vector_t next_vector_fn,
1048                                 void *next_vector_private);
1049
1050 /**
1051  * @brief Receive the PDU blob read from the stream.
1052  *
1053  * @param[in]  req      The tevent request from tstream_readv_pdu_queue_send().
1054  *
1055  * @param[out] perrno   The error set to the actual errno.
1056  *
1057  * @return              The number of bytes read on success, -1 on error with
1058  *                      perrno set to the actual errno.
1059  */
1060 int tstream_readv_pdu_queue_recv(struct tevent_req *req, int *perrno);
1061
1062 /**
1063  * @brief Queue an iovector for sending through the socket
1064  *
1065  * This function queues an iovector for sending to destination through an
1066  * existing stream socket. The async callback is triggered when the whole
1067  * vectror has been delivered to the underlying system socket.
1068  *
1069  * The caller needs to make sure that all non-scalar input parameters hang
1070  * around for the whole lifetime of the request.
1071  *
1072  * @param[in]  mem_ctx  The memory context for the result.
1073  *
1074  * @param[in]  ev       The tevent_context to run on.
1075  *
1076  * @param[in]  stream   The stream to send data through.
1077  *
1078  * @param[in]  queue    The existing send queue.
1079  *
1080  * @param[in]  vector   The iovec vector so write.
1081  *
1082  * @param[in]  count    The size of the vector.
1083  *
1084  * @return              The async request handle. NULL on fatal error.
1085  */
1086 struct tevent_req *tstream_writev_queue_send(TALLOC_CTX *mem_ctx,
1087                                              struct tevent_context *ev,
1088                                              struct tstream_context *stream,
1089                                              struct tevent_queue *queue,
1090                                              const struct iovec *vector,
1091                                              size_t count);
1092
1093 /**
1094  * @brief Receive the result of the sent iovector.
1095  *
1096  * @param[in]  req      The tevent request from tstream_writev_queue_send().
1097  *
1098  * @param[out] perrno   The error set to the actual errno.
1099  *
1100  * @return              The length of the iovector (0 is never returned!), -1 on
1101  *                      error with perrno set to the actual errno.
1102  */
1103 int tstream_writev_queue_recv(struct tevent_req *req, int *perrno);
1104
1105 /**
1106  * @}
1107  */
1108
1109 #endif /* _TSOCKET_H */
1110