test/parallel/test-tls-pause.js

   1 // Copyright Joyent, Inc. and other Node contributors.
   2 //
   3 // Permission is hereby granted, free of charge, to any person obtaining a
   4 // copy of this software and associated documentation files (the
   5 // "Software"), to deal in the Software without restriction, including
   6 // without limitation the rights to use, copy, modify, merge, publish,
   7 // distribute, sublicense, and/or sell copies of the Software, and to permit
   8 // persons to whom the Software is furnished to do so, subject to the
   9 // following conditions:
  10 //
  11 // The above copyright notice and this permission notice shall be included
  12 // in all copies or substantial portions of the Software.
  13 //
  14 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  15 // OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  16 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN
  17 // NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
  18 // DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
  19 // OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
  20 // USE OR OTHER DEALINGS IN THE SOFTWARE.
  21
  22 if (!process.versions.openssl) {
  23   console.error('Skipping because node compiled without OpenSSL.');
  24   process.exit(0);
  25 }
  26
  27 var common = require('../common');
  28 var assert = require('assert');
  29 var tls = require('tls');
  30 var fs = require('fs');
  31 var path = require('path');
  32
  33 var options = {
  34   key: fs.readFileSync(path.join(common.fixturesDir, 'test_key.pem')),
  35   cert: fs.readFileSync(path.join(common.fixturesDir, 'test_cert.pem'))
  36 };
  37
  38 var bufSize = 1024 * 1024;
  39 var sent = 0;
  40 var received = 0;
  41
  42 var server = tls.Server(options, function(socket) {
  43   socket.pipe(socket);
  44   socket.on('data', function(c) {
  45     console.error('data', c.length);
  46   });
  47 });
  48
  49 server.listen(common.PORT, function() {
  50   var resumed = false;
  51   var client = tls.connect({
  52     port: common.PORT,
  53     rejectUnauthorized: false
  54   }, function() {
  55     console.error('connected');
  56     client.pause();
  57     common.debug('paused');
  58     send();
  59     function send() {
  60       console.error('sending');
  61       var ret = client.write(new Buffer(bufSize));
  62       console.error('write => %j', ret);
  63       if (false !== ret) {
  64         console.error('write again');
  65         sent += bufSize;
  66         assert.ok(sent < 100 * 1024 * 1024); // max 100MB
  67         return process.nextTick(send);
  68       }
  69       sent += bufSize;
  70       common.debug('sent: ' + sent);
  71       resumed = true;
  72       client.resume();
  73       console.error('resumed', client);
  74     }
  75   });
  76   client.on('data', function(data) {
  77     console.error('data');
  78     assert.ok(resumed);
  79     received += data.length;
  80     console.error('received', received);
  81     console.error('sent', sent);
  82     if (received >= sent) {
  83       common.debug('received: ' + received);
  84       client.end();
  85       server.close();
  86     }
  87   });
  88 });
  89
  90 process.on('exit', function() {
  91   assert.equal(sent, received);
  92 });