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
00530
00531 if (p->obj->status != KResolver::Canceled
00532 && p->worker)
00533 p->worker->postprocess();
00534
00535
00536
00537 KResolverResults& r = p->obj->results;
00538 if (p->worker)
00539 {
00540 r = p->worker->results;
00541
00542 r.setAddress(p->input->node, p->input->service);
00543
00544
00545
00546
00547 if (p->obj->status != KResolver::Canceled)
00548 {
00549 p->obj->errorcode = r.error();
00550 p->obj->syserror = r.systemError();
00551 p->obj->status = !r.isEmpty() ?
00552 KResolver::Success : KResolver::Failed;
00553 }
00554 else
00555 {
00556 p->obj->status = KResolver::Canceled;
00557 p->obj->errorcode = KResolver::Canceled;
00558 p->obj->syserror = 0;
00559 }
00560 }
00561 else
00562 {
00563 r.empty();
00564 r.setError(p->obj->errorcode, p->obj->syserror);
00565 }
00566
00567
00568 if (!p->obj->waiting && parent)
00569
00570
00571
00572 QApplication::postEvent(parent, new QEvent((QEvent::Type)(ResolutionCompleted)));
00573
00574
00575 p->obj->mutex.unlock();
00576
00577 if (parent == 0L)
00578
00579
00580 delete p->obj;
00581 }
00582 else
00583 {
00584
00585 if (p->worker)
00586 p->worker->postprocess();
00587 }
00588
00589 delete p->worker;
00590
00591
00592
00593
00594 delete p;
00595
00596
00597 notifyWaiters.wakeAll();
00598 }
00599
00600
00601
00602
00603 void KResolverManager::enqueue(KResolver *obj, RequestData *requestor)
00604 {
00605 RequestData *newrequest = new RequestData;
00606 newrequest->nRequests = 0;
00607 newrequest->obj = obj->d;
00608 newrequest->input = &obj->d->input;
00609 newrequest->requestor = requestor;
00610
00611
00612
00613 if ((newrequest->worker = findWorker(obj->d)) == 0L)
00614 {
00615
00616
00617 obj->d->status = KResolver::Failed;
00618 obj->d->errorcode = KResolver::UnsupportedFamily;
00619 obj->d->syserror = 0;
00620
00621 doNotifying(newrequest);
00622 return;
00623 }
00624
00625
00626
00627 if (requestor)
00628 requestor->nRequests++;
00629
00630 if (!newrequest->worker->m_finished)
00631 dispatch(newrequest);
00632 else if (newrequest->nRequests > 0)
00633 {
00634 mutex.lock();
00635 currentRequests.append(newrequest);
00636 mutex.unlock();
00637 }
00638 else
00639
00640 doNotifying(newrequest);
00641 }
00642
00643
00644
00645 void KResolverManager::dispatch(RequestData *data)
00646 {
00647
00648
00649
00650
00651 QMutexLocker locker(&mutex);
00652
00653
00654 newRequests.append(data);
00655
00656
00657
00658
00659
00660
00661
00662
00663
00664
00665
00666
00667
00668
00669
00670
00671
00672
00673
00674
00675
00676
00677
00678
00679
00680
00681
00682 if (availableThreads == 0 && runningThreads < maxThreads)
00683 {
00684
00685
00686
00687 KResolverThread *th = workers.first();
00688 while (th && th->running())
00689 th = workers.next();
00690
00691 if (th == 0L)
00692
00693 th = new KResolverThread;
00694 else
00695 workers.take();
00696
00697 th->start();
00698 workers.append(th);
00699 runningThreads++;
00700 }
00701
00702 feedWorkers.wakeAll();
00703
00704
00705 workers.first();
00706 while (workers.current())
00707 {
00708 if (!workers.current()->running())
00709 workers.remove();
00710 else
00711 workers.next();
00712 }
00713 }
00714
00715
00716
00717 bool KResolverManager::dequeueNew(KResolver* obj)
00718 {
00719
00720
00721
00722
00723
00724 if (!obj->isRunning())
00725 {
00726 KResolverPrivate *d = obj->d;
00727
00728
00729 RequestData *curr = newRequests.first();
00730 while (curr)
00731 if (curr->obj == d)
00732 {
00733
00734
00735 d->status = KResolver::Canceled;
00736 newRequests.take();
00737 doNotifying(curr);
00738
00739 return true;
00740 }
00741 else
00742 curr = currentRequests.next();
00743
00744
00745 return true;
00746 }
00747
00748
00749
00750 obj->d->mutex.lock();
00751 obj->d->status = KResolver::Canceled;
00752 obj->d->mutex.unlock();
00753 return false;
00754 }
00755
00756
00757
00758 void KResolverManager::dequeue(KResolver *obj)
00759 {
00760 QMutexLocker locker(&mutex);
00761 dequeueNew(obj);
00762 }
00763
00764
00765
00766 void KResolverManager::aboutToBeDeleted(KResolver *obj)
00767 {
00768
00769
00770
00771
00772 QMutexLocker locker(&mutex);
00773 KResolverPrivate* d = obj->d;
00774
00775 if (!dequeueNew(obj))
00776 {
00777
00778
00779 RequestData *curr = currentRequests.first();
00780 while (curr)
00781 if (curr->obj == d)
00782 {
00783
00784
00785 d->parent = 0L;
00786 d->emitSignal = false;
00787 d->status = KResolver::Canceled;
00788
00789 return;
00790 }
00791 else
00792 curr = currentRequests.next();
00793
00794
00795 qWarning("KResolverManager::aboutToBeDeleted: pid = %u, obj = %p: could not find obj in list",
00796 pid, (void*)obj);
00797 }
00798
00799
00800
00801 delete d;
00802 }
00803
00804 }
This file is part of the documentation for kdecore Library Version 3.3.0.