]> git.sur5r.net Git - openldap/blobdiff - servers/slapd/sets.c
use a placeholder to mark that there is no process in the runqueue (ITS#4318)
[openldap] / servers / slapd / sets.c
index 9fbc04afb6f7e336e4970954cc9b4a92ac946b37..fa7b92e90c1ef1d5d5a9989740bc743ca2ed7940 100644 (file)
@@ -1,7 +1,7 @@
 /* $OpenLDAP$ */
 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
  *
- * Copyright 2000-2004 The OpenLDAP Foundation.
+ * Copyright 2000-2006 The OpenLDAP Foundation.
  * All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
@@ -53,37 +53,91 @@ slap_set_isempty( BerVarray set )
 }
 
 static void
-slap_set_dispose( SetCookie *cp, BerVarray set )
+slap_set_dispose( SetCookie *cp, BerVarray set, unsigned flags )
 {
-       ber_bvarray_free_x( set, cp->op->o_tmpmemctx );
+       if ( flags & SLAP_SET_REFVAL ) {
+               if ( ! ( flags & SLAP_SET_REFARR ) ) {
+                       cp->set_op->o_tmpfree( set, cp->set_op->o_tmpmemctx );
+               }
+
+       } else {
+               ber_bvarray_free_x( set, cp->set_op->o_tmpmemctx );
+       }
+}
+
+static BerVarray
+set_dup( SetCookie *cp, BerVarray set, unsigned flags )
+{
+       BerVarray       newset = NULL;
+
+       if ( set == NULL ) {
+               return NULL;
+       }
+
+       if ( flags & SLAP_SET_REFARR ) {
+               int     i;
+
+               for ( i = 0; !BER_BVISNULL( &set[ i ] ); i++ )
+                       ;
+               newset = cp->set_op->o_tmpcalloc( i + 1,
+                               sizeof(struct berval), 
+                               cp->set_op->o_tmpmemctx );
+               if ( newset == NULL ) {
+                       return NULL;
+               }
+
+               if ( flags & SLAP_SET_REFVAL ) {
+                       for ( i = 0; !BER_BVISNULL( &set[ i ] ); i++ ) {
+                               ber_dupbv_x( &newset[ i ], &set[ i ],
+                                               cp->set_op->o_tmpmemctx );
+                       }
+
+               } else {
+                       AC_MEMCPY( newset, set, ( i + 1 ) * sizeof( struct berval ) );
+               }
+               
+       } else {
+               newset = set;
+       }
+
+       return newset;
 }
 
 BerVarray
-slap_set_join( SetCookie *cp, BerVarray lset, int op, BerVarray rset )
+slap_set_join(
+       SetCookie       *cp,
+       BerVarray       lset,
+       unsigned        op_flags,
+       BerVarray       rset )
 {
        BerVarray       set;
        long            i, j, last;
+       unsigned        op = ( op_flags & SLAP_SET_OPMASK );
 
        set = NULL;
-       if ( op == '|' ) {
+       switch ( op ) {
+       case '|':       /* union */
                if ( lset == NULL || BER_BVISNULL( lset ) ) {
                        if ( rset == NULL ) {
                                if ( lset == NULL ) {
-                                       return cp->op->o_tmpcalloc( 1, sizeof(struct berval),
-                                                       cp->op->o_tmpmemctx);
+                                       set = cp->set_op->o_tmpcalloc( 1,
+                                                       sizeof(struct berval),
+                                                       cp->set_op->o_tmpmemctx );
+                                       BER_BVZERO( set );
+                                       return set;
                                }
-                               return lset;
+                               return set_dup( cp, lset, SLAP_SET_LREF2REF( op_flags ) );
                        }
-                       slap_set_dispose( cp, lset );
-                       return rset;
+                       slap_set_dispose( cp, lset, SLAP_SET_LREF2REF( op_flags ) );
+                       return set_dup( cp, rset, SLAP_SET_RREF2REF( op_flags ) );
                }
                if ( rset == NULL || BER_BVISNULL( rset ) ) {
-                       slap_set_dispose( cp, rset );
-                       return lset;
+                       slap_set_dispose( cp, rset, SLAP_SET_RREF2REF( op_flags ) );
+                       return set_dup( cp, lset, SLAP_SET_LREF2REF( op_flags ) );
                }
 
                i = slap_set_size( lset ) + slap_set_size( rset ) + 1;
-               set = cp->op->o_tmpcalloc( i, sizeof(struct berval), cp->op->o_tmpmemctx );
+               set = cp->set_op->o_tmpcalloc( i, sizeof(struct berval), cp->set_op->o_tmpmemctx );
                if ( set != NULL ) {
                        /* set_chase() depends on this routine to
                         * keep the first elements of the result
@@ -91,45 +145,69 @@ slap_set_join( SetCookie *cp, BerVarray lset, int op, BerVarray rset )
                         * as the left-set.
                         */
                        for ( i = 0; !BER_BVISNULL( &lset[ i ] ); i++ ) {
-                               set[ i ] = lset[ i ];
+                               if ( op_flags & SLAP_SET_LREFVAL ) {
+                                       ber_dupbv_x( &set[ i ], &lset[ i ], cp->set_op->o_tmpmemctx );
+
+                               } else {
+                                       set[ i ] = lset[ i ];
+                               }
                        }
-                       cp->op->o_tmpfree( lset, cp->op->o_tmpmemctx );
+
+                       last = i;
+
                        for ( i = 0; !BER_BVISNULL( &rset[ i ] ); i++ ) {
+                               int     exists = 0;
+
                                for ( j = 0; !BER_BVISNULL( &set[ j ] ); j++ ) {
-                                       if ( dn_match( &rset[ i ], &set[ j ] ) )
+                                       if ( bvmatch( &rset[ i ], &set[ j ] ) )
                                        {
-                                               cp->op->o_tmpfree( rset[ i ].bv_val, cp->op->o_tmpmemctx );
-                                               BER_BVZERO( &rset[ i ] );
+                                               if ( !( op_flags & SLAP_SET_RREFVAL ) ) {
+                                                       cp->set_op->o_tmpfree( rset[ i ].bv_val, cp->set_op->o_tmpmemctx );
+                                                       BER_BVZERO( &rset[ i ] );
+                                               }
+                                               exists = 1;
                                                break;          
                                        }       
                                }
-                               if ( !BER_BVISNULL( &rset[ i ] ) ) {
-                                       set[ j ] = rset[ i ];
+
+                               if ( !exists ) {
+                                       if ( op_flags & SLAP_SET_RREFVAL ) {
+                                               ber_dupbv_x( &set[ last ], &rset[ i ], cp->set_op->o_tmpmemctx );
+
+                                       } else {
+                                               set[ last ] = rset[ i ];
+                                       }
+                                       last++;
                                }
                        }
-                       cp->op->o_tmpfree( rset, cp->op->o_tmpmemctx );
+                       BER_BVZERO( &set[ last ] );
                }
-               return set;
-       }
+               break;
 
-       if ( op == '&' ) {
+       case '&':       /* intersection */
                if ( lset == NULL || BER_BVISNULL( lset )
                                || rset == NULL || BER_BVISNULL( rset ) )
                {
-                       set = cp->op->o_tmpcalloc( 1, sizeof(struct berval),
-                                       cp->op->o_tmpmemctx );
+                       set = cp->set_op->o_tmpcalloc( 1, sizeof(struct berval),
+                                       cp->set_op->o_tmpmemctx );
+                       BER_BVZERO( set );
+
                } else {
-                       set = lset;
+                       set = set_dup( cp, lset, SLAP_SET_LREF2REF( op_flags ) );
+                       if ( set == NULL ) {
+                               break;
+                       }
                        lset = NULL;
                        last = slap_set_size( set ) - 1;
                        for ( i = 0; !BER_BVISNULL( &set[ i ] ); i++ ) {
                                for ( j = 0; !BER_BVISNULL( &rset[ j ] ); j++ ) {
-                                       if ( dn_match( &set[ i ], &rset[ j ] ) ) {
+                                       if ( bvmatch( &set[ i ], &rset[ j ] ) ) {
                                                break;
                                        }
                                }
+
                                if ( BER_BVISNULL( &rset[ j ] ) ) {
-                                       cp->op->o_tmpfree( set[ i ].bv_val, cp->op->o_tmpmemctx );
+                                       cp->set_op->o_tmpfree( set[ i ].bv_val, cp->set_op->o_tmpmemctx );
                                        set[ i ] = set[ last ];
                                        BER_BVZERO( &set[ last ] );
                                        last--;
@@ -137,10 +215,63 @@ slap_set_join( SetCookie *cp, BerVarray lset, int op, BerVarray rset )
                                }
                        }
                }
+               break;
+
+       case '+':       /* string concatenation */
+               i = slap_set_size( rset );
+               j = slap_set_size( lset );
+
+               set = cp->set_op->o_tmpcalloc( i * j + 1, sizeof(struct berval),
+                               cp->set_op->o_tmpmemctx );
+               if ( set == NULL ) {
+                       break;
+               }
+
+               for ( last = 0, i = 0; !BER_BVISNULL( &lset[ i ] ); i++ ) {
+                       for ( j = 0; !BER_BVISNULL( &rset[ j ] ); j++ ) {
+                               struct berval   bv;
+                               long            k;
+
+                               bv.bv_len = lset[ i ].bv_len + rset[ j ].bv_len;
+                               bv.bv_val = cp->set_op->o_tmpalloc( bv.bv_len + 1,
+                                               cp->set_op->o_tmpmemctx );
+                               if ( bv.bv_val == NULL ) {
+                                       slap_set_dispose( cp, set, 0 );
+                                       set = NULL;
+                                       goto done;
+                               }
+                               AC_MEMCPY( bv.bv_val, lset[ i ].bv_val, lset[ i ].bv_len );
+                               AC_MEMCPY( &bv.bv_val[ lset[ i ].bv_len ], rset[ j ].bv_val, rset[ j ].bv_len );
+                               bv.bv_val[ bv.bv_len ] = '\0';
+
+                               for ( k = 0; k < last; k++ ) {
+                                       if ( bvmatch( &set[ k ], &bv ) ) {
+                                               cp->set_op->o_tmpfree( bv.bv_val, cp->set_op->o_tmpmemctx );
+                                               break;
+                                       }
+                               }
+
+                               if ( k == last ) {
+                                       set[ last++ ] = bv;
+                               }
+                       }
+               }
+               BER_BVZERO( &set[ last ] );
+               break;
+
+       default:
+               break;
+       }
+
+done:;
+       if ( !( op_flags & SLAP_SET_LREFARR ) && lset != NULL ) {
+               cp->set_op->o_tmpfree( lset, cp->set_op->o_tmpmemctx );
+       }
+
+       if ( !( op_flags & SLAP_SET_RREFARR ) && rset != NULL ) {
+               cp->set_op->o_tmpfree( rset, cp->set_op->o_tmpmemctx );
        }
 
-       slap_set_dispose( cp, lset );
-       slap_set_dispose( cp, rset );
        return set;
 }
 
@@ -151,16 +282,20 @@ set_chase( SLAP_SET_GATHER gatherer,
        BerVarray       vals, nset;
        int             i;
 
-       if ( set == NULL )
-               return cp->op->o_tmpcalloc( 1, sizeof(struct berval),
-                               cp->op->o_tmpmemctx );
+       if ( set == NULL ) {
+               set = cp->set_op->o_tmpcalloc( 1, sizeof(struct berval),
+                               cp->set_op->o_tmpmemctx );
+               BER_BVZERO( set );
+               return set;
+       }
 
-       if ( BER_BVISNULL( set ) )
+       if ( BER_BVISNULL( set ) ) {
                return set;
+       }
 
-       nset = cp->op->o_tmpcalloc( 1, sizeof(struct berval), cp->op->o_tmpmemctx );
+       nset = cp->set_op->o_tmpcalloc( 1, sizeof(struct berval), cp->set_op->o_tmpmemctx );
        if ( nset == NULL ) {
-               slap_set_dispose( cp, set );
+               slap_set_dispose( cp, set, 0 );
                return NULL;
        }
        for ( i = 0; !BER_BVISNULL( &set[ i ] ); i++ ) {
@@ -169,7 +304,7 @@ set_chase( SLAP_SET_GATHER gatherer,
                        nset = slap_set_join( cp, nset, '|', vals );
                }
        }
-       slap_set_dispose( cp, set );
+       slap_set_dispose( cp, set, 0 );
 
        if ( closure ) {
                for ( i = 0; !BER_BVISNULL( &nset[ i ] ); i++ ) {
@@ -191,19 +326,21 @@ slap_set_filter( SLAP_SET_GATHER gatherer,
        SetCookie *cp, struct berval *fbv,
        struct berval *user, struct berval *target, BerVarray *results )
 {
+#define STACK_SIZE     64
 #define IS_SET(x)      ( (unsigned long)(x) >= 256 )
 #define IS_OP(x)       ( (unsigned long)(x) < 256 )
 #define SF_ERROR(x)    do { rc = -1; goto _error; } while (0)
 #define SF_TOP()       ( (BerVarray)( (stp < 0) ? 0 : stack[ stp ] ) )
 #define SF_POP()       ( (BerVarray)( (stp < 0) ? 0 : stack[ stp-- ] ) )
 #define SF_PUSH(x)     do { \
-               if (stp >= 63) SF_ERROR(overflow); \
+               if (stp >= (STACK_SIZE - 1)) SF_ERROR(overflow); \
                stack[ ++stp ] = (BerVarray)(long)(x); \
        } while (0)
 
        BerVarray       set, lset;
-       BerVarray       stack[64] = { 0 };
-       int             len, op, rc, stp;
+       BerVarray       stack[ STACK_SIZE ] = { 0 };
+       int             len, rc, stp;
+       unsigned long   op;
        char            c, *filter = fbv->bv_val;
 
        if ( results ) {
@@ -238,7 +375,7 @@ slap_set_filter( SLAP_SET_GATHER gatherer,
                                set = NULL;
 
                        } else if ( IS_OP( SF_TOP() ) ) {
-                               op = (long)SF_POP();
+                               op = (unsigned long)SF_POP();
                                lset = SF_POP();
                                SF_POP();
                                set = slap_set_join( cp, lset, op, set );
@@ -253,8 +390,9 @@ slap_set_filter( SLAP_SET_GATHER gatherer,
                        }
                        break;
 
-               case '&':
-               case '|':
+               case '|':       /* union */
+               case '&':       /* intersection */
+               case '+':       /* string concatenation */
                        set = SF_POP();
                        if ( IS_OP( set ) ) {
                                SF_ERROR( syntax );
@@ -264,7 +402,7 @@ slap_set_filter( SLAP_SET_GATHER gatherer,
                                set = NULL;
 
                        } else if ( IS_OP( SF_TOP() ) ) {
-                               op = (long)SF_POP();
+                               op = (unsigned long)SF_POP();
                                lset = SF_POP();
                                set = slap_set_join( cp, lset, op, set );
                                if ( set == NULL ) {
@@ -289,13 +427,13 @@ slap_set_filter( SLAP_SET_GATHER gatherer,
                                SF_ERROR(syntax);
                        }
                        
-                       set = cp->op->o_tmpcalloc( 2, sizeof(struct berval),
-                                       cp->op->o_tmpmemctx );
+                       set = cp->set_op->o_tmpcalloc( 2, sizeof(struct berval),
+                                       cp->set_op->o_tmpmemctx );
                        if ( set == NULL ) {
                                SF_ERROR(memory);
                        }
-                       set->bv_val = cp->op->o_tmpcalloc( len + 1, sizeof(char),
-                                       cp->op->o_tmpmemctx );
+                       set->bv_val = cp->set_op->o_tmpcalloc( len + 1, sizeof(char),
+                                       cp->set_op->o_tmpmemctx );
                        if ( BER_BVISNULL( set ) ) {
                                SF_ERROR( memory );
                        }
@@ -340,15 +478,16 @@ slap_set_filter( SLAP_SET_GATHER gatherer,
                                if ( ( SF_TOP() == (void *)'/' ) || IS_SET( SF_TOP() ) ) {
                                        SF_ERROR( syntax );
                                }
-                               set = cp->op->o_tmpcalloc( 2, sizeof(struct berval),
-                                               cp->op->o_tmpmemctx );
+                               set = cp->set_op->o_tmpcalloc( 2, sizeof(struct berval),
+                                               cp->set_op->o_tmpmemctx );
                                if ( set == NULL ) {
                                        SF_ERROR( memory );
                                }
-                               ber_dupbv_x( set, target, cp->op->o_tmpmemctx );
+                               ber_dupbv_x( set, target, cp->set_op->o_tmpmemctx );
                                if ( BER_BVISNULL( set ) ) {
                                        SF_ERROR( memory );
                                }
+                               BER_BVZERO( &set[ 1 ] );
                                
                        } else if ( len == 4
                                && memcmp( "user", filter, len ) == 0 ) 
@@ -356,15 +495,16 @@ slap_set_filter( SLAP_SET_GATHER gatherer,
                                if ( ( SF_TOP() == (void *)'/' ) || IS_SET( SF_TOP() ) ) {
                                        SF_ERROR( syntax );
                                }
-                               set = cp->op->o_tmpcalloc( 2, sizeof(struct berval),
-                                               cp->op->o_tmpmemctx );
+                               set = cp->set_op->o_tmpcalloc( 2, sizeof(struct berval),
+                                               cp->set_op->o_tmpmemctx );
                                if ( set == NULL ) {
                                        SF_ERROR( memory );
                                }
-                               ber_dupbv_x( set, user, cp->op->o_tmpmemctx );
+                               ber_dupbv_x( set, user, cp->set_op->o_tmpmemctx );
                                if ( BER_BVISNULL( set ) ) {
                                        SF_ERROR( memory );
                                }
+                               BER_BVZERO( &set[ 1 ] );
                                
                        } else if ( SF_TOP() != (void *)'/' ) {
                                SF_ERROR( syntax );
@@ -410,7 +550,7 @@ slap_set_filter( SLAP_SET_GATHER gatherer,
                /* FIXME: ok ? */ ;
 
        } else if ( IS_OP( SF_TOP() ) ) {
-               op = (long)SF_POP();
+               op = (unsigned long)SF_POP();
                lset = SF_POP();
                set = slap_set_join( cp, lset, op, set );
                if ( set == NULL ) {
@@ -429,11 +569,11 @@ slap_set_filter( SLAP_SET_GATHER gatherer,
 
 _error:
        if ( IS_SET( set ) ) {
-               slap_set_dispose( cp, set );
+               slap_set_dispose( cp, set, 0 );
        }
        while ( ( set = SF_POP() ) ) {
                if ( IS_SET( set ) ) {
-                       slap_set_dispose( cp, set );
+                       slap_set_dispose( cp, set, 0 );
                }
        }
        return rc;