kresolvermanager.cpp
00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025 #include "config.h"
00026
00027 #include <sys/types.h>
00028 #include <netinet/in.h>
00029 #include <limits.h>
00030 #include <unistd.h>
00031
00032 #ifdef HAVE_RES_INIT
00033 # include <sys/stat.h>
00034 # include <resolv.h>
00035 #endif
00036
00037 #include <qapplication.h>
00038 #include <qstring.h>
00039 #include <qcstring.h>
00040 #include <qptrlist.h>
00041 #include <qtimer.h>
00042 #include <qmutex.h>
00043 #include <qthread.h>
00044 #include <qwaitcondition.h>
00045 #include <qsemaphore.h>
00046
00047 #include "kresolver.h"
00048 #include "kresolver_p.h"
00049 #include "kresolverworkerbase.h"
00050 #include "kresolverstandardworkers_p.h"
00051
00052 using namespace KNetwork;
00053 using namespace KNetwork::Internal;
00054
00055
00056
00057
00058
00059
00060
00061
00062
00063
00064
00065
00066
00067
00068
00069
00070
00071
00072
00073
00074
00075
00076
00077
00078
00079
00080
00081
00082
00083
00084
00085
00086
00087
00088
00089
00090
00091
00092
00093
00094
00095
00096
00097
00098
00099
00100
00101 namespace
00102 {
00103
00104
00105
00106
00107
00108
00109
00110
00111
00112
00113
00114 class ResInitUsage
00115 {
00116 #ifdef HAVE_RES_INIT
00117 time_t mTime;
00118 QWaitCondition cond;
00119 QMutex mutex;
00120 int useCount;
00121
00122 bool shouldResInit()
00123 {
00124
00125 struct stat st;
00126 if (stat("/etc/resolv.conf", &st) != 0)
00127 return false;
00128
00129 if (mTime < st.st_mtime)
00130 {
00131
00132 return true;
00133 }
00134 return false;
00135 }
00136
00137 void reResInit()
00138 {
00139
00140 res_init();
00141
00142 struct stat st;
00143 if (stat("/etc/resolv.conf", &st) == 0)
00144 mTime = st.st_mtime;
00145 }
00146
00147 public:
00148 ResInitUsage()
00149 : mTime(0), useCount(0)
00150 { }
00151
00152
00153
00154
00155 void operator--(int)
00156 {
00157 mutex.lock();
00158 if (--useCount == 0)
00159
00160 cond.wakeAll();
00161 mutex.unlock();
00162 }
00163
00164
00165
00166
00167 void operator++(int)
00168 {
00169 mutex.lock();
00170
00171 if (shouldResInit())
00172 {
00173 if (useCount)
00174 {
00175
00176
00177
00178 cond.wait(&mutex);
00179 }
00180 reResInit();
00181 }
00182 useCount++;
00183 mutex.unlock();
00184 }
00185
00186 #else
00187 public:
00188 ResInitUsage()
00189 { }
00190
00191 void operator--(int)
00192 { }
00193
00194 void operator++(int)
00195 { }
00196 #endif
00197
00198 } resInit;
00199
00200
00201
00202
00203
00204
00205
00206 static const int maxThreadWaitTime = 20000;
00207 static const int maxThreads = 5;
00208
00209 static pid_t pid;
00210
00211 KResolverThread::KResolverThread()
00212 : data(0L)
00213 {
00214 }
00215
00216
00217 void KResolverThread::run()
00218 {
00219
00220
00221
00222
00223 KResolverManager::manager()->registerThread(this);
00224 while (true)
00225 {
00226 data = KResolverManager::manager()->requestData(this, ::maxThreadWaitTime);
00227
00228
00229 if (data)
00230 {
00231
00232
00233
00234
00235 ;
00236
00237
00238 data->worker->run();
00239
00240
00241 KResolverManager::manager()->releaseData(this, data);
00242
00243
00244 }
00245 else
00246 break;
00247 }
00248
00249 KResolverManager::manager()->unregisterThread(this);
00250
00251 }
00252
00253 static KResolverManager *globalManager;
00254
00255 KResolverManager* KResolverManager::manager()
00256 {
00257 if (globalManager == 0L)
00258 new KResolverManager();
00259 return globalManager;
00260 }
00261
00262 KResolverManager::KResolverManager()
00263 : runningThreads(0), availableThreads(0)
00264 {
00265 globalManager = this;
00266 workers.setAutoDelete(true);
00267 currentRequests.setAutoDelete(true);
00268 initStandardWorkers();
00269
00270 pid = getpid();
00271 }
00272
00273 KResolverManager::~KResolverManager()
00274 {
00275
00276
00277
00278 for (workers.first(); workers.current(); workers.next())
00279 workers.current()->terminate();
00280 }
00281
00282 void KResolverManager::registerThread(KResolverThread* )
00283 {
00284 }
00285
00286 void KResolverManager::unregisterThread(KResolverThread*)
00287 {
00288 }
00289
00290
00291 RequestData* KResolverManager::requestData(KResolverThread *th, int maxWaitTime)
00292 {
00294
00296
00297 resInit++;
00298
00299
00300
00301 QMutexLocker locker(&mutex);
00302 RequestData *data = findData(th);
00303
00304 if (data)
00305
00306 return data;
00307
00308
00309 availableThreads++;
00310 feedWorkers.wait(&mutex, maxWaitTime);
00311 availableThreads--;
00312
00313 data = findData(th);
00314 if (data == 0L)
00315 {
00316
00317 runningThreads--;
00318 resInit--;
00319 }
00320 return data;
00321 }
00322
00323 RequestData* KResolverManager::findData(KResolverThread* th)
00324 {
00326
00327
00329
00330
00331 for (RequestData *curr = newRequests.first(); curr; curr = newRequests.next())
00332 if (!curr->worker->m_finished)
00333 {
00334
00335 if (curr->obj)
00336 curr->obj->status = KResolver::InProgress;
00337 curr->worker->th = th;
00338
00339
00340 currentRequests.append(newRequests.take());
00341
00342 return curr;
00343 }
00344
00345
00346 return 0L;
00347 }
00348
00349
00350 void KResolverManager::releaseData(KResolverThread *, RequestData* data)
00351 {
00353
00355
00356 resInit--;
00357
00358
00359
00360
00361 if (data->obj)
00362 {
00363 if (data->nRequests > 0)
00364
00365
00366 data->obj->status = KResolver::PostProcessing;
00367 else
00368
00369 data->obj->status = data->worker->results.isEmpty() ? KResolver::Failed : KResolver::Success;
00370 }
00371
00372 data->worker->m_finished = true;
00373 data->worker->th = 0L;
00374
00375
00376 handleFinished();
00377 }
00378
00379
00380 void KResolverManager::handleFinished()
00381 {
00382 bool redo = false;
00383 QPtrQueue<RequestData> doneRequests;
00384
00385 mutex.lock();
00386
00387
00388
00389
00390 RequestData *curr = currentRequests.last();
00391 while (curr)
00392 {
00393 if (curr->worker->th == 0L)
00394 {
00395 if (handleFinishedItem(curr))
00396 {
00397 doneRequests.enqueue(currentRequests.take());
00398 if (curr->requestor &&
00399 curr->requestor->nRequests == 0 &&
00400 curr->requestor->worker->m_finished)
00401
00402 redo = true;
00403 }
00404 }
00405
00406 curr = currentRequests.prev();
00407 }
00408
00409
00410 while (RequestData *d = doneRequests.dequeue())
00411 doNotifying(d);
00412
00413 mutex.unlock();
00414
00415 if (redo)
00416 {
00417
00418
00419 handleFinished();
00420 }
00421 }
00422
00423
00424 bool KResolverManager::handleFinishedItem(RequestData* curr)
00425
00426 {
00427
00428
00429
00430 if (curr->worker->m_finished && curr->nRequests == 0)
00431 {
00432
00433 if (curr->obj)
00434 curr->obj->status = KResolver::Success;
00435
00436 if (curr->requestor)
00437 --curr->requestor->nRequests;
00438
00439
00440
00441 return true;
00442 }
00443 return false;
00444 }
00445
00446
00447
00448 void KResolverManager::registerNewWorker(KResolverWorkerFactoryBase *factory)
00449 {
00450 workerFactories.append(factory);
00451 }
00452
00453 KResolverWorkerBase* KResolverManager::findWorker(KResolverPrivate* p)
00454 {
00456
00458
00459
00460
00461
00462
00463
00464
00465
00466
00467
00468 KResolverWorkerBase *worker;
00469 for (KResolverWorkerFactoryBase *factory = workerFactories.first(); factory;
00470 factory = workerFactories.next())
00471 {
00472 worker = factory->create();
00473
00474
00475 worker->input = &p->input;
00476
00477 if (worker->preprocess())
00478 {
00479
00480 if (worker->m_finished)
00481 p->status = !worker->results.isEmpty() ?
00482 KResolver::Success : KResolver::Failed;
00483 else
00484 p->status = KResolver::Queued;
00485 return worker;
00486 }
00487
00488
00489 delete worker;
00490 }
00491
00492
00493 return 0L;
00494 }
00495
00496 void KResolverManager::doNotifying(RequestData *p)
00497 {
00499
00500
00502
00503
00504
00505
00506
00507
00508
00509
00510
00511
00512
00513
00514
00515
00516
00517
00518
00519
00520
00521
00522
00523
00524 if (p->obj)
00525 {
00526
00527 p->obj->mutex.lock();
00528 KResolver* parent = p->obj->parent;
00529 KResolverResults& r = p->obj->results;
00530
00531 if (p->obj->status == KResolver::Canceled)
00532 {
00533 p->obj->status = KResolver::Canceled;
00534 p->obj->errorcode = KResolver::Canceled;
00535 p->obj->syserror = 0;
00536 r.setError(KResolver::Canceled, 0);
00537 }
00538 else if (p->worker)
00539 {
00540
00541 p->worker->postprocess();
00542
00543
00544
00545 r = p->worker->results;
00546
00547
00548 r.setAddress(p->input->node, p->input->service);
00549
00550
00551
00552
00553 p->obj->errorcode = r.error();
00554 p->obj->syserror = r.systemError();
00555 p->obj->status = !r.isEmpty() ?
00556 KResolver::Success : KResolver::Failed;
00557 }
00558 else
00559 {
00560 r.empty();
00561 r.setError(p->obj->errorcode, p->obj->syserror);
00562 }
00563
00564
00565 if (!p->obj->waiting && parent)
00566
00567
00568
00569 QApplication::postEvent(parent, new QEvent((QEvent::Type)(ResolutionCompleted)));
00570
00571
00572 p->obj->mutex.unlock();
00573 }
00574 else
00575 {
00576
00577 if (p->worker)
00578 p->worker->postprocess();
00579 }
00580
00581 delete p->worker;
00582
00583
00584
00585
00586 delete p;
00587
00588
00589 notifyWaiters.wakeAll();
00590 }
00591
00592
00593
00594
00595 void KResolverManager::enqueue(KResolver *obj, RequestData *requestor)
00596 {
00597 RequestData *newrequest = new RequestData;
00598 newrequest->nRequests = 0;
00599 newrequest->obj = obj->d;
00600 newrequest->input = &obj->d->input;
00601 newrequest->requestor = requestor;
00602
00603
00604
00605 if ((newrequest->worker = findWorker(obj->d)) == 0L)
00606 {
00607
00608
00609 obj->d->status = KResolver::Failed;
00610 obj->d->errorcode = KResolver::UnsupportedFamily;
00611 obj->d->syserror = 0;
00612
00613 doNotifying(newrequest);
00614 return;
00615 }
00616
00617
00618
00619 if (requestor)
00620 requestor->nRequests++;
00621
00622 if (!newrequest->worker->m_finished)
00623 dispatch(newrequest);
00624 else if (newrequest->nRequests > 0)
00625 {
00626 mutex.lock();
00627 currentRequests.append(newrequest);
00628 mutex.unlock();
00629 }
00630 else
00631
00632 doNotifying(newrequest);
00633 }
00634
00635
00636
00637 void KResolverManager::dispatch(RequestData *data)
00638 {
00639
00640
00641
00642
00643 QMutexLocker locker(&mutex);
00644
00645
00646 newRequests.append(data);
00647
00648
00649
00650
00651
00652
00653
00654
00655
00656
00657
00658
00659
00660
00661
00662
00663
00664
00665
00666
00667
00668
00669
00670
00671
00672
00673
00674 if (availableThreads == 0 && runningThreads < maxThreads)
00675 {
00676
00677
00678
00679 KResolverThread *th = workers.first();
00680 while (th && th->running())
00681 th = workers.next();
00682
00683 if (th == 0L)
00684
00685 th = new KResolverThread;
00686 else
00687 workers.take();
00688
00689 th->start();
00690 workers.append(th);
00691 runningThreads++;
00692 }
00693
00694 feedWorkers.wakeAll();
00695
00696
00697 workers.first();
00698 while (workers.current())
00699 {
00700 if (!workers.current()->running())
00701 workers.remove();
00702 else
00703 workers.next();
00704 }
00705 }
00706
00707
00708 bool KResolverManager::dequeueNew(KResolver* obj)
00709 {
00710
00711
00712
00713
00714 KResolverPrivate *d = obj->d;
00715
00716
00717 RequestData *curr = newRequests.first();
00718 while (curr)
00719 if (curr->obj == d)
00720 {
00721
00722
00723 d->status = KResolver::Canceled;
00724 d->errorcode = KResolver::Canceled;
00725 d->syserror = 0;
00726 newRequests.take();
00727
00728 delete curr->worker;
00729 delete curr;
00730
00731 return true;
00732 }
00733 else
00734 curr = newRequests.next();
00735
00736
00737 curr = currentRequests.first();
00738 while (curr)
00739 if (curr->obj == d)
00740 {
00741
00742
00743 d->mutex.lock();
00744
00745 d->status = KResolver::Canceled;
00746 d->errorcode = KResolver::Canceled;
00747 d->syserror = 0;
00748
00749
00750 curr->obj = 0L;
00751 curr->input = 0L;
00752 if (curr->worker)
00753 curr->worker->input = 0L;
00754
00755 d->mutex.unlock();
00756 }
00757 else
00758 curr = currentRequests.next();
00759
00760 return false;
00761 }
00762
00763
00764
00765 void KResolverManager::dequeue(KResolver *obj)
00766 {
00767 QMutexLocker locker(&mutex);
00768 dequeueNew(obj);
00769 }
00770
00771 }
This file is part of the documentation for kdecore Library Version 3.3.0.
Documentation copyright © 1996-2004 the KDE developers.
Generated on Thu Sep 23 17:11:37 2004 by
doxygen 1.3.8-20040913 written by
Dimitri van Heesch, © 1997-2003