-
-
Notifications
You must be signed in to change notification settings - Fork 135
/
kitty_commun.c
429 lines (388 loc) · 15 KB
/
kitty_commun.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
/*
* Fichier contenant les procedures communes à tous les programmes putty, pscp, psftp, plink, pageant
*/
#include "kitty_commun.h"
#include "kitty_tools.h"
// Flag permettant d'activer l'acces a du code particulier permettant d'avoir plus d'info dans le kitty.dmp
int debug_flag = 0 ;
#ifdef MOD_PERSO
#ifndef SAVEMODE_REG
#define SAVEMODE_REG 0
#endif
#ifndef SAVEMODE_FILE
#define SAVEMODE_FILE 1
#endif
#ifndef SAVEMODE_DIR
#define SAVEMODE_DIR 2
#endif
// Flag pour le fonctionnement en mode "portable" (gestion par fichiers)
#ifdef MOD_PORTABLE
int IniFileFlag = SAVEMODE_DIR ;
#else
int IniFileFlag = SAVEMODE_REG ;
#endif
int GetIniFileFlag(void) { return IniFileFlag ; }
void SetIniFileFlag( const int flag ) { IniFileFlag = flag ; }
void SwitchIniFileFlag(void) { if( IniFileFlag == SAVEMODE_REG ) { IniFileFlag = SAVEMODE_DIR ; } else if ( IniFileFlag == SAVEMODE_DIR ) { IniFileFlag = SAVEMODE_REG ; } }
// Flag permettant la gestion de l'arborscence (dossier=folder) dans le cas d'un savemode=dir
#ifdef MOD_PORTABLE
int DirectoryBrowseFlag = 1 ;
#else
int DirectoryBrowseFlag = 0 ;
#endif
// Flag pour repasser en mode Putty basic
int PuttyFlag = 0 ;
int GetPuttyFlag(void) { return PuttyFlag ; }
void SetPuttyFlag( const int flag ) { PuttyFlag = flag ; }
// Flag permettant de sauvegarder automatique les cles SSH des serveurs
static int AutoStoreSSHKeyFlag = 0 ;
int GetAutoStoreSSHKeyFlag(void) { return AutoStoreSSHKeyFlag ; }
void SetAutoStoreSSHKeyFlag( const int flag ) { AutoStoreSSHKeyFlag = flag ; }
// Flag permettant de desactiver la sauvegarde automatique des informations de connexion (user/password) Ã la connexion SSH
static int UserPassSSHNoSave = 0 ;
int GetUserPassSSHNoSave(void) { return UserPassSSHNoSave ; }
void SetUserPassSSHNoSave( const int flag ) { UserPassSSHNoSave = flag ; }
// Flag permettant de gérer la demande confirmation à l'usage d'une clé privée (1=always; 0=never; 2=based on "comment")
static int AskConfirmationFlag=2 ;
int GetAskConfirmationFlag(void) { return AskConfirmationFlag ; }
void SetAskConfirmationFlag( const int flag ) { AskConfirmationFlag = flag ; }
// Flag pour empêcher l'écriture des fichiers (default settings, jump file list ...)
// [KiTTY] readonly=no
static int ReadOnlyFlag = 0 ;
int GetReadOnlyFlag(void) { return ReadOnlyFlag ; }
void SetReadOnlyFlag( const int flag ) { ReadOnlyFlag = flag ; }
// Flag pour gérer le "mélange" des clés dans pageant
static int ScrumbleKeyFlag = 0 ;
int GetScrumbleKeyFlag(void) { return ScrumbleKeyFlag ; }
void SetScrumbleKeyFlag( const int flag ) { ScrumbleKeyFlag = flag ; }
#ifdef MOD_ADB
// Flag pour inhiber le support d'ADB
static int ADBFlag = 1 ;
int GetADBFlag(void) { return ADBFlag ; }
void SetADBFlag( const int flag ) { ADBFlag = flag ; }
#endif
#ifdef MOD_ZMODEM
// Flag pour inhiber les fonctions ZMODEM
static int ZModemFlag = 0 ;
int GetZModemFlag(void) { return ZModemFlag ; }
void SetZModemFlag( const int flag ) { ZModemFlag = flag ; }
#endif
// Flag pour afficher l'image de fond
#if (defined MOD_BACKGROUNDIMAGE) && (!defined FLJ)
// Suite à PuTTY 0.61, le patch covidimus ne fonctionne plus tres bien
// Il impose de demarrer les sessions avec -load meme depuis la config box (voir CONFIG.C)
// Le patch est desactive par defaut
int BackgroundImageFlag = 0 ;
#else
int BackgroundImageFlag = 0 ;
#endif
int GetBackgroundImageFlag(void) { return BackgroundImageFlag ; }
void SetBackgroundImageFlag( const int flag ) { BackgroundImageFlag = flag ; }
// Pour supprimer le salt dans la cryptography: issue: https://github.com/cyd01/KiTTY/issues/113
int RandomActiveFlag = 1 ;
int GetRandomActiveFlag() { return RandomActiveFlag ; }
void SetRandomActiveFlag( const int flag ) { RandomActiveFlag = flag ; }
// Pour supprimer le sel dans le cryptage du mot de passe
int CryptSaltFlag = 0 ;
int GetCryptSaltFlag() { return CryptSaltFlag ; }
void SetCryptSaltFlag( int flag ) { CryptSaltFlag = flag ; }
// Répertoire de sauvegarde de la configuration (savemode=dir)
char * ConfigDirectory = NULL ;
char * GetConfigDirectory( void ) { return ConfigDirectory ; }
int stricmp(const char *s1, const char *s2) ;
int readINI( const char * filename, const char * section, const char * key, char * pStr) ;
char * SetSessPath( const char * dec ) ;
// Nettoie les noms de folder en remplaçant les "/" par des "\" et les " \ " par des " \"
void CleanFolderName( char * folder ) {
int i, j ;
if( folder == NULL ) return ;
if( strlen( folder ) == 0 ) return ;
for( i=0 ; i<strlen(folder) ; i++ ) if( folder[i]=='/' ) folder[i]='\\' ;
for( i=0 ; i<(strlen(folder)-1) ; i++ )
if( folder[i]=='\\' )
while( folder[i+1]==' ' ) for( j=i+1 ; j<strlen(folder) ; j++ ) folder[j]=folder[j+1] ;
for( i=(strlen(folder)-1) ; i>0 ; i-- )
if( folder[i]=='\\' )
while( folder[i-1]==' ' ) {
for( j=i-1 ; j<strlen(folder) ; j++ ) folder[j]=folder[j+1] ;
i-- ;
}
}
#include <sys/types.h>
#include <dirent.h>
#define MAX_VALUE_NAME 16383
// Supprime une arborescence
int _rmdir( const char *dirname ) ;
void DelDir( const char * directory ) {
DIR * dir ;
struct dirent * de ;
char fullpath[MAX_VALUE_NAME] ;
if( (dir=opendir(directory)) != NULL ) {
while( (de=readdir( dir ) ) != NULL )
if( strcmp(de->d_name,".") && strcmp(de->d_name,"..") ) {
sprintf( fullpath, "%s\\%s", directory, de->d_name ) ;
if( GetFileAttributes( fullpath ) & FILE_ATTRIBUTE_DIRECTORY ) { DelDir( fullpath ) ; }
else if( !(GetFileAttributes( fullpath ) & FILE_ATTRIBUTE_DIRECTORY) ) { unlink( fullpath ) ; }
}
closedir( dir ) ;
_rmdir( directory ) ;
}
}
// Lit un parametre soit dans le fichier de configuration, soit dans le registre
char * IniFile = NULL ;
char INIT_SECTION[10];
int ReadParameterLight( const char * key, const char * name, char * value ) {
char buffer[4096] ;
strcpy( buffer, "" ) ;
if( GetValueData( HKEY_CURRENT_USER, TEXT(PUTTY_REG_POS), name, buffer ) == NULL ) {
if( !readINI( IniFile, key, name, buffer ) ) {
strcpy( buffer, "" ) ;
}
}
strcpy( value, buffer ) ;
return strcmp( buffer, "" ) ;
}
/* test if we are in portable mode by looking for putty.ini or kitty.ini in running directory */
int LoadParametersLight( void ) {
FILE * fp = NULL ;
int ret = 0 ;
char buffer[4096] ;
#ifndef FLJ
if( (getenv("KITTY_INI_FILE")!=NULL) && ((fp = fopen( getenv("KITTY_INI_FILE"), "r" )) != NULL) ) {
fclose(fp ) ;
IniFile = (char*)malloc(strlen(getenv("KITTY_INI_FILE"))+1) ;
strcpy( IniFile,getenv("KITTY_INI_FILE") ) ;
strcpy(INIT_SECTION,"KiTTY");
if( readINI( IniFile, "KiTTY", "savemode", buffer ) ) {
while( (buffer[strlen(buffer)-1]=='\n')||(buffer[strlen(buffer)-1]=='\r')
||(buffer[strlen(buffer)-1]==' ')
||(buffer[strlen(buffer)-1]=='\t') ) buffer[strlen(buffer)-1]='\0';
if( !stricmp( buffer, "registry" ) ) IniFileFlag = SAVEMODE_REG ;
else if( !stricmp( buffer, "file" ) ) IniFileFlag = SAVEMODE_FILE ;
else if( !stricmp( buffer, "dir" ) ) { IniFileFlag = SAVEMODE_DIR ; ret = 1 ; }
}
if( IniFileFlag == SAVEMODE_DIR ) {
if( readINI( IniFile, "KiTTY", "browsedirectory", buffer ) ) {
if( !stricmp( buffer, "NO" )&&(IniFileFlag==SAVEMODE_DIR) ) DirectoryBrowseFlag = 0 ;
else DirectoryBrowseFlag = 1 ;
}
if( readINI( IniFile, "KiTTY", "configdir", buffer ) ) {
if( strlen( buffer ) > 0 ) {
ConfigDirectory = (char*)malloc( strlen(buffer) + 1 ) ;
strcpy( ConfigDirectory, buffer ) ;
}
}
} else DirectoryBrowseFlag = 0 ;
} else if( (fp = fopen( "kitty.ini", "r" )) != NULL ) {
IniFile = (char*)malloc(11) ; strcpy(IniFile,"kitty.ini");
strcpy(INIT_SECTION,"KiTTY");
fclose(fp ) ;
if( readINI( "kitty.ini", "KiTTY", "savemode", buffer ) ) {
while( (buffer[strlen(buffer)-1]=='\n')||(buffer[strlen(buffer)-1]=='\r')
||(buffer[strlen(buffer)-1]==' ')
||(buffer[strlen(buffer)-1]=='\t') ) buffer[strlen(buffer)-1]='\0';
if( !stricmp( buffer, "registry" ) ) IniFileFlag = SAVEMODE_REG ;
else if( !stricmp( buffer, "file" ) ) IniFileFlag = SAVEMODE_FILE ;
else if( !stricmp( buffer, "dir" ) ) { IniFileFlag = SAVEMODE_DIR ; ret = 1 ; }
}
if( IniFileFlag == SAVEMODE_DIR ) {
if( readINI( "kitty.ini", "KiTTY", "browsedirectory", buffer ) ) {
if( !stricmp( buffer, "NO" )&&(IniFileFlag==SAVEMODE_DIR) ) DirectoryBrowseFlag = 0 ;
else DirectoryBrowseFlag = 1 ;
}
if( readINI( "kitty.ini", "KiTTY", "configdir", buffer ) ) {
if( strlen( buffer ) > 0 ) {
ConfigDirectory = (char*)malloc( strlen(buffer) + 1 ) ;
strcpy( ConfigDirectory, buffer ) ;
}
}
} else DirectoryBrowseFlag = 0 ;
} else
#endif
if( (fp = fopen( "putty.ini", "r" )) != NULL ) {
IniFile = (char*)malloc(11) ; strcpy(IniFile,"putty.ini");
strcpy(INIT_SECTION,"PuTTY");
fclose(fp ) ;
if( readINI( "putty.ini", "PuTTY", "savemode", buffer ) ) {
while( (buffer[strlen(buffer)-1]=='\n')||(buffer[strlen(buffer)-1]=='\r')
||(buffer[strlen(buffer)-1]==' ')
||(buffer[strlen(buffer)-1]=='\t') ) buffer[strlen(buffer)-1]='\0';
if( !stricmp( buffer, "registry" ) ) IniFileFlag = SAVEMODE_REG ;
else if( !stricmp( buffer, "file" ) ) IniFileFlag = SAVEMODE_FILE ;
else if( !stricmp( buffer, "dir" ) ) { IniFileFlag = SAVEMODE_DIR ; DirectoryBrowseFlag = 1 ; ret = 1 ; }
}
if( IniFileFlag == SAVEMODE_DIR ) {
if( readINI( "putty.ini", "PuTTY", "browsedirectory", buffer ) ) {
if( !stricmp( buffer, "NO" )&&(IniFileFlag==SAVEMODE_DIR) ) DirectoryBrowseFlag = 0 ;
else DirectoryBrowseFlag = 1 ;
}
if( readINI( "putty.ini", "PuTTY", "configdir", buffer ) ) {
if( strlen( buffer ) > 0 ) {
ConfigDirectory = (char*)malloc( strlen(buffer) + 1 ) ;
strcpy( ConfigDirectory, buffer ) ;
}
}
} else DirectoryBrowseFlag = 0 ;
} else {
#ifndef FLJ
sprintf( buffer, "%s/KiTTY/kitty.ini", getenv("APPDATA") );
if( (fp = fopen( buffer, "r" )) != NULL ) {
IniFile = (char*)malloc(strlen(buffer)+1) ;
strcpy(IniFile,buffer);
strcpy(INIT_SECTION,"KiTTY");
fclose(fp);
} else {
#endif
sprintf( buffer, "%s/PuTTY/putty.ini", getenv("APPDATA") );
if( (fp = fopen( buffer, "r" )) != NULL ) {
IniFile = (char*)malloc(strlen(buffer)+1) ;
strcpy(IniFile,buffer);
strcpy(INIT_SECTION,"PuTTY");
fclose(fp);
}
#ifndef FLJ
}
#endif
}
if( ReadParameterLight( INIT_SECTION, "fileextension", buffer ) ) {
if( strlen(buffer) > 0 ) {
if( buffer[0] != '.' ) { strcpy( FileExtension, "." ) ; } else { strcpy( FileExtension, "" ) ; }
strcat( FileExtension, buffer ) ;
while( FileExtension[strlen(FileExtension)-1]==' ' ) { FileExtension[strlen(FileExtension)-1] = '\0' ; }
}
}
if( ReadParameterLight( INIT_SECTION, "autostoresshkey", buffer ) ) { if( !stricmp( buffer, "YES" ) ) SetAutoStoreSSHKeyFlag( 1 ) ; }
if( ReadParameterLight( "Agent", "messageonkeyusage", buffer ) ) { if( !stricmp( buffer, "YES" ) ) SetShowBalloonOnKeyUsage() ; }
if( ReadParameterLight( "Agent", "askconfirmation", buffer ) ) {
if( !stricmp( buffer, "YES" ) ) SetAskConfirmationFlag(1) ;
if( !stricmp( buffer, "NO" ) ) SetAskConfirmationFlag(0) ;
if( !stricmp( buffer, "AUTO" ) ) SetAskConfirmationFlag(2) ;
}
if( ReadParameterLight( "Agent", "scrumble", buffer ) ) { if( !stricmp( buffer, "YES" ) ) SetScrumbleKeyFlag(1) ; }
return ret ;
}
// Positionne un flag permettant de determiner si on est connecte
int is_backend_connected = 0 ;
#ifdef MOD_RECONNECT
int is_backend_first_connected = 0 ;
void SetSSHConnected( int flag ) {
is_backend_connected = flag ;
if( flag ) is_backend_first_connected = 1 ;
}
#else
void SetSSHConnected( int flag ) {
is_backend_connected = flag ;
}
#endif
//PVOID SecureZeroMemory( PVOID ptr, SIZE_T cnt) { return memset( ptr, 0, cnt ) ; }
// Fonction permettant de changer le statut du stockage automatique des ssh host keys
void SetAutoStoreSSHKey( void ) {
AutoStoreSSHKeyFlag = 1 ;
}
/* Fonction permettant de changer le statut de la generation d'un affichage de balloon dans le system tray lorsqu'une clé privée est utilisée */
static int PrintBalloonOnKeyUsageFlag = 0 ;
void SetShowBalloonOnKeyUsage( void ) {
PrintBalloonOnKeyUsageFlag = 1 ;
}
int GetShowBalloonOnKeyUsage( void ) { return PrintBalloonOnKeyUsageFlag ; }
/* Fonction permettant d'affiche un message sur l'icone dans le tray */
char *strncpy(
char *strDest,
const char * strSource,
size_t count
);
int ShowBalloonTip( NOTIFYICONDATA tnid, TCHAR title[], TCHAR msg[] ) {
if( PrintBalloonOnKeyUsageFlag==0 ) return 0 ;
BOOL res;
//NOTIFYICONDATA tnid;
//tnid.cbSize = sizeof(NOTIFYICONDATA) ;
//tnid.hWnd = hwnd ;
tnid.uFlags = NIF_INFO ;
tnid.dwInfoFlags = NIIF_INFO ;
tnid.uTimeout = 3000 ; /*timeout*/
strncpy( tnid.szInfo, msg, sizeof( tnid.szInfo ) );
strncpy( tnid.szInfoTitle, title, sizeof( tnid.szInfoTitle ) );
res = Shell_NotifyIcon(NIM_MODIFY, &tnid);
Sleep(3000);
tnid.szInfo[0] = 0 ;
res = Shell_NotifyIcon(NIM_MODIFY, &tnid);
return res ;
}
// Fonctions permettant de formatter les chaînes de caractères avec %XY
void mungestr( const char *in, char *out ) {
char hex[16] = "0123456789ABCDEF";
int candot = 0 ;
while( *in ) {
if( *in == ' ' || *in == '\\' || *in == '*' || *in == '?' ||
*in ==':' || *in =='/' || *in =='\"' || *in =='<' || *in =='>' || *in =='|' ||
*in == '%' || *in < ' ' || *in > '~' || (*in == '.'
&& !candot) ) {
*out++ = '%' ;
*out++ = hex[((unsigned char) *in) >> 4] ;
*out++ = hex[((unsigned char) *in) & 15] ;
} else {
*out++ = *in ;
}
in++ ;
candot = 1 ;
}
*out = '\0' ;
return ;
}
void unmungestr( const char *in, char *out, int outlen ) {
while (*in) {
if( *in == '%' && in[1] && in[2] ) {
int i, j ;
i = in[1] - '0' ;
i -= (i > 9 ? 7 : 0) ;
j = in[2] - '0' ;
j -= (j > 9 ? 7 : 0) ;
*out++ = (i << 4) + j ;
if( !--outlen )
return ;
in += 3 ;
} else {
*out++ = *in++ ;
if( !--outlen )
return;
}
}
*out = '\0' ;
return ;
}
// Fonctions de gestion du mot de passe
extern Conf *conf;
void GetPasswordInConfig( char * p ) {
if( strlen(conf_get_str(conf,CONF_password)) == 0 ) return ;
/* On decrypte le password */
char bufpass[4096] ;
int len = strlen( conf_get_str(conf,CONF_password) ) ;
if( len>4095 ) { len = 4095 ; }
memcpy( bufpass, conf_get_str(conf,CONF_password), len ) ;
bufpass[len]='\0';
MASKPASS(GetCryptSaltFlag(),bufpass) ;
//DebugAddPassword( "GetPasswordInConfig", bufpass ) ; // in settings.c
memcpy( p, bufpass, strlen(bufpass)+1 ) ;
memset( bufpass,0,strlen(bufpass) ) ;
}
int IsPasswordInConf(void) {
int len = 0 ;
if( strlen(conf_get_str(conf,CONF_password)) == 0 ) return 0 ;
/* On decrypte le password */
char bufpass[4096] ;
len = strlen( conf_get_str(conf,CONF_password) ) ;
if( len>4095 ) { len = 4095 ; }
memcpy( bufpass, conf_get_str(conf,CONF_password), len ) ;
bufpass[len]='\0';
MASKPASS(GetCryptSaltFlag(),bufpass);
//DebugAddPassword( "IsPasswordInConf", bufpass ) ; // in settings.c
len = strlen( bufpass ) ;
memset(bufpass,0,strlen(bufpass));
return len ;
}
void CleanPassword( char * p ) { memset(p,0,strlen(p)); }
// Extention pour les fichiers de session en mode portable (peut être ktx)
char FileExtension[15] = "" ;
// Répertoire courant pourle mode portable
char CurrentFolder[1024] = "Default" ;
#endif