opensource/HeidiSQL
1
0
Fork 0
mirror of https://github.com/HeidiSQL/HeidiSQL.git synced 2025-08-06 18:24:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add safety check for numeric history item name, so we don't run into follow up errors when there is some non-numeric history item. Fixes issue #2790.

Browse Source
This commit is contained in:
Ansgar Becker
2012-03-20 19:34:04 +00:00
parent 8a1fa0efc8
commit e9eb68b293
1 changed files with 14 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
source/main.pas
View File

@ -83,7 +83,7 @@ type
FMaxDuration: Cardinal;
public
property MaxDuration: Cardinal read FMaxDuration;
function ReadItem(RegValue: Integer): TQueryHistoryItem;
function ReadItem(RegValue: String): TQueryHistoryItem;
end;
TQueryHistoryItemComparer = class(TComparer<TQueryHistoryItem>)
function Compare(const Left, Right: TQueryHistoryItem): Integer; override;
@ -2469,7 +2469,7 @@ begin
MainReg.GetValueNames(AllRegItems);
History := TQueryHistory.Create;
for RegItem in AllRegItems do begin
History.ReadItem(StrToInt(RegItem));
History.ReadItem(RegItem);
end;
// Find lowest unused item number
@ -10365,7 +10365,7 @@ begin
MainReg.GetValueNames(Values);
History := TQueryHistory.Create;
for v in Values do begin
Item := History.ReadItem(StrToInt(v));
Item := History.ReadItem(v);
QueryDay := DateToStr(Item.Time);
if Tab.HistoryDays.IndexOf(QueryDay) = -1 then
Tab.HistoryDays.Add(QueryDay);
@ -10387,7 +10387,7 @@ begin
Values := TStringList.Create;
MainReg.GetValueNames(Values);
for v in Values do begin
Item := History.ReadItem(StrToInt(v));
Item := History.ReadItem(v);
QueryDay := DateToStr(Item.Time);
if QueryDay <> Tab.HistoryDays[Node.Index] then
History.Remove(Item);
@ -10857,14 +10857,20 @@ end;
{ TQueryHistory }
function TQueryHistory.ReadItem(RegValue: Integer): TQueryHistoryItem;
function TQueryHistory.ReadItem(RegValue: String): TQueryHistoryItem;
var
p: Integer;
i, p: Integer;
Raw: String;
begin
i := StrToIntDef(RegValue, -1);
// Prevent from running into serious errors when registry has some non-numeric value
if i=-1 then begin
Result := nil;
Exit;
end;
Result := TQueryHistoryItem.Create;
Result.RegValue := RegValue;
Raw := MainReg.ReadString(IntToStr(RegValue));
Result.RegValue := i;
Raw := MainReg.ReadString(RegValue);
p := Pos(DELIM, Raw);
Result.Time := StrToDateTime(Copy(Raw, 1, p-1));
System.Delete(Raw, 1, p);