diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/file.c | 4 | ||||
| -rw-r--r-- | src/main.c | 11 | ||||
| -rw-r--r-- | src/platform.c | 17 |
3 files changed, 22 insertions, 10 deletions
@@ -306,7 +306,9 @@ int file_fill_dst(char *target, struct list_head *file_list) list_for_each_entry(f, file_list, list) { f->dst_remote = dst_remote; - snprintf(f->dst_path, PATH_MAX, "%s/%s", dst_path, f->path); + strncat(f->dst_path, dst_path, PATH_MAX); + strncat(f->dst_path, "/", PATH_MAX); + strncat(f->dst_path, f->path, PATH_MAX); } return 0; @@ -307,6 +307,10 @@ int main(int argc, char **argv) struct sscp_thread *t = &threads[n]; t->sscp = &sscp; t->finished = false; + t->sftp = ssh_make_sftp_session(sscp.host, sscp.opts); + if (!t->sftp) + goto join_out; + ret = pthread_create(&t->tid, NULL, sscp_copy_thread, t); if (ret < 0) { pr_err("pthread_create error: %d\n", ret); @@ -352,14 +356,9 @@ void *sscp_copy_thread(void *arg) { struct sscp_thread *t = arg; struct sscp *sscp = t->sscp; - sftp_session sftp; + sftp_session sftp = t->sftp; struct chunk *c; - /* create sftp session */ - sftp = ssh_make_sftp_session(sscp->host, sscp->opts); - if (!sftp) - return NULL; - pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, NULL); pthread_setcanceltype(PTHREAD_CANCEL_DEFERRED, NULL); pthread_cleanup_push(sscp_copy_thread_cleanup, t); diff --git a/src/platform.c b/src/platform.c index 5e0573e..1a87cd3 100644 --- a/src/platform.c +++ b/src/platform.c @@ -1,14 +1,15 @@ -#include <util.h> -#include <platform.h> - #ifdef __APPLE__ #include <sys/types.h> #include <sys/sysctl.h> #elif linux +#define _GNU_SOURCE +#include <sched.h> #else #error unsupported platform #endif +#include <util.h> +#include <platform.h> #ifdef __APPLE__ int nr_cpus() @@ -25,3 +26,13 @@ int nr_cpus() } #endif +#ifdef linux +int nr_cpus() +{ + cpu_set_t cpu_set; + if (sched_getaffinity(0, sizeof(cpu_set_t), &cpu_set) == 0) + return CPU_COUNT(&cpu_set); + return -1; +} +#endif + |