ViewVC Help
View File | Revision Log | Show Annotations | Download File | View Changeset | Root Listing
root/public/ibx/branches/udr/client/IBUtils.pas
(Generate patch)

Comparing ibx/trunk/fbintf/IBUtils.pas (file contents):
Revision 47 by tony, Mon Jan 9 15:31:51 2017 UTC vs.
Revision 231 by tony, Mon Apr 16 08:32:21 2018 UTC

# Line 32 | Line 32
32   {************************************************************************}
33  
34   unit IBUtils;
35 + {$IFDEF MSWINDOWS}
36 + {$DEFINE WINDOWS}
37 + {$ENDIF}
38  
39   {$IFDEF FPC}
40   {$Mode Delphi}
41   {$codepage UTF8}
42 + {$define HASREQEX}
43   {$ENDIF}
44  
45 +
46   interface
47  
48 < uses
44 < {$IFDEF WINDOWS }
45 <  Windows,
46 < {$ELSE}
47 <  unix,
48 < {$ENDIF}
49 <  Classes, SysUtils;
48 > uses Classes, SysUtils, IB;
49  
50   const
51    CRLF = #13 + #10;
# Line 55 | Line 54 | const
54    TAB  = #9;
55    NULL_TERMINATOR = #0;
56  
57 <  sqlReservedWords: array [0..197] of string = (
57 >  sqlReservedWords: array [0..198] of string = (
58    'ADD',
59    'ADMIN',
60    'ALL',
# Line 147 | Line 146 | const
146    'INTO',
147    'IS',
148    'JOIN',
149 +  'KEY',
150    'LEADING',
151    'LEFT',
152    'LIKE',
# Line 258 | Line 258 | const
258  
259   function Max(n1, n2: Integer): Integer;
260   function Min(n1, n2: Integer): Integer;
261 < function RandomString(iLength: Integer): String;
261 > function RandomString(iLength: Integer): AnsiString;
262   function RandomInteger(iLow, iHigh: Integer): Integer;
263 < function StripString(st: String; CharsToStrip: String): String;
264 < function FormatIdentifier(Dialect: Integer; Value: String): String;
265 < function FormatIdentifierValue(Dialect: Integer; Value: String): String;
266 < function FormatIdentifierValueNC(Dialect: Integer; Value: String): String;
267 < function ExtractIdentifier(Dialect: Integer; Value: String): String;
268 < function QuoteIdentifier(Dialect: Integer; Value: String): String;
269 < function QuoteIdentifierIfNeeded(Dialect: Integer; Value: String): String;
270 < function Space2Underscore(s: string): string;
271 < function SQLSafeString(const s: string): string;
263 > function StripString(st: AnsiString; CharsToStrip: AnsiString): AnsiString;
264 > function ExtractIdentifier(Dialect: Integer; Value: AnsiString): AnsiString;
265 > function QuoteIdentifier(Dialect: Integer; Value: AnsiString): AnsiString;
266 > function QuoteIdentifierIfNeeded(Dialect: Integer; Value: AnsiString): AnsiString;
267 > function Space2Underscore(s: AnsiString): AnsiString;
268 > function SQLSafeString(const s: AnsiString): AnsiString;
269 > function IsSQLIdentifier(Value: AnsiString): boolean;
270 > function ExtractConnectString(const CreateSQL: AnsiString; var ConnectString: AnsiString): boolean;
271 > function MakeConnectString(ServerName, DatabaseName: AnsiString; Protocol: TProtocol;
272 >              PortNo: AnsiString = ''): AnsiString;
273 > function ParseConnectString(ConnectString: AnsiString;
274 >              var ServerName, DatabaseName: AnsiString; var Protocol: TProtocolAll;
275 >              var PortNo: AnsiString): boolean;
276 > function GetProtocol(ConnectString: AnsiString): TProtocolAll;
277  
278   implementation
279  
280 + {$IFDEF HASREQEX}
281 + uses RegExpr;
282 + {$ENDIF}
283 +
284   function Max(n1, n2: Integer): Integer;
285   begin
286    if (n1 > n2) then
# Line 288 | Line 297 | begin
297      result := n2;
298   end;
299  
300 < function RandomString(iLength: Integer): String;
300 > function RandomString(iLength: Integer): AnsiString;
301   begin
302    result := '';
303    while Length(result) < iLength do
# Line 302 | Line 311 | begin
311    result := Trunc(Random(iHigh - iLow)) + iLow;
312   end;
313  
314 < function StripString(st: String; CharsToStrip: String): String;
314 > function StripString(st: AnsiString; CharsToStrip: AnsiString): AnsiString;
315   var
316    i: Integer;
317   begin
# Line 313 | Line 322 | begin
322    end;
323   end;
324  
325 < function FormatIdentifier(Dialect: Integer; Value: String): String;
317 < begin
318 <  Value := Trim(Value);
319 <  if Dialect = 1 then
320 <    Value := AnsiUpperCase(Value)
321 <  else
322 <    if (Value <> '') and (Value[1] = '"') then
323 <      Value := '"' + StringReplace (TrimRight(Value), '"', '""', [rfReplaceAll]) + '"'
324 <    else
325 <      Value := AnsiUpperCase(Value);
326 <  Result := Value;
327 < end;
325 > {Extracts SQL Identifier typically from a  Dialect 3 encoding}
326  
327 < function FormatIdentifierValue(Dialect: Integer; Value: String): String;
327 > function ExtractIdentifier(Dialect: Integer; Value: AnsiString): AnsiString;
328   begin
329    Value := Trim(Value);
330    if Dialect = 1 then
# Line 345 | Line 343 | begin
343    Result := Value;
344   end;
345  
346 < function FormatIdentifierValueNC(Dialect: Integer; Value: String): String;
346 > {Returns true if "w" is a Firebird SQL reserved word}
347 >
348 > function IsReservedWord(w: AnsiString): boolean;
349 > var i: integer;
350 > begin
351 >     Result := true;
352 >     for i := 0 to Length(sqlReservedWords) - 1 do
353 >         if w = sqlReservedWords[i] then
354 >            Exit;
355 >     Result := false;
356 > end;
357 >
358 > {Format an SQL Identifier according to SQL Dialect}
359 >
360 > function QuoteIdentifier(Dialect: Integer; Value: AnsiString): AnsiString;
361   begin
350  Value := Trim(Value);
362    if Dialect = 1 then
363 <    Value := AnsiUpperCase(Value)
363 >    Value := AnsiUpperCase(Trim(Value))
364    else
365 +    Value := '"' + StringReplace (Value, '""', '"', [rfReplaceAll]) + '"';
366 +  Result := Value;
367 + end;
368 +
369 + const
370 +  ValidSQLIdentifierChars = ['A'..'Z','a'..'z','0'..'9','_','$'];
371 +
372 + {Returns true if the value is a valid SQL Identifier - note lower case accepted}
373 +
374 + function IsSQLIdentifier(Value: AnsiString): boolean;
375 + var i: integer;
376 + begin
377 +  Result := false;
378 +  for i := 1 to Length(Value) do
379 +    if not (Value[i] in ValidSQLIdentifierChars) then Exit;
380 +  Result := true;
381 + end;
382 +
383 + {Extracts the Database Connect string from a Create Database Statement}
384 +
385 + {$IFDEF HASREQEX}
386 + function ExtractConnectString(const CreateSQL: AnsiString;
387 +  var ConnectString: AnsiString): boolean;
388 + var RegexObj: TRegExpr;
389 + begin
390 +  RegexObj := TRegExpr.Create;
391 +  try
392 +    {extact database file spec}
393 +    RegexObj.ModifierG := false; {turn off greedy matches}
394 +    RegexObj.ModifierI := true; {case insensitive match}
395 +    RegexObj.Expression := '^ *CREATE +(DATABASE|SCHEMA) +''(.*)''';
396 +    Result := RegexObj.Exec(CreateSQL);
397 +    if Result then
398 +      ConnectString := RegexObj.Match[2];
399 +  finally
400 +    RegexObj.Free;
401 +  end;
402 + end;
403 +
404 + function ParseConnectString(ConnectString: AnsiString; var ServerName,
405 +  DatabaseName: AnsiString; var Protocol: TProtocolAll; var PortNo: AnsiString
406 +  ): boolean;
407 +
408 +  function GetProtocol(scheme: AnsiString): TProtocolAll;
409    begin
410 <    if (Value <> '') and (Value[1] = '"') then
410 >    scheme := AnsiUpperCase(scheme);
411 >    if scheme = 'INET' then
412 >      Result := inet
413 >    else
414 >    if scheme = 'INET4' then
415 >      Result := inet4
416 >    else
417 >    if scheme = 'INET6' then
418 >      Result := inet6
419 >    else
420 >    if scheme = 'XNET' then
421 >      Result := xnet
422 >    else
423 >    if scheme = 'WNET' then
424 >      Result := wnet
425 >  end;
426 >
427 > var RegexObj: TRegExpr;
428 > begin
429 >  ServerName := '';
430 >  DatabaseName := ConnectString;
431 >  PortNo := '';
432 >  Protocol := unknownProtocol;
433 >  RegexObj := TRegExpr.Create;
434 >  try
435 >    {extact database file spec}
436 >    RegexObj.ModifierG := false; {turn off greedy matches}
437 >    RegexObj.Expression := '^([a-zA-Z46]+)://([a-zA-Z0-9\-\.]*)(|:[0-9a-zA-Z\-]+)/(.*)$';
438 >    Result := RegexObj.Exec(ConnectString);
439 >    if Result then
440      begin
441 <      Delete(Value, 1, 1);
442 <      Delete(Value, Length(Value), 1);
443 <      Value := AnsiUpperCase(StringReplace (Value, '""', '"', [rfReplaceAll]));
441 >      {URL type connect string}
442 >      Protocol := GetProtocol(RegexObj.Match[1]);
443 >      ServerName := RegexObj.Match[2];
444 >      if RegexObj.MatchLen[3] > 0 then
445 >        PortNo := system.Copy(ConnectString,RegexObj.MatchPos[3]+1,RegexObj.MatchLen[3]-1);
446 >      DatabaseName := RegexObj.Match[4];
447 >      if ServerName = '' then
448 >        DatabaseName := '/' + DatabaseName;
449      end
450      else
451 <      Value := AnsiUpperCase(Value);
451 >    begin
452 >      {URL type connect string - local loop}
453 >      RegexObj.Expression := '^([a-zA-Z46]+)://(.*)$';
454 >      Result := RegexObj.Exec(ConnectString);
455 >      if Result then
456 >      begin
457 >        Protocol := GetProtocol(RegexObj.Match[1]);
458 >        DatabaseName := RegexObj.Match[2];
459 >      end
460 >      else
461 >      begin
462 >        RegexObj.Expression := '^([a-zA-Z]:\\.*)';
463 >        Result := RegexObj.Exec(ConnectString);
464 >        if Result then
465 >          Protocol := Local {Windows with leading drive ID}
466 >        else
467 >        begin
468 >          RegexObj.Expression := '^([a-zA-Z0-9\-\.]+)(|/[0-9a-zA-Z\-]+):(.*)$';
469 >          Result := RegexObj.Exec(ConnectString);
470 >          if Result then
471 >          begin
472 >            {Legacy TCP Format}
473 >            ServerName := RegexObj.Match[1];
474 >            if RegexObj.MatchLen[2] > 0 then
475 >              PortNo := system.Copy(ConnectString,RegexObj.MatchPos[2]+1,RegexObj.MatchLen[2]-1);
476 >            DatabaseName := RegexObj.Match[3];
477 >            Protocol := TCP;
478 >          end
479 >          else
480 >          begin
481 >            RegexObj.Expression := '^\\\\([a-zA-Z0-9\-\.]+)(|@[0-9a-zA-Z\-]+)\\(.*)$';
482 >            Result := RegexObj.Exec(ConnectString);
483 >            if Result then
484 >            begin
485 >              {Netbui}
486 >              ServerName := RegexObj.Match[1];
487 >              if RegexObj.MatchLen[2] > 0 then
488 >                PortNo := system.Copy(ConnectString,RegexObj.MatchPos[2]+1,RegexObj.MatchLen[2]-1);
489 >              DatabaseName := RegexObj.Match[3];
490 >              Protocol := NamedPipe
491 >            end
492 >            else
493 >            begin
494 >              Result := true;
495 >              Protocol := Local; {Assume local}
496 >            end;
497 >          end;
498 >        end;
499 >      end;
500 >    end;
501 >  finally
502 >    RegexObj.Free;
503    end;
364  Result := Value;
504   end;
505  
506 < function ExtractIdentifier(Dialect: Integer; Value: String): String;
506 > function GetProtocol(ConnectString: AnsiString): TProtocolAll;
507 > var ServerName,
508 >    DatabaseName: AnsiString;
509 >    PortNo: AnsiString;
510   begin
511 <  Value := Trim(Value);
512 <  if Dialect = 1 then
513 <    Value := AnsiUpperCase(Value)
514 <  else
511 >  ParseConnectString(ConnectString,ServerName,DatabaseName,Result,PortNo);
512 > end;
513 >
514 > {$ELSE}
515 > {cruder version of above for Delphi. Older versions lack regular expression
516 > handling.}
517 > function ExtractConnectString(const CreateSQL: AnsiString;
518 >  var ConnectString: AnsiString): boolean;
519 > var i: integer;
520 > begin
521 >  Result := false;
522 >  i := Pos('''',CreateSQL);
523 >  if i > 0 then
524    begin
525 <    if (Value <> '') and (Value[1] = '"') then
525 >    ConnectString := CreateSQL;
526 >    delete(ConnectString,1,i);
527 >    i := Pos('''',ConnectString);
528 >    if i > 0 then
529      begin
530 <      Delete(Value, 1, 1);
531 <      Delete(Value, Length(Value), 1);
532 <      Value := StringReplace (Value, '""', '"', [rfReplaceAll]);
379 <    end
380 <    else
381 <      Value := AnsiUpperCase(Value);
530 >      delete(ConnectString,i,Length(ConnectString)-i+1);
531 >      Result := true;
532 >    end;
533    end;
383  Result := Value;
534   end;
535  
536 < function IsReservedWord(w: string): boolean;
387 < var i: integer;
536 > function GetProtocol(ConnectString: AnsiString): TProtocolAll;
537   begin
538 <     Result := true;
390 <     for i := 0 to Length(sqlReservedWords) - 1 do
391 <         if w = sqlReservedWords[i] then
392 <            Exit;
393 <     Result := false;
538 >  Result := unknownProtocol; {not implemented for Delphi}
539   end;
540  
541 < function QuoteIdentifier(Dialect: Integer; Value: String): String;
541 > function ParseConnectString(ConnectString: AnsiString;
542 >              var ServerName, DatabaseName: AnsiString; var Protocol: TProtocolAll;
543 >              var PortNo: AnsiString): boolean;
544   begin
545 <  if Dialect = 1 then
546 <    Value := AnsiUpperCase(Trim(Value))
547 <  else
548 <    Value := '"' + Value + '"';
549 <  Result := Value;
545 >  Result := false;
546 > end;
547 >
548 > {$ENDIF}
549 >
550 > {Make a connect string in format appropriate protocol}
551 >
552 > function MakeConnectString(ServerName, DatabaseName: AnsiString;
553 >  Protocol: TProtocol; PortNo: AnsiString): AnsiString;
554 >
555 >  function FormatURL: AnsiString;
556 >  begin
557 >    if (ServerName = '') and (Pos('/',DatabaseName) <= 1) then
558 >      Result := DatabaseName
559 >    else
560 >      Result := ServerName + '/' + DatabaseName;
561 >  end;
562 >
563 > begin
564 >  if PortNo <> '' then
565 >    case Protocol of
566 >    NamedPipe:
567 >      ServerName := ServerName + '@' + PortNo;
568 >    Local,
569 >    SPX,
570 >    xnet: {do nothing};
571 >    TCP:
572 >      ServerName := ServerName + '/' + PortNo;
573 >    else
574 >      ServerName := ServerName + ':' + PortNo;
575 >    end;
576 >
577 >  case Protocol of
578 >    TCP:        Result := ServerName + ':' + DatabaseName; {do not localize}
579 >    SPX:        Result := ServerName + '@' + DatabaseName; {do not localize}
580 >    NamedPipe:  Result := '\\' + ServerName + '\' + DatabaseName; {do not localize}
581 >    Local:      Result := DatabaseName; {do not localize}
582 >    inet:       Result := 'inet://' + FormatURL; {do not localize}
583 >    inet4:       Result := 'inet4://' + FormatURL; {do not localize}
584 >    inet6:       Result := 'inet6://' + FormatURL; {do not localize}
585 >    wnet:       Result := 'wnet://' + FormatURL; {do not localize}
586 >    xnet:       Result := 'xnet://' + FormatURL;  {do not localize}
587 >  end;
588   end;
589  
590 < function QuoteIdentifierIfNeeded(Dialect: Integer; Value: String): String;
590 > {Format an SQL Identifier according to SQL Dialect with encapsulation if necessary}
591 >
592 > function QuoteIdentifierIfNeeded(Dialect: Integer; Value: AnsiString): AnsiString;
593   begin
594    if (Dialect = 3) and
595 <    ((AnsiUpperCase(Value) <> Value) or IsReservedWord(Value)) then
596 <     Result := '"' + Value + '"'
595 >    (IsReservedWord(Value) or not IsSQLIdentifier(Value) or (AnsiUpperCase(Value) <> Value)) then
596 >     Result := '"' + StringReplace (TrimRight(Value), '"', '""', [rfReplaceAll]) + '"'
597    else
598      Result := Value
599   end;
600  
601 < function Space2Underscore(s: string): string;
601 > {Replaces unknown characters in a string with underscores}
602 >
603 > function Space2Underscore(s: AnsiString): AnsiString;
604   var
605     k: integer;
606   begin
607       Result := s;
608       for k := 1 to Length(s) do
609 <         if not (Result[k] in ['0'..'9','A'..'Z','_','$'])  then
609 >         if not (Result[k] in ValidSQLIdentifierChars)  then
610              Result[k] := '_';
611   end;
612  
613 < function SQLSafeString(const s: string): string;
613 > {Reformats an SQL string with single quotes duplicated.}
614 >
615 > function SQLSafeString(const s: AnsiString): AnsiString;
616   begin
617    Result := StringReplace(s,'''','''''',[rfReplaceAll]);
618   end;

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines