char *funrle_data_to(const u8 *data, ssize_t size, int fd_out)
{
int ret;
const u8 *data_end;
u8 tmp;
char buff[MAX_BUFFER_LEN];
char *p, *end_p;
struct progress_bar prg_bar;
progress_bar_init(&prg_bar, size);
p = buff;
end_p = buff + sizeof(buff);
for (data_end = data + size; data < data_end; data += 2)
{
tmp = *data;
println("tmp = %d", tmp);
while (p + tmp >= end_p)
{
memset(p, data[1], end_p - p);
ret = ffile_writeto(fd_out, buff, MAX_BUFFER_LEN, 0);
if (ret < 0)
{
error_msg("write file failed");
return NULL;
}
tmp -= end_p - p;
p = buff;
}
memset(p, data[1], tmp);
p += tmp;
progress_bar_add(&prg_bar, 2);
}
progress_bar_finish(&prg_bar);
if (p > buff)
{
ret = ffile_writeto(fd_out, buff, p - buff, 0);
if (ret < 0)
{
error_msg("write file failed");
return NULL;
}
}
return (char *) data;
}
char *frle_data_to(const u8 *data, ssize_t size, int fd_out)
{
int ret;
const u8 *data_end = data + size;
u8 tmp;
u8 buff[MAX_BUFFER_LEN];
const u8 *p;
u8 *q, *end_q;
struct progress_bar prg_bar;
q = buff;
end_q = buff + sizeof(buff) - 1;
progress_bar_init(&prg_bar, size);
while (data < data_end)
{
for (p = data, tmp = *p; tmp == *p && p < data_end; p++);
if (q >= end_q)
{
ret = ffile_writeto(fd_out, buff, q - buff, 0);
if (ret < 0)
{
error_msg("write file failed");
return NULL;
}
q = buff;
}
*q++ = p - data;
*q++ = tmp;
progress_bar_add(&prg_bar, p - data);
data = p;
}
progress_bar_finish(&prg_bar);
if (q > buff)
{
ret = ffile_writeto(fd_out, buff, q - buff, 0);
if (ret < 0)
{
error_msg("write file failed");
return NULL;
}
}
return (char *) data;
}
int cftp_server_send_file(struct cftp_descriptor *desc, const char *filename, u32 offset, size_t size)
{
int fd;
int ret;
union cftp_message *msg;
struct cftp_data_package *data_pkg;
ssize_t recvlen, sendlen, readlen;
u16 blk_num;
struct progress_bar bar;
size_t max_xfer_length, max_data_length;
struct stat st;
max_xfer_length = desc->max_xfer_length;
max_data_length = max_xfer_length - sizeof(msg->data_pkg);
msg = malloc(max_xfer_length);
if (msg == NULL) {
pr_err_info("malloc");
return -ENOMEM;
}
fd = open(filename, O_RDONLY);
if (fd < 0) {
pr_err_info("open file %s faild", filename);
cftp_send_error_message(desc, (struct cftp_error_message *) msg, "open file %s faild", filename);
ret = fd;
goto out_free_msg;
}
ret = fstat(fd, &st);
if (ret < 0) {
pr_err_info("fstat");
cftp_send_error_message(desc, (struct cftp_error_message *) msg, "fstat");
goto out_close_file;
}
if (offset) {
ret = lseek(fd, offset, SEEK_SET);
if (ret < 0) {
pr_err_info("lseek");
cftp_send_error_message(desc, (struct cftp_error_message *) msg, "seek file %s faild", filename);
goto out_close_file;
}
}
if (size == 0) {
size = st.st_size - offset;
}
data_pkg = malloc(max_xfer_length);
if (data_pkg == NULL) {
pr_err_info("malloc");
cftp_send_error_message(desc, (struct cftp_error_message *) msg, "malloc");
ret = -ENOMEM;
goto out_close_file;
}
println("filename = %s", filename);
println("offset = %s", size2text(offset));
println("size = %s", size2text(size));
blk_num = 0;
data_pkg->type = CFTP_PACKAGE_DATA;
progress_bar_init(&bar, size, 0, PROGRESS_BAR_TYPE_DATA);
while (1) {
readlen = read(fd, data_pkg->data, max_data_length);
if (readlen < 0) {
pr_err_info("read file failed");
cftp_send_error_message(desc, (struct cftp_error_message *) msg, "read file failed");
ret = readlen;
goto out_free_data_pkg;
}
data_pkg->blk_num = blk_num++;
label_send_data:
sendlen = cftp_send_data_retry(desc, data_pkg, sizeof(*data_pkg) + readlen, desc->retry_count);
if (sendlen < 0) {
pr_err_info("cftp_send_ack_message");
ret = sendlen;
goto out_free_data_pkg;
}
recvlen = cftp_receive_data(desc, msg, max_xfer_length);
if (recvlen < 0) {
pr_err_info("cftp_receive_data");
cftp_send_error_message(desc, (struct cftp_error_message *) msg, "receive file failed");
ret = recvlen;
goto out_free_data_pkg;
}
switch (msg->type) {
case CFTP_PACKAGE_ERROR:
show_error_message((struct cftp_error_message *) msg);
ret = -EFAULT;
goto out_free_data_pkg;
case CFTP_PACKAGE_ACK:
if (msg->ack_msg.blk_num != blk_num) {
pr_warn_info("blk_num %d != %d", msg->ack_msg.blk_num, blk_num);
goto label_send_data;
}
if ((size_t) readlen < max_data_length) {
progress_bar_finish(&bar);
println("Send data complete");
ret = 0;
goto out_free_data_pkg;
}
progress_bar_add(&bar, max_data_length);
break;
default:
pr_err_info("invalid package type");
cftp_send_error_message(desc, (struct cftp_error_message *) msg, "invalid package type");
ret = -EINVAL;
goto out_free_data_pkg;
}
}
out_free_data_pkg:
free(data_pkg);
out_close_file:
close(fd);
out_free_msg:
free(msg);
return ret;
}
int cftp_server_receive_file(struct cftp_descriptor *desc, const char *filename, mode_t mode, u32 offset, size_t size)
{
int fd;
int ret;
union cftp_message *msg;
ssize_t recvlen, sendlen;
u16 blk_num;
struct progress_bar bar;
size_t max_xfer_length;
max_xfer_length = desc->max_xfer_length;
msg = malloc(max_xfer_length);
if (msg == NULL) {
pr_err_info("malloc");
return -ENOMEM;
}
fd = open(filename, O_WRONLY | O_CREAT | O_TRUNC, mode);
if (fd < 0) {
pr_err_info("open file %s faild", filename);
cftp_send_error_message(desc, (struct cftp_error_message *) msg, "open file %s faild", filename);
ret = fd;
goto out_free_msg;
}
if (offset) {
ret = lseek(fd, offset, SEEK_SET);
if (ret < 0) {
pr_err_info("lseek");
cftp_send_error_message(desc, (struct cftp_error_message *) msg, "seek file %s faild", filename);
goto out_free_msg;
}
}
println("offset = %s", size2text(offset));
println("size = %s", size2text(size));
blk_num = 0;
progress_bar_init(&bar, size, 0, PROGRESS_BAR_TYPE_DATA);
while (1) {
sendlen = cftp_send_ack_message(desc, (struct cftp_ack_message *) msg, blk_num, desc->retry_count);
if (sendlen < 0) {
pr_err_info("cftp_send_ack_message");
ret = sendlen;
goto out_close_file;
}
recvlen = cftp_receive_data(desc, msg, max_xfer_length);
if (recvlen < 0) {
pr_err_info("cftp_receive_data");
cftp_send_error_message(desc, (struct cftp_error_message *) msg, "receive file failed");
ret = recvlen;
goto out_close_file;
}
switch (msg->type) {
case CFTP_PACKAGE_ERROR:
show_error_message((struct cftp_error_message *) msg);
ret = -EFAULT;
goto out_close_file;
case CFTP_PACKAGE_DATA:
if (msg->data_pkg.blk_num != blk_num) {
pr_warn_info("blk_num %d != %d", msg->data_pkg.blk_num, blk_num);
continue;
}
sendlen = write(fd, msg->data_pkg.data, recvlen - sizeof(msg->data_pkg));
if (sendlen < 0) {
pr_err_info("write");
cftp_send_error_message(desc, (struct cftp_error_message *) msg, "write file failed");
ret = sendlen;
goto out_close_file;
}
blk_num++;
if ((size_t) recvlen < max_xfer_length) {
cftp_send_ack_message(desc, (struct cftp_ack_message *) msg, blk_num, 0);
progress_bar_finish(&bar);
println("Receive data complete");
ret = 0;
goto out_close_file;
} else {
progress_bar_add(&bar, sendlen);
}
break;
default:
pr_err_info("invalid package type");
cftp_send_error_message(desc, (struct cftp_error_message *) msg, "invalid package type");
ret = -EINVAL;
goto out_close_file;
}
}
out_close_file:
close(fd);
out_free_msg:
free(msg);
return ret;
}
int cftp_client_send_file(struct cftp_descriptor *desc, const char *file_in, u32 offset_in, const char *file_out, u32 offset_out, size_t size)
{
int fd;
int ret;
struct stat st;
ssize_t recvlen, sendlen, readlen;
u16 blk_num;
union cftp_message *msg;
struct cftp_data_package *data_msg;
size_t max_xfer_length, max_data_length;
struct progress_bar bar;
ret = file_lstat(file_in, &st);
if (ret < 0) {
pr_err_info("fstat");
return ret;
}
max_xfer_length = desc->max_xfer_length;
msg = malloc(max_xfer_length);
if (msg == NULL) {
pr_err_info("malloc");
return -ENOMEM;
}
switch (st.st_mode & S_IFMT) {
case S_IFREG:
println("Send File: %s => %s", file_in, file_out);
break;
case S_IFBLK:
case S_IFCHR:
println("Send Devicce: %s => %s", file_in, file_out);
return cftp_client_send_special_file(desc, &st, &msg->file_req, text_copy(msg->file_req.filename, file_out) - msg->file_req.filename + 1);
case S_IFLNK:
println("Send Symlink: %s => %s", file_in, file_out); {
char *p;
p = text_copy(msg->file_req.filename, file_out) + 1;
ret = readlink(file_in, p, 1024);
if (ret < 0) {
pr_err_info("readlink");
return ret;
}
return cftp_client_send_special_file(desc, &st, &msg->file_req, p - msg->file_req.filename + ret + 1);
}
case S_IFDIR:
println("Send Directory: %s => %s", file_in, file_out);
ret = cftp_client_send_special_file(desc, &st, &msg->file_req, text_copy(msg->file_req.filename, file_out) - msg->file_req.filename + 1);
if (ret < 0) {
pr_err_info("cftp_client_send_special_file");
return ret;
}
return cftp_client_send_directory(desc, file_in, file_out);
default:
pr_warn_info("File %s type is unknown", file_in);
return 0;
}
fd = open(file_in, O_RDONLY);
if (fd < 0) {
pr_err_info("open file %s failed", file_in);
goto out_free_msg;
}
if (offset_in) {
ret = lseek(fd, offset_in, SEEK_SET);
if (ret < 0) {
pr_err_info("lseek");
goto out_close_file;
}
}
if (size == 0) {
size = st.st_size - offset_in;
}
data_msg = malloc(max_xfer_length);
if (data_msg == NULL) {
pr_err_info("malloc");
ret = -ENOMEM;
goto out_close_file;
}
sendlen = cftp_send_file_reuest(desc, (void *) msg, file_out, &st, offset_out, size, 0);
if (sendlen < 0) {
ret = sendlen;
pr_err_info("cftp_send_data_retry");
goto out_free_data_msg;
}
println("seek = %s", size2text(offset_out));
println("skip = %s", size2text(offset_in));
println("size = %s", size2text(size));
blk_num = 0;
data_msg->type = CFTP_PACKAGE_DATA;
max_data_length = max_xfer_length - sizeof(*data_msg);
progress_bar_init(&bar, size, 0, PROGRESS_BAR_TYPE_DATA);
readlen = max_data_length;
while (1) {
recvlen = cftp_receive_data(desc, msg, max_xfer_length);
if (recvlen < 0) {
pr_err_info("cftp_receive_data");
ret = recvlen;
goto out_free_data_msg;
}
switch (msg->type) {
case CFTP_PACKAGE_ERROR:
show_error_message((struct cftp_error_message *) msg);
ret = -EFAULT;
goto out_free_data_msg;
case CFTP_PACKAGE_ACK:
if (msg->ack_msg.blk_num == blk_num) {
if ((size_t) readlen < max_data_length) {
ret = 0;
progress_bar_finish(&bar);
println("Send data complete");
goto out_free_data_msg;
}
readlen = read(fd, data_msg->data, max_data_length);
if (readlen < 0) {
pr_err_info("read");
cftp_send_error_message(desc, (struct cftp_error_message *) msg, "read file failed");
ret = readlen;
goto out_free_data_msg;
}
data_msg->blk_num = blk_num++;
progress_bar_add(&bar, readlen);
} else {
pr_warn_info("%d != %d", msg->ack_msg.blk_num, blk_num);
if (blk_num == 0) {
continue;
}
}
sendlen = cftp_send_data_retry(desc, data_msg, sizeof(*data_msg) + readlen, desc->retry_count);
if (sendlen < 0) {
pr_err_info("cftp_send_data_retry");
goto out_free_data_msg;
}
break;
default:
pr_err_info("invalid package type");
cftp_send_error_message(desc, (struct cftp_error_message *) msg, "invalid package type");
ret = -EINVAL;
goto out_free_data_msg;
}
}
out_free_data_msg:
free(data_msg);
out_close_file:
close(fd);
out_free_msg:
free(msg);
return ret;
}
