tyler.durden/pixiewps
1
0
Fork 0
mirror of https://github.com/wiire-a/pixiewps.git synced 2025-07-27 07:44:36 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #58 from rofl0r/pthread_stack

Browse Source 
Set a reasonable stacksize for threads to prevent huge waste with GLIBC
This commit is contained in:
wiire-a 2017-10-15 20:47:17 +02:00 committed by GitHub
commit 537dd74029
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/pixiewps.c
View File

@ -19,6 +19,8 @@
* along with this program. If not, see <http://www.gnu.org/licenses/>. * along with this program. If not, see <http://www.gnu.org/licenses/>.
*/ */
#define _POSIX_C_SOURCE 200809L
#include <stdio.h> #include <stdio.h>
#include <stdlib.h> #include <stdlib.h>
#include <unistd.h> #include <unistd.h>
@ -26,6 +28,7 @@
#include <string.h> #include <string.h>
#include <getopt.h> #include <getopt.h>
#include <pthread.h> #include <pthread.h>
#include <limits.h>
#include <sys/types.h> #include <sys/types.h>
#include <sys/time.h> #include <sys/time.h>
@ -121,6 +124,10 @@ static void *crack_thread(void *arg) {
return 0; return 0;
} }
static size_t getminstacksize(size_t minimum) {
return (minimum < PTHREAD_STACK_MIN) ? PTHREAD_STACK_MIN : minimum;
}
static void init_crack_jobs(struct global *wps) { static void init_crack_jobs(struct global *wps) {
job_control.jobs = wps->jobs; job_control.jobs = wps->jobs;
job_control.end = wps->end; job_control.end = wps->end;
@ -138,10 +145,15 @@ static void init_crack_jobs(struct global *wps) {
job_control.randr_enonce[i] |= wps->e_nonce[j++]; job_control.randr_enonce[i] |= wps->e_nonce[j++];
} }
job_control.crack_jobs = malloc(wps->jobs * sizeof (struct job_control)); job_control.crack_jobs = malloc(wps->jobs * sizeof (struct job_control));
size_t stacksize = getminstacksize(64*1024);
time_t curr = wps->start; time_t curr = wps->start;
for(i = 0; i < wps->jobs; i++) { for(i = 0; i < wps->jobs; i++) {
job_control.crack_jobs[i].start = curr; job_control.crack_jobs[i].start = curr;
pthread_create(&job_control.crack_jobs[i].thr, NULL, crack_thread, &job_control.crack_jobs[i]); pthread_attr_t attr;
int attr_ok = pthread_attr_init(&attr) == 0 ;
if(attr_ok) pthread_attr_setstacksize(&attr, stacksize);
pthread_create(&job_control.crack_jobs[i].thr, &attr, crack_thread, &job_control.crack_jobs[i]);
if(attr_ok) pthread_attr_destroy(&attr);
curr -= SECS_PER_JOB_BLOCK; curr -= SECS_PER_JOB_BLOCK;
} }
} }