mirror of
https://github.com/NLnetLabs/unbound.git
synced 2025-12-20 23:00:56 -05:00
libworker.
git-svn-id: file:///svn/unbound/trunk@808 be551aaa-1e26-0410-a405-d3ace91eadb9
This commit is contained in:
parent
a2311972e7
commit
9374c1465a
4 changed files with 242 additions and 0 deletions
|
|
@ -1,5 +1,6 @@
|
||||||
5 December 2007: Wouter
|
5 December 2007: Wouter
|
||||||
- locking in context_new() inside the function.
|
- locking in context_new() inside the function.
|
||||||
|
- setup of libworker.
|
||||||
|
|
||||||
4 December 2007: Wouter
|
4 December 2007: Wouter
|
||||||
- minor Makefile fixup.
|
- minor Makefile fixup.
|
||||||
|
|
|
||||||
|
|
@ -45,6 +45,7 @@
|
||||||
#include "util/rbtree.h"
|
#include "util/rbtree.h"
|
||||||
#include "services/modstack.h"
|
#include "services/modstack.h"
|
||||||
#include "libunbound/unbound.h"
|
#include "libunbound/unbound.h"
|
||||||
|
#include "util/data/packed_rrset.h"
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* The context structure
|
* The context structure
|
||||||
|
|
@ -128,6 +129,13 @@ struct ctx_query {
|
||||||
/** for async query, the callback user arg */
|
/** for async query, the callback user arg */
|
||||||
void* cb_arg;
|
void* cb_arg;
|
||||||
|
|
||||||
|
/** answer message, result from resolver lookup. */
|
||||||
|
uint8_t* msg;
|
||||||
|
/** resulting message length. */
|
||||||
|
size_t msg_len;
|
||||||
|
/** validation status on security */
|
||||||
|
enum sec_status msg_security;
|
||||||
|
|
||||||
/** result structure, also contains original query, type, class.
|
/** result structure, also contains original query, type, class.
|
||||||
* malloced ptr ready to hand to the client. */
|
* malloced ptr ready to hand to the client. */
|
||||||
struct ub_val_result* res;
|
struct ub_val_result* res;
|
||||||
|
|
|
||||||
133
libunbound/worker.c
Normal file
133
libunbound/worker.c
Normal file
|
|
@ -0,0 +1,133 @@
|
||||||
|
/*
|
||||||
|
* libunbound/worker.c - worker thread or process that resolves
|
||||||
|
*
|
||||||
|
* Copyright (c) 2007, NLnet Labs. All rights reserved.
|
||||||
|
*
|
||||||
|
* This software is open source.
|
||||||
|
*
|
||||||
|
* Redistribution and use in source and binary forms, with or without
|
||||||
|
* modification, are permitted provided that the following conditions
|
||||||
|
* are met:
|
||||||
|
*
|
||||||
|
* Redistributions of source code must retain the above copyright notice,
|
||||||
|
* this list of conditions and the following disclaimer.
|
||||||
|
*
|
||||||
|
* Redistributions in binary form must reproduce the above copyright notice,
|
||||||
|
* this list of conditions and the following disclaimer in the documentation
|
||||||
|
* and/or other materials provided with the distribution.
|
||||||
|
*
|
||||||
|
* Neither the name of the NLNET LABS nor the names of its contributors may
|
||||||
|
* be used to endorse or promote products derived from this software without
|
||||||
|
* specific prior written permission.
|
||||||
|
*
|
||||||
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||||
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
|
||||||
|
* TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
||||||
|
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE
|
||||||
|
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||||
|
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||||
|
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||||
|
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||||
|
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||||
|
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||||
|
* POSSIBILITY OF SUCH DAMAGE.
|
||||||
|
*/
|
||||||
|
|
||||||
|
/**
|
||||||
|
* \file
|
||||||
|
*
|
||||||
|
* This file contains the worker process or thread that performs
|
||||||
|
* the DNS resolving and validation. The worker is called by a procedure
|
||||||
|
* and if in the background continues until exit, if in the foreground
|
||||||
|
* returns from the procedure when done.
|
||||||
|
*/
|
||||||
|
#include "config.h"
|
||||||
|
#include "libunbound/worker.h"
|
||||||
|
#include "libunbound/context.h"
|
||||||
|
#include "libunbound/unbound.h"
|
||||||
|
|
||||||
|
/** delete libworker struct */
|
||||||
|
static void
|
||||||
|
libworker_delete(struct libworker* w)
|
||||||
|
{
|
||||||
|
if(!w) return;
|
||||||
|
lock_basic_lock(&w->ctx->cfglock);
|
||||||
|
/* return alloc */
|
||||||
|
lock_basic_unlock(&w->ctx->cfglock);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
/** setup fresh libworker struct */
|
||||||
|
static struct libworker*
|
||||||
|
libworker_setup(struct ub_val_ctx* ctx)
|
||||||
|
{
|
||||||
|
struct libworker* w = (struct libworker*)calloc(1, sizeof(*w));
|
||||||
|
if(!w) return NULL;
|
||||||
|
w->ctx = ctx;
|
||||||
|
lock_basic_lock(&ctx->cfglock);
|
||||||
|
/* obtain a new alloc */
|
||||||
|
lock_basic_unlock(&ctx->cfglock);
|
||||||
|
/* setup event base */
|
||||||
|
/* setup outnet */
|
||||||
|
/* setup rnd */
|
||||||
|
/* setup env */
|
||||||
|
/* setup env.scratch */
|
||||||
|
/* setup env.scratch_buffer */
|
||||||
|
/* setup env.worker */
|
||||||
|
/* setup env.mesh */
|
||||||
|
/* setup env.alloc */
|
||||||
|
/* setup env.rnd */
|
||||||
|
/* setup env - callback ptrs */
|
||||||
|
return w;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** the background thread func */
|
||||||
|
static void*
|
||||||
|
libworker_dobg(void* arg)
|
||||||
|
{
|
||||||
|
struct ub_val_ctx* ctx = (struct ub_val_ctx*)arg;
|
||||||
|
struct libworker* w = libworker_setup(ctx);
|
||||||
|
log_thread_set(&w->thread_num);
|
||||||
|
/* TODO do bg thread */
|
||||||
|
libworker_delete(w);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
int libworker_bg(struct ub_val_ctx* ctx)
|
||||||
|
{
|
||||||
|
/* fork or threadcreate */
|
||||||
|
lock_basic_lock(&ctx->cfglock);
|
||||||
|
if(ctx->dothread) {
|
||||||
|
ub_thread_t t;
|
||||||
|
lock_basic_unlock(&ctx->cfglock);
|
||||||
|
ub_thread_create(&t, libworker_dobg, ctx);
|
||||||
|
/* ctx.tid = t or so */
|
||||||
|
} else {
|
||||||
|
pid_t p;
|
||||||
|
lock_basic_unlock(&ctx->cfglock);
|
||||||
|
switch((p=fork())) {
|
||||||
|
case 0:
|
||||||
|
(void)libworker_dobg(ctx);
|
||||||
|
exit(0);
|
||||||
|
break;
|
||||||
|
case -1:
|
||||||
|
/* TODO make UB_FORKFAILED */
|
||||||
|
return UB_SOCKET;
|
||||||
|
default:
|
||||||
|
/* ctx.pid = p or so */
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return UB_NOERROR;
|
||||||
|
}
|
||||||
|
|
||||||
|
int libworker_fg(struct ub_val_ctx* ctx, struct ctx_query* q)
|
||||||
|
{
|
||||||
|
struct libworker* w = libworker_setup(ctx);
|
||||||
|
if(!w)
|
||||||
|
return UB_NOMEM;
|
||||||
|
/* TODO */
|
||||||
|
q->res->bogus = 1;
|
||||||
|
libworker_delete(w);
|
||||||
|
return UB_NOERROR;
|
||||||
|
}
|
||||||
100
libunbound/worker.h
Normal file
100
libunbound/worker.h
Normal file
|
|
@ -0,0 +1,100 @@
|
||||||
|
/*
|
||||||
|
* libunbound/worker.h - worker thread or process that resolves
|
||||||
|
*
|
||||||
|
* Copyright (c) 2007, NLnet Labs. All rights reserved.
|
||||||
|
*
|
||||||
|
* This software is open source.
|
||||||
|
*
|
||||||
|
* Redistribution and use in source and binary forms, with or without
|
||||||
|
* modification, are permitted provided that the following conditions
|
||||||
|
* are met:
|
||||||
|
*
|
||||||
|
* Redistributions of source code must retain the above copyright notice,
|
||||||
|
* this list of conditions and the following disclaimer.
|
||||||
|
*
|
||||||
|
* Redistributions in binary form must reproduce the above copyright notice,
|
||||||
|
* this list of conditions and the following disclaimer in the documentation
|
||||||
|
* and/or other materials provided with the distribution.
|
||||||
|
*
|
||||||
|
* Neither the name of the NLNET LABS nor the names of its contributors may
|
||||||
|
* be used to endorse or promote products derived from this software without
|
||||||
|
* specific prior written permission.
|
||||||
|
*
|
||||||
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||||
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
|
||||||
|
* TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
||||||
|
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE
|
||||||
|
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||||
|
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||||
|
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||||
|
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||||
|
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||||
|
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||||
|
* POSSIBILITY OF SUCH DAMAGE.
|
||||||
|
*/
|
||||||
|
|
||||||
|
/**
|
||||||
|
* \file
|
||||||
|
*
|
||||||
|
* This file contains the worker process or thread that performs
|
||||||
|
* the DNS resolving and validation. The worker is called by a procedure
|
||||||
|
* and if in the background continues until exit, if in the foreground
|
||||||
|
* returns from the procedure when done.
|
||||||
|
*/
|
||||||
|
#ifndef LIBUNBOUND_WORKER_H
|
||||||
|
#define LIBUNBOUND_WORKER_H
|
||||||
|
struct ub_val_ctx;
|
||||||
|
struct module_env;
|
||||||
|
struct comm_base;
|
||||||
|
struct outside_network;
|
||||||
|
struct ub_randstate;
|
||||||
|
struct ctx_query;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* The library-worker status structure
|
||||||
|
* Internal to the worker.
|
||||||
|
*/
|
||||||
|
struct libworker {
|
||||||
|
/** every worker has a unique thread_num. (first in struct) */
|
||||||
|
int thread_num;
|
||||||
|
/** context we are operating under */
|
||||||
|
struct ub_val_ctx* ctx;
|
||||||
|
|
||||||
|
/** is this a background worker? */
|
||||||
|
int is_bg;
|
||||||
|
|
||||||
|
/** copy of the module environment with worker local entries. */
|
||||||
|
struct module_env* env;
|
||||||
|
/** the event base this worker works with */
|
||||||
|
struct comm_base* base;
|
||||||
|
/** the backside outside network interface to the auth servers */
|
||||||
|
struct outside_network* back;
|
||||||
|
|
||||||
|
/** random() table for this worker. */
|
||||||
|
struct ub_randstate* rndstate;
|
||||||
|
/** do we need to exit (when done) */
|
||||||
|
int need_to_exit;
|
||||||
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Create a background worker
|
||||||
|
* @param ctx: is updated with pid/tid of the background worker.
|
||||||
|
* a new allocation cache is obtained from ctx. It contains the
|
||||||
|
* threadnumber and unique id for further (shared) cache insertions.
|
||||||
|
* @return 0 if OK, else error.
|
||||||
|
* Further communication is done via the pipes in ctx.
|
||||||
|
*/
|
||||||
|
int libworker_bg(struct ub_val_ctx* ctx);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Create a foreground worker.
|
||||||
|
* This worker will join the threadpool of resolver threads.
|
||||||
|
* It exits when the query answer has been obtained (or error).
|
||||||
|
* This routine blocks until the worker is finished.
|
||||||
|
* @param ctx: new allocation cache obtained and returned to it.
|
||||||
|
* @param q: query (result is stored in here).
|
||||||
|
* @return 0 if finished OK, else error.
|
||||||
|
*/
|
||||||
|
int libworker_fg(struct ub_val_ctx* ctx, struct ctx_query* q);
|
||||||
|
|
||||||
|
#endif /* LIBUNBOUND_WORKER_H */
|
||||||
Loading…
Reference in a new issue