Skip to content

Commit

Permalink
* samtools-0.1.6-9 (r470)
Browse files Browse the repository at this point in the history
 * make knetfile.c compatible with MinGW (thank Martin Morgan for the patch)
  • Loading branch information
Heng Li committed Sep 29, 2009
1 parent daf3efd commit 089233f
Show file tree
Hide file tree
Showing 4 changed files with 47 additions and 11 deletions.
2 changes: 1 addition & 1 deletion Makefile.mingw
@@ -1,7 +1,7 @@
CC= gcc.exe
AR= ar.exe
CFLAGS= -g -Wall -O2
DFLAGS= -D_FILE_OFFSET_BITS=64 -D_CURSES_LIB=2 -D_USE_KNETFILE
DFLAGS= -D_CURSES_LIB=2 -D_USE_KNETFILE
LOBJS= bgzf.o kstring.o bam_aux.o bam.o bam_import.o sam.o bam_index.o \
bam_pileup.o bam_lpileup.o bam_md.o glf.o razf.o faidx.o bam_sort.o \
knetfile.o
Expand Down
2 changes: 1 addition & 1 deletion bamtk.c
Expand Up @@ -9,7 +9,7 @@
#endif

#ifndef PACKAGE_VERSION
#define PACKAGE_VERSION "0.1.6-8 (r469)"
#define PACKAGE_VERSION "0.1.6-9 (r470)"
#endif

int bam_taf2baf(int argc, char *argv[]);
Expand Down
50 changes: 43 additions & 7 deletions knetfile.c
Expand Up @@ -71,7 +71,14 @@ static int socket_wait(int fd, int is_read)
if (is_read) fdr = &fds;
else fdw = &fds;
ret = select(fd+1, fdr, fdw, 0, &tv);
#ifndef _WIN32
if (ret == -1) perror("select");
#else
if (ret == 0)
fprintf(stderr, "select time-out\n");
else if (ret == SOCKET_ERROR)
fprintf(stderr, "select: %d\n", WSAGetLastError());
#endif
return ret;
}

Expand Down Expand Up @@ -104,16 +111,28 @@ static int socket_connect(const char *host, const char *port)
}
#else
/* MinGW's printf has problem with "%lld" */
char *uint64tostr(char *buf, uint64_t x)
char *int64tostr(char *buf, int64_t x)
{
int i, cnt;
for (i = 0; x; x /= 10) buf[i++] = '0' + x%10;
int cnt;
int i = 0;
do {
buf[i++] = '0' + x % 10;
x /= 10;
} while (x);
buf[i] = 0;
for (cnt = i, i = 0; i < cnt/2; ++i) {
int c = buf[i]; buf[i] = buf[cnt-i-1]; buf[cnt-i-1] = c;
}
return buf;
}

int64_t strtoint64(const char *buf)
{
int64_t x;
for (x = 0; *buf != '\0'; ++buf)
x = x * 10 + ((int64_t) *buf - 48);
return x;
}
/* In windows, the first thing is to establish the TCP connection. */
int knet_win32_init()
{
Expand All @@ -130,7 +149,11 @@ void knet_win32_destroy()
* non-Windows OS, I do not use this one. */
static SOCKET socket_connect(const char *host, const char *port)
{
#define __err_connect(func) do { perror(func); return -1; } while (0)
#define __err_connect(func) \
do { \
fprintf(stderr, "%s: %d\n", func, WSAGetLastError()); \
return -1; \
} while (0)

int on = 1;
SOCKET fd;
Expand Down Expand Up @@ -183,7 +206,11 @@ static off_t my_netread(int fd, void *buf, off_t len)

static int kftp_get_response(knetFile *ftp)
{
#ifndef _WIN32
unsigned char c;
#else
char c;
#endif
int n = 0;
char *p;
if (socket_wait(ftp->ctrl_fd, 1) <= 0) return 0;
Expand Down Expand Up @@ -260,6 +287,7 @@ int kftp_reconnect(knetFile *ftp)
ftp->ctrl_fd = -1;
}
netclose(ftp->fd);
ftp->fd = -1;
return kftp_connect(ftp);
}

Expand Down Expand Up @@ -300,18 +328,26 @@ int kftp_connect_file(knetFile *fp)
}
kftp_pasv_prep(fp);
kftp_send_cmd(fp, fp->size_cmd, 1);
#ifndef _WIN32
if ( sscanf(fp->response,"%*d %lld", &file_size) != 1 )
{
fprintf(stderr,"[kftp_connect_file] %s\n", fp->response);
return -1;
} else fp->file_size = file_size;
}
#else
const char *p = fp->response;
while (*p != ' ') ++p;
while (*p < '0' || *p > '9') ++p;
file_size = strtoint64(p);
#endif
fp->file_size = file_size;
if (fp->offset>=0) {
char tmp[32];
#ifndef _WIN32
sprintf(tmp, "REST %lld\r\n", (long long)fp->offset);
#else
strcpy(tmp, "REST ");
uint64tostr(tmp + 5, fp->offset);
int64tostr(tmp + 5, fp->offset);
strcat(tmp, "\r\n");
#endif
kftp_send_cmd(fp, tmp, 1);
Expand Down Expand Up @@ -492,7 +528,7 @@ off_t knet_read(knetFile *fp, void *buf, off_t len)
return l;
}

off_t knet_seek(knetFile *fp, off_t off, int whence)
off_t knet_seek(knetFile *fp, int64_t off, int whence)
{
if (whence == SEEK_SET && off == fp->offset) return 0;
if (fp->type == KNF_TYPE_LOCAL) {
Expand Down
4 changes: 2 additions & 2 deletions knetfile.h
Expand Up @@ -9,7 +9,7 @@
#define netwrite(fd, ptr, len) write(fd, ptr, len)
#define netclose(fd) close(fd)
#else
#include <winsock.h>
#include <winsock2.h>
#define netread(fd, ptr, len) recv(fd, ptr, len, 0)
#define netwrite(fd, ptr, len) send(fd, ptr, len, 0)
#define netclose(fd) closesocket(fd)
Expand Down Expand Up @@ -65,7 +65,7 @@ extern "C" {
This routine only sets ->offset and ->is_ready=0. It does not
communicate with the FTP server.
*/
off_t knet_seek(knetFile *fp, off_t off, int whence);
off_t knet_seek(knetFile *fp, int64_t off, int whence);
int knet_close(knetFile *fp);

#ifdef __cplusplus
Expand Down

0 comments on commit 089233f

Please sign in to comment.