|
@@ -0,0 +1,227 @@
|
|
|
+//---------------------------------------------------------------------------
|
|
|
+// __________________ _________ _____ _____ .__ ._.
|
|
|
+// \______ \______ \ / _____/ / \ / _ \ |__| ____ | |
|
|
|
+// | | _/| | \ \_____ \ / \ / \ / /_\ \| _/ __ \ | |
|
|
|
+// | | \| ` \/ / Y \ / | | \ ___/ \|
|
|
|
+// |______ /_______ /_______ \____|__ / /\ \____|__ |__|\___ | __
|
|
|
+// \/ \/ \/ \/ )/ \/ \/ \/
|
|
|
+//
|
|
|
+// This file is part of libdsm. Copyright © 2014 VideoLabs SAS
|
|
|
+//
|
|
|
+// Author: Julien 'Lta' BALLET <contact@lta.io>
|
|
|
+//
|
|
|
+// This program is free software. It comes without any warranty, to the extent
|
|
|
+// permitted by applicable law. You can redistribute it and/or modify it under
|
|
|
+// the terms of the Do What The Fuck You Want To Public License, Version 2, as
|
|
|
+// published by Sam Hocevar. See the COPYING file for more details.
|
|
|
+//----------------------------------------------------------------------------
|
|
|
+
|
|
|
+#include <stdlib.h>
|
|
|
+#include <string.h>
|
|
|
+#include <stdio.h>
|
|
|
+#include <assert.h>
|
|
|
+
|
|
|
+#include "bdsm/smb_trans2.h"
|
|
|
+
|
|
|
+static smb_file_t *smb_find_parse(smb_message_t *msg)
|
|
|
+{
|
|
|
+ smb_trans2_resp_t *tr2;
|
|
|
+ smb_tr2_find2_params_t *params;
|
|
|
+ smb_tr2_find2_entry_t *iter;
|
|
|
+ smb_file_t *files, *tmp;
|
|
|
+ uint8_t *eod;
|
|
|
+ size_t count, i;
|
|
|
+
|
|
|
+ assert(msg != NULL);
|
|
|
+
|
|
|
+ // Let's parse the answer we got from server
|
|
|
+ tr2 = (smb_trans2_resp_t *)msg->packet->payload;
|
|
|
+ params = (smb_tr2_find2_params_t *)tr2->payload;
|
|
|
+ iter = (smb_tr2_find2_entry_t *)(tr2->payload + sizeof(smb_tr2_find2_params_t));
|
|
|
+ eod = msg->packet->payload + msg->payload_size;
|
|
|
+ count = params->count;
|
|
|
+ files = tmp = NULL;
|
|
|
+
|
|
|
+ for(i = 0; i < count && (uint8_t *)iter < eod; i++)
|
|
|
+ {
|
|
|
+ tmp = calloc(1, sizeof(smb_file_t));
|
|
|
+ assert(tmp != NULL);
|
|
|
+
|
|
|
+ tmp->name_len = iter->name_len + 2;
|
|
|
+ tmp->name = malloc(tmp->name_len);
|
|
|
+ assert(tmp->name != NULL);
|
|
|
+ memcpy(tmp->name, iter->name, tmp->name_len - 2);
|
|
|
+ tmp->name[tmp->name_len - 1] = tmp->name[tmp->name_len - 2] = 0;
|
|
|
+
|
|
|
+ tmp->created = iter->created;
|
|
|
+ tmp->accessed = iter->accessed;
|
|
|
+ tmp->written = iter->written;
|
|
|
+ tmp->changed = iter->changed;
|
|
|
+ tmp->size = iter->size;
|
|
|
+ tmp->alloc_size = iter->alloc_size;
|
|
|
+ tmp->attr = iter->attr;
|
|
|
+ tmp->is_dir = tmp->attr & SMB_ATTR_DIR;
|
|
|
+
|
|
|
+ tmp->next = files;
|
|
|
+ files = tmp;
|
|
|
+
|
|
|
+ iter = (smb_tr2_find2_entry_t *)(((char *)iter) + iter->next_entry);
|
|
|
+ }
|
|
|
+
|
|
|
+ return (files);
|
|
|
+}
|
|
|
+
|
|
|
+smb_file_t *smb_find(smb_session_t *s, smb_tid tid, const char *pattern)
|
|
|
+{
|
|
|
+ smb_message_t *msg, reply;
|
|
|
+ smb_trans2_req_t *tr2;
|
|
|
+ smb_tr2_find2_t *find;
|
|
|
+ size_t pattern_len, msg_len;
|
|
|
+ int res;
|
|
|
+
|
|
|
+ assert(s != NULL && pattern != NULL && tid);
|
|
|
+
|
|
|
+ pattern_len = strlen(pattern) + 1;
|
|
|
+ msg_len = sizeof(smb_trans2_req_t) + sizeof(smb_tr2_find2_t);
|
|
|
+ msg_len += pattern_len * 2 + 3;
|
|
|
+
|
|
|
+ msg = smb_message_new(SMB_CMD_TRANS2, msg_len);
|
|
|
+ smb_message_set_default_flags(msg);
|
|
|
+ msg->packet->header.tid = tid;
|
|
|
+
|
|
|
+ tr2 = (smb_trans2_req_t *)msg->packet->payload;
|
|
|
+ tr2->wct = 15;
|
|
|
+ tr2->total_param_count = pattern_len * 2 + sizeof(smb_tr2_find2_t);
|
|
|
+ tr2->max_param_count = 10; // ?? Why not the same or 12 ?
|
|
|
+ tr2->max_data_count = 0xffff;
|
|
|
+ tr2->param_count = tr2->total_param_count;
|
|
|
+ tr2->param_offset = 68; // Offset of find_first_params in packet;
|
|
|
+ tr2->data_count = 0;
|
|
|
+ tr2->data_offset = 88; // Offset of pattern in packet
|
|
|
+ tr2->setup_count = 1;
|
|
|
+ tr2->cmd = SMB_TR2_FIND_FIRST;
|
|
|
+ tr2->bct = sizeof(smb_tr2_find2_t) + pattern_len * 2 + 3;
|
|
|
+
|
|
|
+ find = (smb_tr2_find2_t *) tr2->payload;
|
|
|
+ find->attrs = SMB_FIND2_ATTR_DEFAULT;
|
|
|
+ find->count = 1366; // ??
|
|
|
+ find->flags = SMB_FIND2_FLAG_DEFAULT;
|
|
|
+ find->interest = 0x0104; // 'Find file both directory info'
|
|
|
+
|
|
|
+ smb_message_advance(msg, sizeof(smb_trans2_req_t));
|
|
|
+ smb_message_advance(msg, sizeof(smb_tr2_find2_t));
|
|
|
+ smb_message_put_utf16(msg, "", pattern, pattern_len);
|
|
|
+
|
|
|
+ // Adds padding at the end if necessary.
|
|
|
+ if (tr2->bct % 4)
|
|
|
+ smb_message_advance(msg, 4 - tr2->bct % 4);
|
|
|
+
|
|
|
+ res = smb_session_send_msg(s, msg);
|
|
|
+ smb_message_destroy(msg);
|
|
|
+ if (!res)
|
|
|
+ {
|
|
|
+ if (BDSM_DEBUG)
|
|
|
+ fprintf(stderr, "Unable to query pattern: %s\n", pattern);
|
|
|
+ return (NULL);
|
|
|
+ }
|
|
|
+
|
|
|
+ if (!smb_session_recv_msg(s, &reply)
|
|
|
+ || reply.packet->header.status != NT_STATUS_SUCCESS)
|
|
|
+ {
|
|
|
+ if (BDSM_DEBUG)
|
|
|
+ fprintf(stderr, "Unable to recv msg or failure for %s\n", pattern);
|
|
|
+ return (NULL);
|
|
|
+ }
|
|
|
+
|
|
|
+ return (smb_find_parse(&reply));
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+smb_file_t *smb_stat(smb_session_t *s, smb_tid tid, const char *path)
|
|
|
+{
|
|
|
+ smb_message_t *msg, reply;
|
|
|
+ smb_trans2_req_t *tr2;
|
|
|
+ smb_trans2_resp_t *tr2_resp;
|
|
|
+ smb_tr2_query_t *query;
|
|
|
+ smb_tr2_path_info_t *info;
|
|
|
+ smb_file_t *file;
|
|
|
+ size_t path_len, msg_len;
|
|
|
+ int res;
|
|
|
+
|
|
|
+ assert(s != NULL && path != NULL && tid);
|
|
|
+
|
|
|
+ path_len = strlen(path) + 1;
|
|
|
+ msg_len = sizeof(smb_trans2_req_t) + sizeof(smb_tr2_query_t);
|
|
|
+ msg_len += path_len * 2 + 3; // +3 for eventual padding
|
|
|
+
|
|
|
+ msg = smb_message_new(SMB_CMD_TRANS2, msg_len);
|
|
|
+ smb_message_set_default_flags(msg);
|
|
|
+ msg->packet->header.tid = tid;
|
|
|
+
|
|
|
+ tr2 = (smb_trans2_req_t *)msg->packet->payload;
|
|
|
+ tr2->wct = 15;
|
|
|
+ tr2->total_param_count = path_len * 2 + sizeof(smb_tr2_query_t);
|
|
|
+ tr2->max_param_count = 2; // ?? Why not the same or 12 ?
|
|
|
+ tr2->max_data_count = 0xffff;
|
|
|
+ tr2->param_count = tr2->total_param_count;
|
|
|
+ tr2->param_offset = 68; // Offset of find_first_params in packet;
|
|
|
+ tr2->data_count = 0;
|
|
|
+ tr2->data_offset = 96; // Offset of pattern in packet
|
|
|
+ tr2->setup_count = 1;
|
|
|
+ tr2->cmd = SMB_TR2_QUERY_PATH;
|
|
|
+ tr2->bct = sizeof(smb_tr2_query_t) + path_len * 2 + 3;
|
|
|
+
|
|
|
+ query = (smb_tr2_query_t *)tr2->payload;
|
|
|
+ query->interest = 0x0107; // Query File All Info
|
|
|
+
|
|
|
+ smb_message_advance(msg, sizeof(smb_trans2_req_t));
|
|
|
+ smb_message_advance(msg, sizeof(smb_tr2_query_t));
|
|
|
+ smb_message_put_utf16(msg, "", path, path_len);
|
|
|
+
|
|
|
+ // Adds padding at the end if necessary.
|
|
|
+ if (msg->cursor % 4)
|
|
|
+ {
|
|
|
+ int padding = 4 - msg->cursor % 4;
|
|
|
+ tr2->bct += padding;
|
|
|
+ for (int i = 0; i < padding; i++)
|
|
|
+ smb_message_put8(msg, 0);
|
|
|
+ }
|
|
|
+
|
|
|
+ res = smb_session_send_msg(s, msg);
|
|
|
+ smb_message_destroy(msg);
|
|
|
+ if (!res)
|
|
|
+ {
|
|
|
+ if (BDSM_DEBUG)
|
|
|
+ fprintf(stderr, "Unable to query pattern: %s\n", path);
|
|
|
+ return (NULL);
|
|
|
+ }
|
|
|
+
|
|
|
+ if (!smb_session_recv_msg(s, &reply)
|
|
|
+ || reply.packet->header.status != NT_STATUS_SUCCESS)
|
|
|
+ {
|
|
|
+ if (BDSM_DEBUG)
|
|
|
+ fprintf(stderr, "Unable to recv msg or failure for %s\n", path);
|
|
|
+ return (NULL);
|
|
|
+ }
|
|
|
+
|
|
|
+ tr2_resp = (smb_trans2_resp_t *)reply.packet->payload;
|
|
|
+ info = (smb_tr2_path_info_t *)tr2_resp->payload;
|
|
|
+ file = calloc(1, sizeof(smb_file_t));
|
|
|
+ assert(file != NULL);
|
|
|
+
|
|
|
+ file->name_len = info->name_len;
|
|
|
+ file->name = malloc(info->name_len);
|
|
|
+ assert(file->name != NULL);
|
|
|
+ memcpy(file->name, info->name, file->name_len);
|
|
|
+
|
|
|
+ file->created = info->created;
|
|
|
+ file->accessed = info->accessed;
|
|
|
+ file->written = info->written;
|
|
|
+ file->changed = info->changed;
|
|
|
+ file->alloc_size = info->alloc_size;
|
|
|
+ file->size = info->size;
|
|
|
+ file->attr = info->attr;
|
|
|
+ file->is_dir = info->is_dir;
|
|
|
+
|
|
|
+ return (file);
|
|
|
+}
|