Switched the atomics to Bools

Author: lobsterjerusalem

c2/cli/basic.go

@@ -15,7 +15,10 @@ import (
 
 // A very basic reverse/bind shell handler.
 func Basic(conn net.Conn) {
-	var shutdown int32 = 0
+
+	var shutdown atomic.Bool
+	shutdown.Store(false)
+
 	// Create channels for communication between goroutines.
 	responseCh := make(chan string)
 
@@ -29,11 +32,11 @@ func Basic(conn net.Conn) {
 		defer func (){
 			// Signals for both routines to stop, this should get triggered when socket is closed
 			// and causes it to fail the read
-			atomic.StoreInt32(&shutdown, 1)
+			shutdown.Store(true)
 		}()
 		responseBuffer := make([]byte, 1024)
 		for {
-			if atomic.LoadInt32(&shutdown) == 1 {
+			if shutdown.Load() {
 				return
 			}
 			select {
@@ -61,7 +64,7 @@ func Basic(conn net.Conn) {
 	go func() {
 		defer wg.Done()
 		for {
-			if atomic.LoadInt32(&shutdown) == 1 {
+			if shutdown.Load() {
 				return
 			}
 			select {
@@ -81,7 +84,7 @@ func Basic(conn net.Conn) {
 		for { 
 			reader := bufio.NewReader(os.Stdin)
 			command,_ := reader.ReadString('\n')
-			if atomic.LoadInt32(&shutdown) == 1 {
+			if shutdown.Load() {
 				break
 			}
 			ok := protocol.TCPWrite(conn, []byte(command))