Diff for /libraries/libldap/sasl.c between versions 1.58.2.2 and 1.64.2.8

version 1.58.2.2, 2005/08/10 21:57:17 version 1.64.2.8, 2010/04/13 20:22:59
Line 1 Line 1
 /* $OpenLDAP: pkg/ldap/libraries/libldap/sasl.c,v 1.60 2005/08/09 22:54:38 hallvard Exp $ */  /* $OpenLDAP: pkg/ldap/libraries/libldap/sasl.c,v 1.64.2.7 2009/10/31 00:11:22 quanah Exp $ */
 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.  /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
  *   *
  * Copyright 1998-2005 The OpenLDAP Foundation.   * Copyright 1998-2010 The OpenLDAP Foundation.
  * All rights reserved.   * All rights reserved.
  *   *
  * Redistribution and use in source and binary forms, with or without   * Redistribution and use in source and binary forms, with or without
Line 12 Line 12
  * top-level directory of the distribution or, alternatively, at   * top-level directory of the distribution or, alternatively, at
  * <http://www.OpenLDAP.org/license.html>.   * <http://www.OpenLDAP.org/license.html>.
  */   */
 /* Portions Copyright (C) The Internet Society (1997)  
  * ASN.1 fragments are from RFC 2251; see RFC for full legal notices.  
  */  
   
 /*  /*
  *      BindRequest ::= SEQUENCE {   *      BindRequest ::= SEQUENCE {
Line 22 Line 19
  *              name            DistinguishedName,       -- who   *              name            DistinguishedName,       -- who
  *              authentication  CHOICE {   *              authentication  CHOICE {
  *                      simple          [0] OCTET STRING -- passwd   *                      simple          [0] OCTET STRING -- passwd
 #ifdef LDAP_API_FEATURE_X_OPENLDAP_V2_KBIND   *                      krbv42ldap      [1] OCTET STRING -- OBSOLETE
  *                      krbv42ldap      [1] OCTET STRING   *                      krbv42dsa       [2] OCTET STRING -- OBSOLETE
  *                      krbv42dsa       [2] OCTET STRING  
 #endif  
  *                      sasl            [3] SaslCredentials     -- LDAPv3   *                      sasl            [3] SaslCredentials     -- LDAPv3
  *              }   *              }
  *      }   *      }
Line 196  ldap_sasl_bind_s( Line 191  ldap_sasl_bind_s(
         }          }
 #endif  #endif
   
         if ( ldap_result( ld, msgid, 1, NULL, &result ) == -1 ) {          if ( ldap_result( ld, msgid, LDAP_MSG_ALL, NULL, &result ) == -1 || !result ) {
                 return( ld->ld_errno ); /* ldap_result sets ld_errno */                  return( ld->ld_errno ); /* ldap_result sets ld_errno */
         }          }
   
Line 206  ldap_sasl_bind_s( Line 201  ldap_sasl_bind_s(
                 rc = ldap_parse_sasl_bind_result( ld, result, &scredp, 0 );                  rc = ldap_parse_sasl_bind_result( ld, result, &scredp, 0 );
         }          }
   
         if ( rc != LDAP_SUCCESS && rc != LDAP_SASL_BIND_IN_PROGRESS ) {          if ( rc != LDAP_SUCCESS ) {
                 ldap_msgfree( result );                  ldap_msgfree( result );
                 return( rc );                  return( rc );
         }          }
Line 294  ldap_parse_sasl_bind_result( Line 289  ldap_parse_sasl_bind_result(
         }          }
   
         if ( ld->ld_version < LDAP_VERSION2 ) {          if ( ld->ld_version < LDAP_VERSION2 ) {
                 tag = ber_scanf( ber, "{ia}",                  tag = ber_scanf( ber, "{iA}",
                         &errcode, &ld->ld_error );                          &errcode, &ld->ld_error );
   
                 if( tag == LBER_ERROR ) {                  if( tag == LBER_ERROR ) {
Line 306  ldap_parse_sasl_bind_result( Line 301  ldap_parse_sasl_bind_result(
         } else {          } else {
                 ber_len_t len;                  ber_len_t len;
   
                 tag = ber_scanf( ber, "{eaa" /*}*/,                  tag = ber_scanf( ber, "{eAA" /*}*/,
                         &errcode, &ld->ld_matched, &ld->ld_error );                          &errcode, &ld->ld_matched, &ld->ld_error );
   
                 if( tag == LBER_ERROR ) {                  if( tag == LBER_ERROR ) {
Line 352  ldap_parse_sasl_bind_result( Line 347  ldap_parse_sasl_bind_result(
                 ldap_msgfree( res );                  ldap_msgfree( res );
         }          }
   
         return( ld->ld_errno );          return( LDAP_SUCCESS );
 }  }
   
 int  int
Line 478  done: Line 473  done:
   
         return rc;          return rc;
 }  }
   
   #ifdef HAVE_CYRUS_SASL
   
   #ifdef HAVE_SASL_SASL_H
   #include <sasl/sasl.h>
   #else
   #include <sasl.h>
   #endif
   
   #endif /* HAVE_CYRUS_SASL */
   
   static int
   sb_sasl_generic_remove( Sockbuf_IO_Desc *sbiod );
   
   static int
   sb_sasl_generic_setup( Sockbuf_IO_Desc *sbiod, void *arg )
   {
           struct sb_sasl_generic_data     *p;
           struct sb_sasl_generic_install  *i;
   
           assert( sbiod != NULL );
   
           i = (struct sb_sasl_generic_install *)arg;
   
           p = LBER_MALLOC( sizeof( *p ) );
           if ( p == NULL )
                   return -1;
           p->ops = i->ops;
           p->ops_private = i->ops_private;
           p->sbiod = sbiod;
           p->flags = 0;
           ber_pvt_sb_buf_init( &p->sec_buf_in );
           ber_pvt_sb_buf_init( &p->buf_in );
           ber_pvt_sb_buf_init( &p->buf_out );
   
           sbiod->sbiod_pvt = p;
   
           p->ops->init( p, &p->min_send, &p->max_send, &p->max_recv );
   
           if ( ber_pvt_sb_grow_buffer( &p->sec_buf_in, p->min_send ) < 0 ) {
                   sb_sasl_generic_remove( sbiod );
                   sock_errset(ENOMEM);
                   return -1;
           }
   
           return 0;
   }
   
   static int
   sb_sasl_generic_remove( Sockbuf_IO_Desc *sbiod )
   {
           struct sb_sasl_generic_data     *p;
   
           assert( sbiod != NULL );
   
           p = (struct sb_sasl_generic_data *)sbiod->sbiod_pvt;
   
           p->ops->fini(p);
   
           ber_pvt_sb_buf_destroy( &p->sec_buf_in );
           ber_pvt_sb_buf_destroy( &p->buf_in );
           ber_pvt_sb_buf_destroy( &p->buf_out );
           LBER_FREE( p );
           sbiod->sbiod_pvt = NULL;
           return 0;
   }
   
   static ber_len_t
   sb_sasl_generic_pkt_length(
           struct sb_sasl_generic_data *p,
           const unsigned char *buf,
           int debuglevel )
   {
           ber_len_t               size;
   
           assert( buf != NULL );
   
           size = buf[0] << 24
                   | buf[1] << 16
                   | buf[2] << 8
                   | buf[3];
   
           if ( size > p->max_recv ) {
                   /* somebody is trying to mess me up. */
                   ber_log_printf( LDAP_DEBUG_ANY, debuglevel,
                           "sb_sasl_generic_pkt_length: "
                           "received illegal packet length of %lu bytes\n",
                           (unsigned long)size );
                   size = 16; /* this should lead to an error. */
           }
   
           return size + 4; /* include the size !!! */
   }
   
   /* Drop a processed packet from the input buffer */
   static void
   sb_sasl_generic_drop_packet (
           struct sb_sasl_generic_data *p,
           int debuglevel )
   {
           ber_slen_t                      len;
   
           len = p->sec_buf_in.buf_ptr - p->sec_buf_in.buf_end;
           if ( len > 0 )
                   AC_MEMCPY( p->sec_buf_in.buf_base, p->sec_buf_in.buf_base +
                           p->sec_buf_in.buf_end, len );
   
           if ( len >= 4 ) {
                   p->sec_buf_in.buf_end = sb_sasl_generic_pkt_length(p,
                           (unsigned char *) p->sec_buf_in.buf_base, debuglevel);
           }
           else {
                   p->sec_buf_in.buf_end = 0;
           }
           p->sec_buf_in.buf_ptr = len;
   }
   
   static ber_slen_t
   sb_sasl_generic_read( Sockbuf_IO_Desc *sbiod, void *buf, ber_len_t len)
   {
           struct sb_sasl_generic_data     *p;
           ber_slen_t                      ret, bufptr;
   
           assert( sbiod != NULL );
           assert( SOCKBUF_VALID( sbiod->sbiod_sb ) );
   
           p = (struct sb_sasl_generic_data *)sbiod->sbiod_pvt;
   
           /* Are there anything left in the buffer? */
           ret = ber_pvt_sb_copy_out( &p->buf_in, buf, len );
           bufptr = ret;
           len -= ret;
   
           if ( len == 0 )
                   return bufptr;
   
           p->ops->reset_buf( p, &p->buf_in );
   
           /* Read the length of the packet */
           while ( p->sec_buf_in.buf_ptr < 4 ) {
                   ret = LBER_SBIOD_READ_NEXT( sbiod, p->sec_buf_in.buf_base +
                           p->sec_buf_in.buf_ptr,
                           4 - p->sec_buf_in.buf_ptr );
   #ifdef EINTR
                   if ( ( ret < 0 ) && ( errno == EINTR ) )
                           continue;
   #endif
                   if ( ret <= 0 )
                           return bufptr ? bufptr : ret;
   
                   p->sec_buf_in.buf_ptr += ret;
           }
   
           /* The new packet always starts at p->sec_buf_in.buf_base */
           ret = sb_sasl_generic_pkt_length(p, (unsigned char *) p->sec_buf_in.buf_base,
                   sbiod->sbiod_sb->sb_debug );
   
           /* Grow the packet buffer if neccessary */
           if ( ( p->sec_buf_in.buf_size < (ber_len_t) ret ) && 
                   ber_pvt_sb_grow_buffer( &p->sec_buf_in, ret ) < 0 )
           {
                   sock_errset(ENOMEM);
                   return -1;
           }
           p->sec_buf_in.buf_end = ret;
   
           /* Did we read the whole encrypted packet? */
           while ( p->sec_buf_in.buf_ptr < p->sec_buf_in.buf_end ) {
                   /* No, we have got only a part of it */
                   ret = p->sec_buf_in.buf_end - p->sec_buf_in.buf_ptr;
   
                   ret = LBER_SBIOD_READ_NEXT( sbiod, p->sec_buf_in.buf_base +
                           p->sec_buf_in.buf_ptr, ret );
   #ifdef EINTR
                   if ( ( ret < 0 ) && ( errno == EINTR ) )
                           continue;
   #endif
                   if ( ret <= 0 )
                           return bufptr ? bufptr : ret;
   
                   p->sec_buf_in.buf_ptr += ret;
           }
   
           /* Decode the packet */
           ret = p->ops->decode( p, &p->sec_buf_in, &p->buf_in );
   
           /* Drop the packet from the input buffer */
           sb_sasl_generic_drop_packet( p, sbiod->sbiod_sb->sb_debug );
   
           if ( ret != 0 ) {
                   ber_log_printf( LDAP_DEBUG_ANY, sbiod->sbiod_sb->sb_debug,
                           "sb_sasl_generic_read: failed to decode packet\n" );
                   sock_errset(EIO);
                   return -1;
           }
   
           bufptr += ber_pvt_sb_copy_out( &p->buf_in, (char*) buf + bufptr, len );
   
           return bufptr;
   }
   
   static ber_slen_t
   sb_sasl_generic_write( Sockbuf_IO_Desc *sbiod, void *buf, ber_len_t len)
   {
           struct sb_sasl_generic_data     *p;
           int                             ret;
           ber_len_t                       len2;
   
           assert( sbiod != NULL );
           assert( SOCKBUF_VALID( sbiod->sbiod_sb ) );
   
           p = (struct sb_sasl_generic_data *)sbiod->sbiod_pvt;
   
           /* Is there anything left in the buffer? */
           if ( p->buf_out.buf_ptr != p->buf_out.buf_end ) {
                   ret = ber_pvt_sb_do_write( sbiod, &p->buf_out );
                   if ( ret < 0 ) return ret;
   
                   /* Still have something left?? */
                   if ( p->buf_out.buf_ptr != p->buf_out.buf_end ) {
                           sock_errset(EAGAIN);
                           return -1;
                   }
           }
   
           len2 = p->max_send - 100;       /* For safety margin */
           len2 = len > len2 ? len2 : len;
   
           /* If we're just retrying a partial write, tell the
            * caller it's done. Let them call again if there's
            * still more left to write.
            */
           if ( p->flags & LDAP_PVT_SASL_PARTIAL_WRITE ) {
                   p->flags ^= LDAP_PVT_SASL_PARTIAL_WRITE;
                   return len2;
           }
   
           /* now encode the next packet. */
           p->ops->reset_buf( p, &p->buf_out );
   
           ret = p->ops->encode( p, buf, len2, &p->buf_out );
   
           if ( ret != 0 ) {
                   ber_log_printf( LDAP_DEBUG_ANY, sbiod->sbiod_sb->sb_debug,
                           "sb_sasl_generic_write: failed to encode packet\n" );
                   sock_errset(EIO);
                   return -1;
           }
   
           ret = ber_pvt_sb_do_write( sbiod, &p->buf_out );
   
           if ( ret < 0 ) {
                   /* error? */
                   int err = sock_errno();
                   /* caller can retry this */
                   if ( err == EAGAIN || err == EWOULDBLOCK || err == EINTR )
                           p->flags |= LDAP_PVT_SASL_PARTIAL_WRITE;
                   return ret;
           } else if ( p->buf_out.buf_ptr != p->buf_out.buf_end ) {
                   /* partial write? pretend nothing got written */
                   len2 = 0;
                   p->flags |= LDAP_PVT_SASL_PARTIAL_WRITE;
           }
   
           /* return number of bytes encoded, not written, to ensure
            * no byte is encoded twice (even if only sent once).
            */
           return len2;
   }
   
   static int
   sb_sasl_generic_ctrl( Sockbuf_IO_Desc *sbiod, int opt, void *arg )
   {
           struct sb_sasl_generic_data     *p;
   
           p = (struct sb_sasl_generic_data *)sbiod->sbiod_pvt;
   
           if ( opt == LBER_SB_OPT_DATA_READY ) {
                   if ( p->buf_in.buf_ptr != p->buf_in.buf_end ) return 1;
           }
   
           return LBER_SBIOD_CTRL_NEXT( sbiod, opt, arg );
   }
   
   Sockbuf_IO ldap_pvt_sockbuf_io_sasl_generic = {
           sb_sasl_generic_setup,          /* sbi_setup */
           sb_sasl_generic_remove,         /* sbi_remove */
           sb_sasl_generic_ctrl,           /* sbi_ctrl */
           sb_sasl_generic_read,           /* sbi_read */
           sb_sasl_generic_write,          /* sbi_write */
           NULL                    /* sbi_close */
   };
   
   int ldap_pvt_sasl_generic_install(
           Sockbuf *sb,
           struct sb_sasl_generic_install *install_arg )
   {
           Debug( LDAP_DEBUG_TRACE, "ldap_pvt_sasl_generic_install\n",
                   0, 0, 0 );
   
           /* don't install the stuff unless security has been negotiated */
   
           if ( !ber_sockbuf_ctrl( sb, LBER_SB_OPT_HAS_IO,
                           &ldap_pvt_sockbuf_io_sasl_generic ) )
           {
   #ifdef LDAP_DEBUG
                   ber_sockbuf_add_io( sb, &ber_sockbuf_io_debug,
                           LBER_SBIOD_LEVEL_APPLICATION, (void *)"sasl_generic_" );
   #endif
                   ber_sockbuf_add_io( sb, &ldap_pvt_sockbuf_io_sasl_generic,
                           LBER_SBIOD_LEVEL_APPLICATION, install_arg );
           }
   
           return LDAP_SUCCESS;
   }
   
   void ldap_pvt_sasl_generic_remove( Sockbuf *sb )
   {
           ber_sockbuf_remove_io( sb, &ldap_pvt_sockbuf_io_sasl_generic,
                   LBER_SBIOD_LEVEL_APPLICATION );
   #ifdef LDAP_DEBUG
           ber_sockbuf_remove_io( sb, &ber_sockbuf_io_debug,
                   LBER_SBIOD_LEVEL_APPLICATION );
   #endif
   }

Removed from v.1.58.2.2  
changed lines
  Added in v.1.64.2.8


______________
© Copyright 1998-2020, OpenLDAP Foundation, info@OpenLDAP.org