From 4a0eeba2372755de574f69ca97525974addec8a3 Mon Sep 17 00:00:00 2001
From: nickago <ngonella@calpoly.edu>
Date: Tue, 25 Aug 2015 09:19:07 -0700
Subject: Added semaphore for resource allocation to echo command

---
 api/command.go | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

(limited to 'api/command.go')

diff --git a/api/command.go b/api/command.go
index 31f676a11..749cbf790 100644
--- a/api/command.go
+++ b/api/command.go
@@ -24,6 +24,8 @@ var commands = []commandHandler{
 	echoCommand,
 }
 
+var echoSem chan bool
+
 func InitCommand(r *mux.Router) {
 	l4g.Debug("Initializing command api routes")
 	r.Handle("/command", ApiUserRequired(command)).Methods("POST")
@@ -95,6 +97,7 @@ func logoutCommand(c *Context, command *model.Command) bool {
 
 func echoCommand(c *Context, command *model.Command) bool {
 	cmd := "/echo"
+	maxThreads := 100
 
 	if !command.Suggest && strings.Index(command.Command, cmd) == 0 {
 		parameters := strings.SplitN(command.Command, " ", 2)
@@ -118,8 +121,24 @@ func echoCommand(c *Context, command *model.Command) bool {
 			}
 		}
 
-		// Fire off asynchronous process
+		if delay > 10000 {
+			c.Err = model.NewAppError("echoCommand", "Delays must be under 10000 seconds", "")
+			return false
+		}
+
+		if echoSem == nil {
+			// We want one additional thread allowed so we never reach channel lockup
+			echoSem = make(chan bool, maxThreads+1)
+		}
+
+		if len(echoSem) >= maxThreads {
+			c.Err = model.NewAppError("echoCommand", "High volume of echo request, cannot process request", "")
+			return false
+		}
+
+		echoSem <- true
 		go func() {
+			defer func() { <-echoSem }()
 			post := &model.Post{}
 			post.ChannelId = command.ChannelId
 			post.Message = message
-- 
cgit v1.2.3-1-g7c22