fix co_free memory leak

lvyes1988 · Jun 14, 2017 · eb75d36 · eb75d36
1 parent b4c4d69
commit eb75d36
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/co_routine.cpp b/co_routine.cpp
@@ -522,14 +522,16 @@ int co_create( stCoRoutine_t **ppco,const stCoRoutineAttr_t *attr,pfn_co_routine
 }
 void co_free( stCoRoutine_t *co )
 {
-	free( co );
+    if (!co->cIsShareStack) 
+    {    
+        free(co->stack_mem->stack_buffer);
+        free(co->stack_mem);
+    }   
+    free( co );
 }
 void co_release( stCoRoutine_t *co )
 {
-	if( co->cEnd )
-	{
-		free( co );
-	}
+    co_free( co );
 }
 
 void co_swap(stCoRoutine_t* curr, stCoRoutine_t* pending_co);