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 }