X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=libmemcached%2Fmemcached_callback.c;h=d1c042429b4d5fd4569a98a649f0ede6f2ac957f;hb=d455dd963105dc3ccc120887215f66dec8f55377;hp=ad1ce2b1e65bb4d2bb81478201d1a76320c1231f;hpb=b369db6ab57bc940618281a2be848542a406bb7c;p=awesomized%2Flibmemcached diff --git a/libmemcached/memcached_callback.c b/libmemcached/memcached_callback.c index ad1ce2b1..d1c04242 100644 --- a/libmemcached/memcached_callback.c +++ b/libmemcached/memcached_callback.c @@ -19,13 +19,23 @@ memcached_return memcached_callback_set(memcached_st *ptr, if (key) { - ptr->prefix_key_length= strlen(key); - if ((ptr->prefix_key_length > MEMCACHED_PREFIX_KEY_MAX_SIZE -1) + size_t key_length= strlen(key); + + if (memcachd_key_test((char **)&key, &key_length, 1) == MEMCACHED_BAD_KEY_PROVIDED) + { + return MEMCACHED_BAD_KEY_PROVIDED; + } + + if ((key_length > MEMCACHED_PREFIX_KEY_MAX_SIZE -1) || (strcpy(ptr->prefix_key, key) == NULL)) { ptr->prefix_key_length= 0; return MEMCACHED_BAD_KEY_PROVIDED; } + else + { + ptr->prefix_key_length= key_length; + } } else {