mirror of
https://github.com/Pithikos/C-Thread-Pool
synced 2024-11-22 05:31:18 +03:00
183 lines
2.9 KiB
Bash
Executable File
183 lines
2.9 KiB
Bash
Executable File
#! /bin/bash
|
|
|
|
#
|
|
# This file has several tests to check for memory leaks.
|
|
# valgrind is used so make sure you have it installed
|
|
#
|
|
|
|
|
|
|
|
# --------------------------- Generic ----------------------------------
|
|
|
|
function needle { #needle #haystack
|
|
python -c "import re; print(re.search(r'$1', '$2').group(0))"
|
|
}
|
|
function extract_num { #needle with number #haystack
|
|
string=$(needle "$1" "$2")
|
|
needle "[0-9]*" "$string"
|
|
}
|
|
function err { #string #log
|
|
echo "------------------- ERROR ------------------------"
|
|
echo "$1"
|
|
echo "$2" >> memleaks.log
|
|
exit 1
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# ---------------------------- Tests -----------------------------------
|
|
|
|
|
|
function test_mass_addition {
|
|
|
|
read -d '' code <<"EOF"
|
|
#include <stdio.h>
|
|
#include <unistd.h>
|
|
#include <time.h>
|
|
#include "../thpool.h"
|
|
|
|
pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
|
|
int sum=0;
|
|
|
|
void add_one() {
|
|
pthread_mutex_lock(&mutex);
|
|
sum ++;
|
|
pthread_mutex_unlock(&mutex);
|
|
}
|
|
|
|
int main(){
|
|
|
|
thpool_t* threadpool;
|
|
threadpool = thpool_init(10);
|
|
|
|
int n;
|
|
for (n=0; n<1000; n++){
|
|
thpool_add_work(threadpool, (void*)add_one, NULL);
|
|
}
|
|
|
|
thpool_wait(threadpool);
|
|
|
|
printf("%d", sum);
|
|
|
|
return 0;
|
|
}
|
|
EOF
|
|
|
|
echo "Adding up to 1000 with 10 threads"
|
|
echo "$code" > _test_.c
|
|
gcc _test_.c ../thpool.c -pthread -o test
|
|
output=$(./test 2>&1 /dev/null)
|
|
if [ "$output" == "1000" ]; then
|
|
return
|
|
fi
|
|
err "Expected 1000 but got $output" "$output"
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
function test_mass_addition2 {
|
|
|
|
read -d '' code <<"EOF"
|
|
#include <stdio.h>
|
|
#include <unistd.h>
|
|
#include <time.h>
|
|
#include "../thpool.h"
|
|
|
|
pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
|
|
int sum=0;
|
|
|
|
void add_one() {
|
|
pthread_mutex_lock(&mutex);
|
|
sum ++;
|
|
pthread_mutex_unlock(&mutex);
|
|
}
|
|
|
|
int main(){
|
|
|
|
thpool_t* threadpool;
|
|
threadpool = thpool_init(1000);
|
|
|
|
int n;
|
|
for (n=0; n<100; n++){
|
|
thpool_add_work(threadpool, (void*)add_one, NULL);
|
|
}
|
|
|
|
thpool_wait(threadpool);
|
|
|
|
printf("%d", sum);
|
|
|
|
return 0;
|
|
}
|
|
EOF
|
|
|
|
echo "Adding up to 100 with 1000 threads"
|
|
echo "$code" > _test_.c
|
|
gcc _test_.c ../thpool.c -pthread -o test
|
|
output=$(./test 2>&1 /dev/null)
|
|
if [ "$output" == "100" ]; then
|
|
return
|
|
fi
|
|
err "Expected 100 but got $output" "$output"
|
|
}
|
|
|
|
|
|
|
|
function test_mass_addition3 {
|
|
|
|
read -d '' code <<"EOF"
|
|
#include <stdio.h>
|
|
#include <unistd.h>
|
|
#include <time.h>
|
|
#include "../thpool.h"
|
|
|
|
pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
|
|
int sum=0;
|
|
|
|
void add_one() {
|
|
pthread_mutex_lock(&mutex);
|
|
sum ++;
|
|
pthread_mutex_unlock(&mutex);
|
|
}
|
|
|
|
int main(){
|
|
|
|
thpool_t* threadpool;
|
|
threadpool = thpool_init(1000);
|
|
|
|
int n;
|
|
for (n=0; n<100000; n++){
|
|
thpool_add_work(threadpool, (void*)add_one, NULL);
|
|
}
|
|
|
|
thpool_wait(threadpool);
|
|
|
|
printf("%d", sum);
|
|
|
|
return 0;
|
|
}
|
|
EOF
|
|
|
|
echo "Adding up to 100000 with 1000 threads"
|
|
echo "$code" > _test_.c
|
|
gcc _test_.c ../thpool.c -pthread -o test
|
|
output=$(./test 2>&1 /dev/null)
|
|
if [ "$output" == "100000" ]; then
|
|
return
|
|
fi
|
|
err "Expected 100000 but got $output" "$output"
|
|
}
|
|
|
|
|
|
|
|
# Run tests
|
|
test_mass_addition
|
|
test_mass_addition2
|
|
test_mass_addition3
|
|
|
|
echo "No errors"
|