iframe-proxy | Sunbelt Computer Software

History

172 lines (151 loc) · 6.55 KB

Raw

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

using System.Text;

using System.Text.Json.Nodes;

using Microsoft.Extensions.Primitives;

using Npgsql;

namespace NpgsqlRest;

public class NpgsqlRestParameter : NpgsqlParameter

{

public int Ordinal { get; private set; }

public string ConvertedName { get; internal set; }

public string ActualName { get; internal set; }

/// <summary>

/// The original PostgreSQL parameter name as defined in the function/source.

/// Never changed by annotations. Used by RoutineSourceParameterFormatter for SQL generation.

/// </summary>

public string OriginalName { get; init; }

/// <summary>

/// Virtual parameters exist for HTTP matching and claim mapping but are NOT bound

/// to the PostgreSQL command. Created by @define_param annotation on SQL file endpoints.

/// </summary>

public bool IsVirtual { get; init; }

public TypeDescriptor TypeDescriptor { get; set; }

/// <summary>

/// For HTTP custom type composite parameters, the field names in order.

/// Used to build the composite text value without depending on CompositeTypeCache.

/// </summary>

public string[]? CompositeFieldNames { get; set; }

public ParamType ParamType { get; set; } = default!;

public StringValues? QueryStringValues { get; set; } = null;

public JsonNode? JsonBodyNode { get; set; } = null;

public NpgsqlRestParameter? HashOf { get; set; } = null;

/// <summary>

/// The original string representation of the parameter value as received from the request

/// (query string or JSON body). Used for cache key generation to ensure consistency.

/// </summary>

public string? OriginalStringValue { get; set; } = null;

public bool IsUploadMetadata { get; set; } = false;

/// <summary>

/// Explicit default value set by annotation (e.g., @param my_param default null).

/// null = no explicit default (use existing HasDefault/PostgreSQL behavior).

/// DBNull.Value = default is SQL NULL.

/// Any other value = the default value as a string to be bound.

/// </summary>

public object? DefaultValue { get; set; } = null;

public bool IsIpAddress { get; set; } = false;

public string? UserClaim { get; set; } = null;

public bool IsUserClaims { get; set; } = false;

public bool IsFromUserClaims => UserClaim is not null || IsUserClaims is true;

public NpgsqlRestParameter(

int ordinal,

string convertedName,

string actualName,

TypeDescriptor typeDescriptor)

{

Ordinal = ordinal;

ConvertedName = convertedName;

OriginalName = actualName;

ActualName = actualName;

TypeDescriptor = typeDescriptor;

NpgsqlDbType = typeDescriptor.ActualDbType;

if (actualName is not null &&

Options.AuthenticationOptions.ParameterNameClaimsMapping.TryGetValue(actualName, out var claimName))

{

UserClaim = claimName;

}

if (actualName is not null &&

string.Equals(Options.AuthenticationOptions.IpAddressParameterName, actualName, StringComparison.OrdinalIgnoreCase))

{

IsIpAddress = true;

}

if (actualName is not null &&

string.Equals(Options.AuthenticationOptions.ClaimsJsonParameterName, actualName, StringComparison.OrdinalIgnoreCase))

{

IsUserClaims = true;

}

if (Options.UploadOptions.UseDefaultUploadMetadataParameter is true)

{

if (string.Equals(Options.UploadOptions.DefaultUploadMetadataParameterName, actualName, StringComparison.OrdinalIgnoreCase))

{

IsUploadMetadata = true;

}

private const char CacheKeySeparator = '\x1F'; // Unit Separator - non-printable ASCII character

// Null/DBNull marker, delimited by the Unit Separator (the same byte used between params). Avoids

// \x00, which is hostile across backends (Redis keys, HybridCache key validation, log collectors).

// Collision-free: a real value can never contain \x1F, so it can never produce this marker.

private const string CacheKeyNull = "\x1FNULL\x1F";

internal string GetCacheStringValue()

{

if (Value is null || Value == DBNull.Value)

{

return CacheKeyNull;

}

// Prefer original string value from request for consistency

if (OriginalStringValue is not null)

{

return OriginalStringValue;

}

// Fallback for internally-set values (user claims, IP address, etc.)

if (TypeDescriptor.IsArray)

{

// Arrays can be stored as List<object?> (from query string parsing) or object[]

IList<object?>? list = Value as IList<object?>;

if (list is null || list.Count == 0)

{

return "[]";

}

var sb = new StringBuilder();

sb.Append('[');

for (int i = 0; i < list.Count; i++)

{

if (i > 0)

{

sb.Append(CacheKeySeparator);

}

sb.Append(list[i]?.ToString() ?? CacheKeyNull);

}

sb.Append(']');

return sb.ToString();

}

return Value.ToString() ?? CacheKeyNull;

}

internal static char GetCacheKeySeparator() => CacheKeySeparator;

#pragma warning disable CS8618 // Non-nullable field must contain a non-null value when exiting constructor. Consider adding the 'required' modifier or declaring as nullable.

#pragma warning disable CS8603 // Possible null reference return.

public NpgsqlRestParameter NpgsqlRestParameterMemberwiseClone() => MemberwiseClone() as NpgsqlRestParameter;

private NpgsqlRestParameter() { }

#pragma warning restore CS8603 // Possible null reference return.

#pragma warning restore CS8618 // Non-nullable field must contain a non-null value when exiting constructor. Consider adding the 'required' modifier or declaring as nullable.

private static readonly NpgsqlRestParameter TextParam = new()

{

NpgsqlDbType = NpgsqlTypes.NpgsqlDbType.Text,

Value = DBNull.Value,

};

public static NpgsqlParameter CreateParamWithType(NpgsqlTypes.NpgsqlDbType type)

{

var result = TextParam.NpgsqlRestParameterMemberwiseClone();

result.NpgsqlDbType = type;

return result;

}

public static NpgsqlParameter CreateTextParam(object? value)

{

var result = TextParam.NpgsqlRestParameterMemberwiseClone();

if (value is null)

{

return result;

}

result.Value = value;

return result;

}

Sunbelt Computer Software

PL/B Language Development and Support

Navigation Menu

Uh oh!

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

Uh oh!

FilesExpand file tree

NpgsqlRestParameter.cs

Latest commit

History

NpgsqlRestParameter.cs

File metadata and controls