Adds seeking, directory creation and improves file writing. Also various other cleanups.

Implements creating directories as well as writing and deleting files in iFuse.

Signed-off-by: Matthew Colyer <mcolyer@mcolyer-laptop.(none)>

1 files changed, 148 insertions, 32 deletions

diff --git a/src/AFC.c b/src/AFC.c
index 6fb5a9e..6a0df50 100644
--- a/src/AFC.c
+++ b/src/AFC.c
@@ -29,13 +29,15 @@ extern int debug;
 
 /* Locking, for thread-safety (well... kind of, hehe) */
 void afc_lock(AFClient *client) {
+        if (debug) printf("In the midst of a lock...\n");
         while (client->lock) {
-                sleep(200);
+                usleep(500); // they say it's obsolete, but whatever
         }
         client->lock = 1;
 }
 
 void afc_unlock(AFClient *client) { // just to be pretty 
+        if (debug) printf("Unlock!\n");
         client->lock = 0; 
 }
 
@@ -102,13 +104,13 @@ int dispatch_AFC_packet(AFClient *client, const char *data, int length) {
                 if ((length) < (client->afc_packet->entire_length - client->afc_packet->this_length)) {
                         if (debug) printf("dispatch_AFC_packet: Length did not resemble what it was supposed to based on the packet.\nlength minus offset: %i\nrest of packet: %i\n", length-offset, client->afc_packet->entire_length - client->afc_packet->this_length);
                         free(buffer);
-                        return 0;
+                        return -1;
                 }
                 if (debug) printf("dispatch_AFC_packet: fucked-up packet method (probably a write)\n");
                 memcpy(buffer+sizeof(AFCPacket), data, offset);
                 bytes = mux_send(client->connection, buffer, client->afc_packet->this_length);
                 free(buffer);
-                if (bytes <= 0) { return 0; }
+                if (bytes <= 0) { return bytes; }
                 if (debug) {
                         printf("dispatch_AFC_packet: sent the first now go with the second\n");
                         printf("Length: %i\n", length-offset);
@@ -118,8 +120,7 @@ int dispatch_AFC_packet(AFClient *client, const char *data, int length) {
                 
                 
                 bytes = mux_send(client->connection, data+offset, length-offset);
-                if (bytes <= 0) { return 0; }
-                else { return bytes; }
+                return bytes;
         } else {
                 if (debug) printf("dispatch_AFC_packet doin things the old way\n");
                 char *buffer = (char*)malloc(sizeof(char) * client->afc_packet->this_length);
@@ -130,10 +131,9 @@ int dispatch_AFC_packet(AFClient *client, const char *data, int length) {
                 if (debug) fwrite(buffer, 1, client->afc_packet->this_length, stdout);
                 if (debug) printf("\n");
                 bytes = mux_send(client->connection, buffer, client->afc_packet->this_length);
-                if (bytes <= 0) return 0;
-                else return bytes;
+                return bytes;
         }
-        return 0;
+        return -1;
 }
 
 int receive_AFC_data(AFClient *client, char **dump_here) {
@@ -148,7 +148,7 @@ int receive_AFC_data(AFClient *client, char **dump_here) {
                 free(buffer);
                 printf("Just didn't get enough.\n");
                 *dump_here = NULL;
-                return 0;
+                return -1;
         }
         
         r_packet = (AFCPacket*)malloc(sizeof(AFCPacket));
@@ -177,19 +177,19 @@ int receive_AFC_data(AFClient *client, char **dump_here) {
                 if(param1 == 0) {
                         if (debug) printf("... false alarm, but still\n");
                         *dump_here = NULL;
-                        return 1;
+                        return 0;
                 }
                 else { if (debug) printf("Errno %i\n", param1); }
                 free(r_packet);
                 *dump_here = NULL;
-                return 0;
+                return -1;
         } else {
                 if (debug) printf("Operation code %x\nFull length %i and this length %i\n", r_packet->operation, r_packet->entire_length, r_packet->this_length);
         }
 
         recv_len = r_packet->entire_length - r_packet->this_length;
         free(r_packet);
-        if (!recv_len)
+        if (!recv_len && r_packet->operation == AFC_SUCCESS_RESPONSE)
         {
                 *dump_here = NULL;
                 return 0;
@@ -240,10 +240,28 @@ char **afc_get_dir_list(AFClient *client, const char *dir) {
         char *blah = NULL, **list = NULL;
         client->afc_packet->entire_length = client->afc_packet->this_length = 0;
         bytes = dispatch_AFC_packet(client, dir, strlen(dir));
-        if (!bytes) { afc_unlock(client); return NULL; }
+        if (bytes <= 0) { afc_unlock(client); return NULL; }
+        
+        bytes = receive_AFC_data(client, &blah);
+        if (bytes < 0 && !blah) { afc_unlock(client); return NULL; }
+        
+        list = make_strings_list(blah, bytes);
+        free(blah);
+        afc_unlock(client);
+        return list;
+}
+
+char **afc_get_devinfo(AFClient *client) {
+        afc_lock(client);
+        client->afc_packet->operation = AFC_GET_DEVINFO;
+        int bytes = 0;
+        char *blah = NULL, **list = NULL;
+        client->afc_packet->entire_length = client->afc_packet->this_length = 0;
+        bytes = dispatch_AFC_packet(client, NULL, 0);
+        if (bytes < 0) { afc_unlock(client); return NULL; }
         
         bytes = receive_AFC_data(client, &blah);
-        if (!bytes && !blah) { afc_unlock(client); return NULL; }
+        if (bytes < 0 && !blah) { afc_unlock(client); return NULL; } 
         
         list = make_strings_list(blah, bytes);
         free(blah);
@@ -251,6 +269,7 @@ char **afc_get_dir_list(AFClient *client, const char *dir) {
         return list;
 }
 
+        
 char **make_strings_list(char *tokens, int true_length) {
         if (!tokens || !true_length) return NULL;
         int nulls = 0, i = 0, j = 0;
@@ -279,7 +298,7 @@ int afc_delete_file(AFClient *client, const char *path) {
         bytes = receive_AFC_data(client, &receive);
         free(receive);
         afc_unlock(client);
-        if (bytes <= 0) { return 0; }
+        if (bytes < 0) { return 0; }
         else return 1;
 }
 
@@ -302,12 +321,28 @@ int afc_rename_file(AFClient *client, const char *from, const char *to) {
         bytes = receive_AFC_data(client, &receive);
         free(receive);
         afc_unlock(client);
-        if (bytes <= 0) return 0;
+        if (bytes < 0) return 0;
         else return 1;
 }
 
+int afc_mkdir(AFClient *client, const char *dir) {
+        if (!client) return 0;
+        afc_lock(client);
+        int bytes = 0;
+        char *recvd = NULL;
         
+        client->afc_packet->operation = AFC_MAKE_DIR;
+        client->afc_packet->this_length = client->afc_packet->entire_length = 0;
+        bytes = dispatch_AFC_packet(client, dir, strlen(dir));
+        if (bytes <= 0) { afc_unlock(client); return 0; }
         
+        bytes = receive_AFC_data(client, &recvd);
+        afc_unlock(client);
+        if (recvd) { free(recvd); recvd = NULL; }
+        if (bytes == 0) return 1;
+        else return 0;
+}
+
 AFCFile *afc_get_file_info(AFClient *client, const char *path) {
         client->afc_packet->operation = AFC_GET_INFO;
         client->afc_packet->entire_length = client->afc_packet->this_length = 0;
@@ -350,7 +385,7 @@ AFCFile *afc_get_file_info(AFClient *client, const char *path) {
 }
 
 AFCFile *afc_open_file(AFClient *client, const char *filename, uint32 file_mode) {
-        if (file_mode != AFC_FILE_READ && file_mode != AFC_FILE_WRITE) return NULL;
+        //if (file_mode != AFC_FILE_READ && file_mode != AFC_FILE_WRITE) return NULL;
         if (!client ||!client->connection || !client->afc_packet) return NULL;
         afc_lock(client);
         char *further_data = (char*)malloc(sizeof(char) * (8 + strlen(filename) + 1));
@@ -372,17 +407,19 @@ AFCFile *afc_open_file(AFClient *client, const char *filename, uint32 file_mode)
                 return NULL;
         } else {
                 length_thing = receive_AFC_data(client, &further_data);
-                if (length_thing && further_data) {
+                if (length_thing > 0 && further_data) {
                         afc_unlock(client); // don't want to hang on the next call... and besides, it'll re-lock, do its thing, and unlock again anyway.
                         file_infos = afc_get_file_info(client, filename);
                         memcpy(&file_infos->filehandle, further_data, 4);
                         return file_infos;
                 } else {
                         if (debug) printf("didn't get further data or something\n");
+                        afc_unlock(client);
                         return NULL;
                 }
         }
         if (debug) printf("what the fuck\n");
+        afc_unlock(client);
         return NULL;
 }
 
@@ -408,7 +445,8 @@ int afc_read_file(AFClient *client, AFCFile *file, char *data, int length) {
 
         // Looping here to get around the maximum amount of data that recieve_AFC_data can handle
         while (current_count < length){
-
+                if (debug) printf("afc_read_file: current count is %i but length is %i\n", current_count, length);
+                
                 // Send the read command
                 AFCFilePacket *packet = (AFCFilePacket*)malloc(sizeof(AFCFilePacket));
                 packet->unknown1 = packet->unknown2 = 0;
@@ -421,6 +459,7 @@ int afc_read_file(AFClient *client, AFCFile *file, char *data, int length) {
                 // If we get a positive reponse to the command gather the data
                 if (bytes > 0) {
                         bytes = receive_AFC_data(client, &input);
+                        if (debug) printf("afc_read_file: bytes returned: %i\n", bytes);
                         if (bytes < 0) {
                                 if (input) free(input);
                                 afc_unlock(client);
@@ -431,7 +470,7 @@ int afc_read_file(AFClient *client, AFCFile *file, char *data, int length) {
                                 return current_count;
                         } else {
                                 if (input) {
-                                        printf("afc_read_file: %d\n", bytes);
+                                        if (debug) printf("afc_read_file: %d\n", bytes);
                                         memcpy(data+current_count, input, (bytes > length) ? length : bytes);
                                         free(input);
                                         input = NULL;
@@ -444,6 +483,7 @@ int afc_read_file(AFClient *client, AFCFile *file, char *data, int length) {
                 }
         }
         afc_unlock(client);
+        if (debug) printf("afc_read_file: returning current_count as %i\n", current_count);
         return current_count;
 }
 
@@ -451,29 +491,53 @@ int afc_write_file(AFClient *client, AFCFile *file, const char *data, int length
         char *acknowledgement = NULL;
         if (!client ||!client->afc_packet || !client->connection || !file) return -1;
         afc_lock(client);
-        client->afc_packet->this_length = sizeof(AFCPacket) + 8;
-        client->afc_packet->entire_length = client->afc_packet->this_length + length;
-        client->afc_packet->operation = AFC_WRITE;
         if (debug) printf("afc_write_file: Write length: %i\n", length);
-        uint32 zero = 0, bytes = 0;
-        
+        const int MAXIMUM_WRITE_SIZE = 1 << 16;
+        uint32 zero = 0, bytes = 0, segments = (length / MAXIMUM_WRITE_SIZE), current_count = 0, i = 0;
         char *out_buffer = NULL;
+        
+        for (i = 0; i < segments; i++) { // Essentially, yeah, divide it into segments.
+                client->afc_packet->this_length = sizeof(AFCPacket) + 8;
+                //client->afc_packet->entire_length = client->afc_packet->this_length + length;
+                client->afc_packet->entire_length = client->afc_packet->this_length + MAXIMUM_WRITE_SIZE;
+                client->afc_packet->operation = AFC_WRITE;
+                out_buffer = (char*)malloc(sizeof(char) * client->afc_packet->entire_length - sizeof(AFCPacket));
+                memcpy(out_buffer, (char*)&file->filehandle, sizeof(uint32));
+                memcpy(out_buffer+4, (char*)&zero, sizeof(uint32));
+                memcpy(out_buffer+8, data+current_count, MAXIMUM_WRITE_SIZE);
+        
+                bytes = dispatch_AFC_packet(client, out_buffer, MAXIMUM_WRITE_SIZE + 8);
+                if (bytes < 0) { afc_unlock(client); return bytes; }
+                free(out_buffer); out_buffer = NULL; // cleanup and hope it works
+                current_count += bytes;
+                bytes = receive_AFC_data(client, &acknowledgement); 
+                if (bytes < 0) { afc_unlock(client); return current_count; }
+        }
+        
+        // By this point, we should be at the end. i.e. the last segment that didn't get sent in the for loop
+        // this length is fine because it's always sizeof(AFCPacket) + 8, but to be sure we do it again
+        if (current_count == length) { afc_unlock(client); return current_count; }
+        client->afc_packet->this_length = sizeof(AFCPacket) + 8;
+        client->afc_packet->entire_length = client->afc_packet->this_length + (length - current_count);
+        // operation is already AFC_WRITE, but set again to be sure
+        client->afc_packet->operation = AFC_WRITE;
         out_buffer = (char*)malloc(sizeof(char) * client->afc_packet->entire_length - sizeof(AFCPacket));
         memcpy(out_buffer, (char*)&file->filehandle, sizeof(uint32));
         memcpy(out_buffer+4, (char*)&zero, sizeof(uint32));
-        memcpy(out_buffer+8, data, length);
-        
-        bytes = dispatch_AFC_packet(client, out_buffer, length + 8);
-        if (!bytes) return -1;
+        memcpy(out_buffer+8, data+current_count, (length - current_count));
+        bytes = dispatch_AFC_packet(client, out_buffer, (length - current_count) + 8);
+        free(out_buffer); out_buffer = NULL;
+        current_count += bytes;
+        if (bytes <= 0) { afc_unlock(client); return current_count; }
         
         zero = bytes;
         bytes = receive_AFC_data(client, &acknowledgement);
         afc_unlock(client);
-        if (bytes <= 0) {
+        if (bytes < 0) {
                 if (debug) printf("afc_write_file: uh oh?\n");
         }
         
-        return zero;
+        return current_count;
 }
 
 void afc_close_file(AFClient *client, AFCFile *file) {
@@ -490,9 +554,61 @@ void afc_close_file(AFClient *client, AFCFile *file) {
 
         free(buffer);
         client->afc_packet->entire_length = client->afc_packet->this_length = 0;
-        if (!bytes) return;
+        if (bytes <= 0) { afc_unlock(client); return; }
         
         bytes = receive_AFC_data(client, &buffer);
         afc_unlock(client);
+        if (buffer) { free(buffer); }
         return;
 }
+
+int afc_seek_file(AFClient *client, AFCFile *file, int seekpos) {
+        afc_lock(client);
+        
+        char *buffer = (char*)malloc(sizeof(char) * 24);
+        uint32 seekto = 0, bytes = 0, zero = 0;
+        if (seekpos < 0) seekpos = file->size - abs(seekpos);
+        seekto = seekpos;
+        
+        memcpy(buffer, &file->filehandle, sizeof(uint32)); // handle
+        memcpy(buffer+4, &zero, sizeof(uint32)); // pad
+        memcpy(buffer+8, &zero, sizeof(uint32)); // fromwhere
+        memcpy(buffer+12, &zero, sizeof(uint32)); // pad
+        memcpy(buffer+16, &seekto, sizeof(uint32)); // offset
+        memcpy(buffer+20, &zero, sizeof(uint32)); // pad
+        client->afc_packet->operation = AFC_FILE_SEEK;
+        client->afc_packet->this_length = client->afc_packet->entire_length = 0;
+        bytes = dispatch_AFC_packet(client, buffer, 23);
+        free(buffer); buffer = NULL;
+        if (bytes <= 0) { afc_unlock(client); return -1; }
+        
+        bytes = receive_AFC_data(client, &buffer);
+        if (buffer) free(buffer);
+        afc_unlock(client);
+        if (bytes >= 0) return 0;
+        else return -1;
+}
+
+int afc_truncate_file(AFClient *client, AFCFile *file, uint32 newsize) {
+        afc_lock(client);
+        
+        char *buffer = (char*)malloc(sizeof(char) * 16);
+        uint32 bytes = 0, zero = 0;
+        
+        memcpy(buffer, &file->filehandle, sizeof(uint32)); // handle
+        memcpy(buffer+4, &zero, sizeof(uint32)); // pad
+        memcpy(buffer+8, &newsize, sizeof(uint32)); // newsize
+        memcpy(buffer+12, &zero, 3); // pad
+        client->afc_packet->operation = AFC_FILE_TRUNCATE;
+        client->afc_packet->this_length = client->afc_packet->entire_length = 0;
+        bytes = dispatch_AFC_packet(client, buffer, 15);
+        free(buffer); buffer = NULL;
+        if (bytes <= 0) { afc_unlock(client); return -1; }
+        
+        bytes = receive_AFC_data(client, &buffer);
+        if (buffer) free(buffer);
+        afc_unlock(client);
+        if (bytes >= 0) return 0;
+        else return -1;
+}
+