diff options
Diffstat (limited to 'src/tests/rot13.cpp')
| -rw-r--r-- | src/tests/rot13.cpp | 80 |
1 files changed, 80 insertions, 0 deletions
diff --git a/src/tests/rot13.cpp b/src/tests/rot13.cpp new file mode 100644 index 0000000..e677440 --- /dev/null +++ b/src/tests/rot13.cpp | |||
| @@ -0,0 +1,80 @@ | |||
| 1 | #include "bu/protocol.h" | ||
| 2 | #include "bu/multiserver.h" | ||
| 3 | #include "bu/client.h" | ||
| 4 | #include "bu/filter.h" | ||
| 5 | |||
| 6 | using namespace Bu; | ||
| 7 | |||
| 8 | class Rot13Filter : public Filter | ||
| 9 | { | ||
| 10 | public: | ||
| 11 | Rot13Filter( Stream &next ) : | ||
| 12 | Filter( next ) | ||
| 13 | { | ||
| 14 | } | ||
| 15 | |||
| 16 | virtual ~Rot13Filter() | ||
| 17 | { | ||
| 18 | } | ||
| 19 | |||
| 20 | virtual void start() | ||
| 21 | { | ||
| 22 | } | ||
| 23 | |||
| 24 | virtual size_t stop() | ||
| 25 | { | ||
| 26 | return 0; | ||
| 27 | } | ||
| 28 | |||
| 29 | virtual size_t read( void *pBuf, size_t nBytes ) | ||
| 30 | { | ||
| 31 | return rNext.read( pBuf, nBytes ); | ||
| 32 | } | ||
| 33 | |||
| 34 | virtual size_t write( const void *pBuf, size_t nBytes ) | ||
| 35 | { | ||
| 36 | const char *cBuf = (const char *)pBuf; | ||
| 37 | char *buf = new char[nBytes]; | ||
| 38 | for( size_t j = 0; j < nBytes; j++ ) | ||
| 39 | { | ||
| 40 | if( cBuf[j] >= 'a' && cBuf[j] <= 'z' ) | ||
| 41 | buf[j] = (cBuf[j]-'a'+13)%26+'a'; | ||
| 42 | else if( cBuf[j] >= 'A' && cBuf[j] <= 'Z' ) | ||
| 43 | buf[j] = (cBuf[j]-'A'+13)%26+'A'; | ||
| 44 | else | ||
| 45 | buf[j] = cBuf[j]; | ||
| 46 | } | ||
| 47 | rNext.write( buf, nBytes ); | ||
| 48 | delete[] buf; | ||
| 49 | return nBytes; | ||
| 50 | } | ||
| 51 | }; | ||
| 52 | |||
| 53 | class Rot13Protocol : public Protocol | ||
| 54 | { | ||
| 55 | public: | ||
| 56 | void onNewConnection( Bu::Client *pClient ) | ||
| 57 | { | ||
| 58 | pClient->pushFilter<Rot13Filter>(); | ||
| 59 | } | ||
| 60 | |||
| 61 | void onNewData( Bu::Client *pClient ) | ||
| 62 | { | ||
| 63 | pClient->write( pClient->getInput().getStr(), pClient->getInputSize() ); | ||
| 64 | pClient->seek( pClient->getInputSize() ); | ||
| 65 | } | ||
| 66 | }; | ||
| 67 | |||
| 68 | int main() | ||
| 69 | { | ||
| 70 | MultiServer xSrv; | ||
| 71 | |||
| 72 | xSrv.setTimeout( 5 ); | ||
| 73 | xSrv.addProtocol( genProtocol<Rot13Protocol>, 9999 ); | ||
| 74 | |||
| 75 | for(;;) | ||
| 76 | xSrv.scan(); | ||
| 77 | |||
| 78 | return 0; | ||
| 79 | } | ||
| 80 | |||
