test/smtp_buffer_test.cpp

   1
   2 #include <ioevent>
   3
   4 #include <EvBufferEvent.hpp>
   5
   6 #include <evhttp.h>
   7
   8 #include <iostream>
   9
  10 enum {
  11         STATE_INIT = 0,
  12         STATE_DATA
  13 };
  14
  15 struct cc {
  16         int state;
  17
  18         EvBufferEvent * be;
  19         struct bufferevent * bev;
  20
  21         const char * cur_line;
  22
  23 };
  24
  25 void react_helo(struct cc * c)
  26 {
  27         bufferevent * bev = c->bev;
  28         evbuffer * b = ::evbuffer_new();
  29
  30         const char * line = c->cur_line;
  31
  32         ::evbuffer_add_printf(b, "250 Hello %s\r\n", line + 5);
  33         ::bufferevent_write_buffer(bev, b);
  34
  35         ::evbuffer_free(b);
  36 }
  37
  38 void react_250_ok(struct cc * c)
  39 {
  40         bufferevent * bev = c->bev;
  41         evbuffer * b = ::evbuffer_new();
  42
  43         ::evbuffer_add_printf(b, "250 Ok\r\n");
  44         ::bufferevent_write_buffer(bev, b);
  45
  46         ::evbuffer_free(b);
  47 }
  48
  49 void react_data(struct cc * c)
  50 {
  51         bufferevent * bev = c->bev;
  52         evbuffer * b = ::evbuffer_new();
  53
  54         ::evbuffer_add_printf(b, "354 End data with <CR><LF>.<CR><LF>\r\n");
  55         ::bufferevent_write_buffer(bev, b);
  56
  57         c->state = STATE_DATA;
  58
  59         ::evbuffer_free(b);
  60 }
  61
  62 void react_data_line(struct cc * c)
  63 {
  64         bufferevent * bev = c->bev;
  65         evbuffer * b = ::evbuffer_new();
  66
  67         ::evbuffer_add_printf(b, "250 Ok\r\n");
  68         ::bufferevent_write_buffer(bev, b);
  69
  70         c->state = STATE_INIT;
  71
  72         ::evbuffer_free(b);
  73 }
  74
  75 void react_bye(struct cc * c)
  76 {
  77         bufferevent * bev = c->bev;
  78         evbuffer * b = ::evbuffer_new();
  79
  80         ::evbuffer_add_printf(b, "221 Bye\r\n");
  81         ::bufferevent_write_buffer(bev, b);
  82
  83         ::evbuffer_free(b);
  84 }
  85
  86 void readcb(struct bufferevent *bev, void * v)
  87 {
  88         char * line;
  89
  90         struct cc * c = (struct cc *)v;
  91
  92         while (line = ::evbuffer_readline(bev->input))
  93         {
  94                 c->cur_line = line;
  95
  96                 if (c->state == STATE_INIT)
  97                 {
  98                         if (!strncmp(line, "HELO ", 5) ||
  99                             !strncmp(line, "EHLO ", 5))
 100                         {
 101                                 react_helo(c);
 102                         }
 103                         else
 104                         if (!strncmp(line, "MAIL", 4))
 105                         {
 106                                 react_250_ok(c);
 107                         }
 108                         else
 109                         if (!strncmp(line, "RCPT", 4))
 110                         {
 111                                 react_250_ok(c);
 112                         }
 113                         else
 114                         if (!strncmp(line, "DATA", 4))
 115                         {
 116                                 react_data(c);
 117                         }
 118                         else
 119                         if (!strncmp(line, "QUIT", 4))
 120                         {
 121                                 react_bye(c);
 122                         }
 123                 }
 124                 else
 125                 {
 126                         if (strcmp(line, "."))
 127                         {
 128                                 react_data_line(c);
 129                         }
 130                 }
 131
 132                 free(line);
 133                 c->cur_line = NULL;
 134         }
 135
 136 }
 137
 138 void writecb(struct bufferevent *, void *)
 139 {
 140
 141 }
 142
 143 void errorcb(struct bufferevent *, short, void *)
 144 {
 145
 146 }
 147
 148 void accept_handler(int s1, short, void *)
 149 {
 150         struct cc * c = new cc;
 151         c->state = STATE_INIT;
 152
 153         EvBufferEvent * be = new EvBufferEvent(readcb, writecb, errorcb, c);
 154
 155         be->accept(s1);
 156
 157         c->be = be;
 158         c->bev = be->_bufferevent;
 159
 160         c->cur_line = NULL;
 161
 162         evbuffer * b = ::evbuffer_new();
 163
 164         ::bufferevent_enable(be->_bufferevent, EV_READ);
 165
 166         be->write("220 www.example.com ESMTP postfix\r\n", 35);
 167
 168 }
 169
 170 main()
 171 {
 172         ::event_init();
 173
 174         int sock = ::bind_socket(NULL, 5525);
 175
 176         ::bufev_socket_listen(sock, 10);
 177
 178         ioevent accept_event(sock, EV_READ, &accept_handler, NULL);
 179         ::event_add(&accept_event, NULL);
 180
 181         ::event_dispatch();
 182
 183 }