pdns/ws-auth.hh

   1 /*
   2     PowerDNS Versatile Database Driven Nameserver
   3     Copyright (C) 2002 - 2009  PowerDNS.COM BV
   4
   5     This program is free software; you can redistribute it and/or modify
   6     it under the terms of the GNU General Public License version 2
   7     as published by the Free Software Foundation
   8
   9     Additionally, the license of this program contains a special
  10     exception which allows to distribute the program in binary form when
  11     it is linked against OpenSSL.
  12
  13     This program is distributed in the hope that it will be useful,
  14     but WITHOUT ANY WARRANTY; without even the implied warranty of
  15     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16     GNU General Public License for more details.
  17
  18     You should have received a copy of the GNU General Public License
  19     along with this program; if not, write to the Free Software
  20     Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
  21 */
  22 #ifndef WS_HH
  23 #define WS_HH
  24 #include <string>
  25 #include <map>
  26 #include <time.h>
  27 #include <pthread.h>
  28 #include "misc.hh"
  29 #include "namespaces.hh"
  30
  31 class Ewma
  32 {
  33 public:
  34   Ewma() : d_last(0), d_10(0), d_5(0), d_1(0), d_max(0){dt.set();}
  35   void submit(int val)
  36   {
  37     int rate=val-d_last;
  38     double difft=dt.udiff()/1000000.0;
  39     dt.set();
  40
  41     d_10=((600.0-difft)*d_10+(difft*rate))/600.0;
  42     d_5=((300.0-difft)*d_5+(difft*rate))/300.0;
  43     d_1=((60.0-difft)*d_1+(difft*rate))/60.0;
  44     d_max=max(d_1,d_max);
  45
  46     d_last=val;
  47   }
  48   double get10()
  49   {
  50     return d_10;
  51   }
  52   double get5()
  53   {
  54     return d_5;
  55   }
  56   double get1()
  57   {
  58     return d_1;
  59   }
  60   double getMax()
  61   {
  62     return d_max;
  63   }
  64 private:
  65   DTime dt;
  66   int d_last;
  67   double d_10, d_5, d_1, d_max;
  68 };
  69
  70 class WebServer;
  71 class HttpRequest;
  72 class HttpResponse;
  73
  74 class AuthWebServer
  75 {
  76 public:
  77   AuthWebServer();
  78   void go();
  79   static string makePercentage(const double& val);
  80
  81 private:
  82   static void *webThreadHelper(void *);
  83   static void *statThreadHelper(void *p);
  84   void indexfunction(HttpRequest* req, HttpResponse* resp);
  85   void cssfunction(HttpRequest* req, HttpResponse* resp);
  86   void jsonstat(HttpRequest* req, HttpResponse* resp);
  87   void registerApiHandler(const string& url, boost::function<void(HttpRequest*, HttpResponse*)> handler);
  88   void printvars(ostringstream &ret);
  89   void printargs(ostringstream &ret);
  90   void webThread();
  91   void statThread();
  92   pthread_t d_tid;
  93
  94   time_t d_start;
  95   double d_min10, d_min5, d_min1;
  96   Ewma d_queries, d_cachehits, d_cachemisses;
  97   Ewma d_qcachehits, d_qcachemisses;
  98   WebServer *d_ws;
  99 };
 100
 101 #endif