tandem
/
PostgreSQL-IBAN
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

Fully working IBAN extension

This commit is contained in:
yorickdewid 2016-03-05 23:17:38 +01:00
parent d654e10425
commit 0a17c9f450
2 changed files with 65 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Makefile
View File

@ -11,7 +11,7 @@ all: iban.so
iban.so: iban.cc iban.so: iban.cc
$(CC) $(CFLAGS) -o iban.o -c iban.cc $(INCLUDEDIRS) $(CC) $(CFLAGS) -o iban.o -c iban.cc $(INCLUDEDIRS)
$(CC) -Wl,--gc-sections -shared -o iban.so iban.o $(LIBDIR) -lpq -lm -lstdc++ $(CC) -Wl,--gc-sections -shared -o iban.so iban.o $(LIBDIR) -lpq -lm -lstdc++
# cp iban.so $(LIBINSTALL) cp iban.so $(LIBINSTALL)
clean: clean:
@$(RM) -rf *.o @$(RM) -rf *.o

80
iban.cc
View File

@ -1,9 +1,29 @@
/*
* IBAN: PostgreSQL functions and datatype
* Copyright © 2016 Yorick de Wid <yorick17@outlook.com>
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see
* <http://www.gnu.org/licenses/>.
*
*********************************************************************/
extern "C" extern "C"
{ {
#include <postgres.h> #include <postgres.h>
#include <string.h> #include <string.h>
#include <fmgr.h> #include <fmgr.h>
#include <utils/builtins.h> #include <utils/builtins.h>
#include <libpq/pqformat.h>
} }
#include <iostream> #include <iostream>
@ -23,11 +43,11 @@ extern "C"
extern Datum ibanin(PG_FUNCTION_ARGS); extern Datum ibanin(PG_FUNCTION_ARGS);
extern Datum ibanout(PG_FUNCTION_ARGS); extern Datum ibanout(PG_FUNCTION_ARGS);
extern Datum ibanrecv(PG_FUNCTION_ARGS);
extern Datum ibansend(PG_FUNCTION_ARGS);
extern Datum iban_validate(PG_FUNCTION_ARGS); extern Datum iban_validate(PG_FUNCTION_ARGS);
typedef struct Iban { typedef char Iban;
char account[34];
} Iban;
} }
void Validate::addSpecification(Specification* specPtr) { void Validate::addSpecification(Specification* specPtr) {
@ -449,25 +469,23 @@ bool account_validate_str(char *iban) {
extern "C" extern "C"
{ {
/* Convert type input */ /**************************************************************************
* Input/Output functions
**************************************************************************/
PG_FUNCTION_INFO_V1(ibanin); PG_FUNCTION_INFO_V1(ibanin);
Datum Datum
ibanin(PG_FUNCTION_ARGS) { ibanin(PG_FUNCTION_ARGS) {
char *str = PG_GETARG_CSTRING(0); char *inputText = PG_GETARG_CSTRING(0);
Iban *result;
if (!account_validate_str(str)) if (!account_validate_str(inputText))
ereport(ERROR, ereport(ERROR,
(errcode(ERRCODE_INVALID_TEXT_REPRESENTATION), (errcode(ERRCODE_INVALID_TEXT_REPRESENTATION),
errmsg("invalid iban format for value: \"%s\"", errmsg("invalid iban format for value: \"%s\"",
str))); inputText)));
result = (Iban *) palloc0(sizeof(Iban)); PG_RETURN_TEXT_P(cstring_to_text(inputText));
memcpy(result->account, str, strlen(str));
PG_RETURN_POINTER(result);
} }
/* Convert type output */ /* Convert type output */
@ -476,11 +494,41 @@ extern "C"
Datum Datum
ibanout(PG_FUNCTION_ARGS) { ibanout(PG_FUNCTION_ARGS) {
Iban *iban = (Iban *) PG_GETARG_POINTER(0); Iban *iban = (Iban *) PG_GETARG_DATUM(0);
char *result;
result = psprintf("%s", iban->account); PG_RETURN_CSTRING(TextDatumGetCString(iban));
PG_RETURN_CSTRING(result); }
/**************************************************************************
* Binary Input/Output functions
**************************************************************************/
PG_FUNCTION_INFO_V1(ibanrecv);
Datum
ibanrecv(PG_FUNCTION_ARGS) {
StringInfo buf = (StringInfo) PG_GETARG_POINTER(0);
text *result;
Iban *str;
int nbytes;
str = pq_getmsgtext(buf, buf->len - buf->cursor, &nbytes);
result = cstring_to_text_with_len(str, nbytes);
pfree(str);
PG_RETURN_TEXT_P(result);
}
PG_FUNCTION_INFO_V1(ibansend);
Datum
ibansend(PG_FUNCTION_ARGS) {
text *t = PG_GETARG_TEXT_PP(0);
StringInfoData buf;
pq_begintypsend(&buf);
pq_sendtext(&buf, VARDATA_ANY(t), VARSIZE_ANY_EXHDR(t));
PG_RETURN_BYTEA_P(pq_endtypsend(&buf));
} }
/* Manually verify a text */ /* Manually verify a text */