Skip to content

Commit 2bacb0c

Browse files
committed
Lint and make it consistent
1 parent f41f77c commit 2bacb0c

File tree

3 files changed

+24
-29
lines changed

3 files changed

+24
-29
lines changed

include/buffer.h

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,11 +38,12 @@ int buffer_read_packet (buffer_t *, packet_t *);
3838

3939
int recv_buffer_write (char *data, int len);
4040
int recv_buffer_read (char *data, int len);
41-
int recv_file_buffer_read (int, int *, int, int);
4241

4342
int send_buffer_write (char *data, int len);
4443
int send_packet_buffer_write (packet_t *);
4544
int send_packet_buffer_read (packet_t *);
46-
int send_file_buffer_write (int, int, int, int);
45+
46+
int64_t recv_file_buffer_read (int, int64_t *, int64_t, int64_t);
47+
int64_t send_file_buffer_write (int, int64_t, int64_t, int64_t);
4748

4849
#endif /* end of include guard: BUFFER_H_IBSOFJRY */

src/api.c

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -107,14 +107,14 @@ int udt_close(socket_t sock)
107107
}
108108

109109
int64_t udt_recvfile(socket_t sock, int file, int64_t *offset, int64_t filesize,
110-
int blocksize)
110+
int blocksize)
111111
{
112112
if (!connection.is_connected) return -1;
113-
return send_file_buffer_read(file, offset, filesize, blocksize);
113+
return recv_file_buffer_read(file, offset, filesize, blocksize);
114114
}
115115

116116
int64_t udt_sendfile(socket_t sock, int file, int64_t offset, int64_t filesize,
117-
int blocksize)
117+
int blocksize)
118118
{
119119
if (!connection.is_connected) return -1;
120120
return send_file_buffer_write(file, offset, filesize, blocksize);

src/send_buffer.c

Lines changed: 18 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -71,37 +71,31 @@ int send_buffer_write(char *data, int len)
7171
return retval;
7272
}
7373

74-
int send_file_buffer_write(int fd, int offset, int size, int blocksize)
74+
int64_t send_file_buffer_write(int fd, int64_t offset, int64_t size, int64_t blocksize)
7575
{
76-
76+
7777
packet_t packet;
7878

79-
off_t pos;
79+
off_t pos;
8080
int retval;
8181
int seqnum;
82-
char *buffer;
82+
char buffer[PACKET_DATA_SIZE];
8383
int boundary;
84-
int readbytescount;
85-
int packet_dsize;
84+
int len;
8685

87-
pos = offset;
86+
pos = offset;
8887
retval = size;
8988
seqnum = 2142894844; /* TODO: generate random number */
90-
buffer = (char *) malloc (size);
9189
boundary = PACKET_BOUNDARY_START;
92-
93-
if (fd < 0 || pos < 0) return -1;
9490

95-
packet_dsize = (blocksize > PACKET_DATA_SIZE) ? PACKET_DATA_SIZE : blocksize;
96-
97-
while (size > 0)
91+
if (fd < 0) return -1;
92+
93+
while (size > 0)
9894
{
99-
100-
lseek(fd, pos, SEEK_SET);
101-
readbytescount = read(fd, buffer, packet_dsize);
102-
size -= readbytescount;
95+
len = pread(fd, buffer, PACKET_DATA_SIZE, offset);
96+
size -= len;
10397

104-
boundary = boundary | (size > 0) ? PACKET_BOUNDARY_NONE : PACKET_BOUNDARY_END;
98+
boundary |= (size > 0) ? PACKET_BOUNDARY_NONE : PACKET_BOUNDARY_END;
10599

106100
packet_clear_header (packet);
107101
packet_set_data (packet);
@@ -112,22 +106,22 @@ int send_file_buffer_write(int fd, int offset, int size, int blocksize)
112106
packet_set_timestamp(packet, 0x0000051c); /* TODO: calculate time */
113107
packet_set_id (packet, 0x08c42c74); /* TODO: generate an id */
114108

115-
packet_new(&packet, buffer, readbytescount);
109+
packet_new(&packet, buffer, len);
116110
send_packet_buffer_write(&packet);
117111

118112
boundary = PACKET_BOUNDARY_NONE;
119-
120-
pos += blocksize;
113+
114+
pos += blocksize;
121115
}
122-
123-
packet_clear_header (packet);
116+
117+
packet_clear_header (packet);
124118
packet_set_ctrl (packet);
125119
packet_set_type (packet, PACKET_TYPE_ACK);
126120
packet_set_timestamp(packet, 0x0000051c); /* TODO: calculate time */
127121
packet_set_id (packet, 0x08c42c74); /* TODO: generate an id */
128122

129123
packet_new(&packet, NULL, 0);
130124
send_packet_buffer_write(&packet);
131-
125+
132126
return retval; /* return total bytes sent */
133127
}

0 commit comments

Comments
 (0)