ls.c (4002B)
1/* 2 * Licensed to the Apache Software Foundation (ASF) under one 3 * or more contributor license agreements. See the NOTICE file 4 * distributed with this work for additional information 5 * regarding copyright ownership. The ASF licenses this file 6 * to you under the Apache License, Version 2.0 (the 7 * "License"); you may not use this file except in compliance 8 * with the License. You may obtain a copy of the License at 9 * 10 * http://www.apache.org/licenses/LICENSE-2.0 11 * 12 * Unless required by applicable law or agreed to in writing, 13 * software distributed under the License is distributed on an 14 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY 15 * KIND, either express or implied. See the License for the 16 * specific language governing permissions and limitations 17 * under the License. 18 */ 19 20#include "fs.h" 21#include "ls.h" 22 23#include <guacamole/client.h> 24#include <guacamole/mem.h> 25#include <guacamole/protocol.h> 26#include <guacamole/socket.h> 27#include <guacamole/stream.h> 28#include <guacamole/user.h> 29#include <winpr/file.h> 30#include <winpr/nt.h> 31#include <winpr/shell.h> 32 33#include <stdlib.h> 34#include <string.h> 35 36int guac_rdp_ls_ack_handler(guac_user* user, guac_stream* stream, 37 char* message, guac_protocol_status status) { 38 39 int blob_written = 0; 40 const char* filename; 41 42 guac_rdp_ls_status* ls_status = (guac_rdp_ls_status*) stream->data; 43 44 /* If unsuccessful, free stream and abort */ 45 if (status != GUAC_PROTOCOL_STATUS_SUCCESS) { 46 guac_rdp_fs_close(ls_status->fs, ls_status->file_id); 47 guac_user_free_stream(user, stream); 48 guac_mem_free(ls_status); 49 return 0; 50 } 51 52 /* While directory entries remain */ 53 while ((filename = guac_rdp_fs_read_dir(ls_status->fs, 54 ls_status->file_id)) != NULL 55 && !blob_written) { 56 57 char absolute_path[GUAC_RDP_FS_MAX_PATH]; 58 59 /* Skip current and parent directory entries */ 60 if (strcmp(filename, ".") == 0 || strcmp(filename, "..") == 0) 61 continue; 62 63 /* Concatenate into absolute path - skip if invalid */ 64 if (!guac_rdp_fs_append_filename(absolute_path, 65 ls_status->directory_name, filename)) { 66 67 guac_user_log(user, GUAC_LOG_DEBUG, 68 "Skipping filename \"%s\" - filename is invalid or " 69 "resulting path is too long", filename); 70 71 continue; 72 } 73 74 /* Attempt to open file to determine type */ 75 int file_id = guac_rdp_fs_open(ls_status->fs, absolute_path, 76 GENERIC_READ, 0, FILE_OPEN, 0); 77 if (file_id < 0) 78 continue; 79 80 /* Get opened file */ 81 guac_rdp_fs_file* file = guac_rdp_fs_get_file(ls_status->fs, file_id); 82 if (file == NULL) { 83 guac_user_log(user, GUAC_LOG_DEBUG, "%s: Successful open produced " 84 "bad file_id: %i", __func__, file_id); 85 return 0; 86 } 87 88 /* Determine mimetype */ 89 const char* mimetype; 90 if (file->attributes & FILE_ATTRIBUTE_DIRECTORY) 91 mimetype = GUAC_USER_STREAM_INDEX_MIMETYPE; 92 else 93 mimetype = "application/octet-stream"; 94 95 /* Write entry */ 96 blob_written |= guac_common_json_write_property(user, stream, 97 &ls_status->json_state, absolute_path, mimetype); 98 99 guac_rdp_fs_close(ls_status->fs, file_id); 100 101 } 102 103 /* Complete JSON and cleanup at end of directory */ 104 if (filename == NULL) { 105 106 /* Complete JSON object */ 107 guac_common_json_end_object(user, stream, &ls_status->json_state); 108 guac_common_json_flush(user, stream, &ls_status->json_state); 109 110 /* Clean up resources */ 111 guac_rdp_fs_close(ls_status->fs, ls_status->file_id); 112 guac_mem_free(ls_status); 113 114 /* Signal of stream */ 115 guac_protocol_send_end(user->socket, stream); 116 guac_user_free_stream(user, stream); 117 118 } 119 120 guac_socket_flush(user->socket); 121 return 0; 122 123} 124