add redis ip limiting
This commit is contained in:
		
							
								
								
									
										39
									
								
								app.js
									
									
									
									
									
								
							
							
						
						
									
										39
									
								
								app.js
									
									
									
									
									
								
							@@ -1,6 +1,7 @@
 | 
				
			|||||||
var express = require('express')
 | 
					var express = require('express')
 | 
				
			||||||
  , app = express.createServer()
 | 
					  , app = express.createServer()
 | 
				
			||||||
  , nodemailer = require('nodemailer')
 | 
					  , nodemailer = require('nodemailer')
 | 
				
			||||||
 | 
					  , redis = require('redis')
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// Express config
 | 
					// Express config
 | 
				
			||||||
app.set('views', __dirname + '/views');
 | 
					app.set('views', __dirname + '/views');
 | 
				
			||||||
@@ -27,16 +28,38 @@ app.get('/', function(req, res) {
 | 
				
			|||||||
});
 | 
					});
 | 
				
			||||||
 | 
					
 | 
				
			||||||
app.post('/text', function(req, res) {
 | 
					app.post('/text', function(req, res) {
 | 
				
			||||||
  if (true) {
 | 
					  var keystr = req.connection.remoteAddress + '_' + dateStr();
 | 
				
			||||||
    sendText(req.body.number, req.body.msg, function(err) {
 | 
					
 | 
				
			||||||
      if (err)
 | 
					  var rclient = redis.createClient();
 | 
				
			||||||
        res.send({success:false});
 | 
					  rclient.incr(keystr, function(err, num) {
 | 
				
			||||||
      else
 | 
					    rclient.quit();
 | 
				
			||||||
        res.send({success:true});
 | 
					
 | 
				
			||||||
    });
 | 
					    if (err) {
 | 
				
			||||||
  }
 | 
					      res.send({success:false,msg:'Could not validate IP quota.'});
 | 
				
			||||||
 | 
					      return;
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					    if (num < 51) {
 | 
				
			||||||
 | 
					      sendText(req.body.number, req.body.msg, function(err) {
 | 
				
			||||||
 | 
					        if (err)
 | 
				
			||||||
 | 
					          res.send({success:false,msg:'Communication with SMS gateway failed.'});
 | 
				
			||||||
 | 
					        else
 | 
				
			||||||
 | 
					          res.send({success:true});
 | 
				
			||||||
 | 
					      });
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					    else {
 | 
				
			||||||
 | 
					      res.send({success:false,msg:'Exceeded quota.'});
 | 
				
			||||||
 | 
					    }
 | 
				
			||||||
 | 
					  });
 | 
				
			||||||
});
 | 
					});
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					function dateStr() {
 | 
				
			||||||
 | 
					  var today = new Date();
 | 
				
			||||||
 | 
					  var dd = today.getDate();
 | 
				
			||||||
 | 
					  var mm = today.getMonth()+1;
 | 
				
			||||||
 | 
					  var yyyy = today.getFullYear();
 | 
				
			||||||
 | 
					  return mm + '/' + dd + '/' + yyyy;
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
function sendText(phone, msg, cb) {
 | 
					function sendText(phone, msg, cb) {
 | 
				
			||||||
  var transport = nodemailer.createTransport("SES", {
 | 
					  var transport = nodemailer.createTransport("SES", {
 | 
				
			||||||
    AWSAccessKeyID: config.aws.access,
 | 
					    AWSAccessKeyID: config.aws.access,
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -4,6 +4,7 @@
 | 
				
			|||||||
      "express": "2.5.2"
 | 
					      "express": "2.5.2"
 | 
				
			||||||
      , "jade": "0.14.2"
 | 
					      , "jade": "0.14.2"
 | 
				
			||||||
      , "nodemailer": "latest"
 | 
					      , "nodemailer": "latest"
 | 
				
			||||||
 | 
					      , "redis": "latest"
 | 
				
			||||||
    }
 | 
					    }
 | 
				
			||||||
  , "devDependencies": {
 | 
					  , "devDependencies": {
 | 
				
			||||||
    }
 | 
					    }
 | 
				
			||||||
 
 | 
				
			|||||||
		Reference in New Issue
	
	Block a user