PostgreSQL Source Code git master
Macros | Functions
testlo64.c File Reference
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include "libpq-fe.h"
#include "libpq/libpq-fs.h"
Include dependency graph for testlo64.c:

Go to the source code of this file.

Macros

#define  BUFSIZE   1024
 

Functions

static Oid  importFile (PGconn *conn, char *filename)
 
static void  pickout (PGconn *conn, Oid lobjId, int64_t start, int len)
 
static void  overwrite (PGconn *conn, Oid lobjId, int64_t start, int len)
 
static void  my_truncate (PGconn *conn, Oid lobjId, int64_t len)
 
static void  exportFile (PGconn *conn, Oid lobjId, char *filename)
 
static void  exit_nicely (PGconn *conn)
 
int  main (int argc, char **argv)
 

Macro Definition Documentation

◆ BUFSIZE

#define BUFSIZE   1024

Definition at line 27 of file testlo64.c.

Function Documentation

◆ exit_nicely()

static void exit_nicely ( PGconn *  conn )
static

Definition at line 216 of file testlo64.c.

217{
218 PQfinish(conn);
219 exit(1);
220}
void PQfinish(PGconn *conn)
Definition: fe-connect.c:5305
PGconn * conn
Definition: streamutil.c:52

References conn, and PQfinish().

Referenced by main().

◆ exportFile()

static void exportFile ( PGconn *  conn,
Oid  lobjId,
char *  filename 
)
static

Definition at line 173 of file testlo64.c.

174{
175 int lobj_fd;
176 char buf[BUFSIZE];
177 int nbytes,
178 tmp;
179 int fd;
180
181 /*
182 * open the large object
183 */
184 lobj_fd = lo_open(conn, lobjId, INV_READ);
185 if (lobj_fd < 0)
186 fprintf(stderr, "cannot open large object %u", lobjId);
187
188 /*
189 * open the file to be written to
190 */
191 fd = open(filename, O_CREAT | O_WRONLY | O_TRUNC, 0666);
192 if (fd < 0)
193 { /* error */
194 fprintf(stderr, "cannot open unix file\"%s\"",
195 filename);
196 }
197
198 /*
199 * read in from the inversion file and write to the Unix file
200 */
201 while ((nbytes = lo_read(conn, lobj_fd, buf, BUFSIZE)) > 0)
202 {
203 tmp = write(fd, buf, nbytes);
204 if (tmp < nbytes)
205 {
206 fprintf(stderr, "error while writing \"%s\"",
207 filename);
208 }
209 }
210
211 lo_close(conn, lobj_fd);
212 close(fd);
213}
int lo_read(int fd, char *buf, int len)
Definition: be-fsstubs.c:154
#define fprintf(file, fmt, msg)
Definition: cubescan.l:21
int lo_close(PGconn *conn, int fd)
Definition: fe-lobj.c:96
int lo_open(PGconn *conn, Oid lobjId, int mode)
Definition: fe-lobj.c:57
#define close(a)
Definition: win32.h:12
#define write(a, b, c)
Definition: win32.h:14
#define INV_READ
Definition: libpq-fs.h:22
static char * filename
Definition: pg_dumpall.c:120
static char * buf
Definition: pg_test_fsync.c:72
static int fd(const char *x, int i)
Definition: preproc-init.c:105
#define BUFSIZE
Definition: testlo64.c:27

References buf, BUFSIZE, close, conn, fd(), filename, fprintf, INV_READ, lo_close(), lo_open(), lo_read(), and write.

◆ importFile()

static Oid importFile ( PGconn *  conn,
char *  filename 
)
static

Definition at line 35 of file testlo64.c.

36{
37 Oid lobjId;
38 int lobj_fd;
39 char buf[BUFSIZE];
40 int nbytes,
41 tmp;
42 int fd;
43
44 /*
45 * open the file to be read in
46 */
47 fd = open(filename, O_RDONLY, 0666);
48 if (fd < 0)
49 { /* error */
50 fprintf(stderr, "cannot open unix file\"%s\"\n", filename);
51 }
52
53 /*
54 * create the large object
55 */
56 lobjId = lo_creat(conn, INV_READ | INV_WRITE);
57 if (lobjId == 0)
58 fprintf(stderr, "cannot create large object");
59
60 lobj_fd = lo_open(conn, lobjId, INV_WRITE);
61
62 /*
63 * read in from the Unix file and write to the inversion file
64 */
65 while ((nbytes = read(fd, buf, BUFSIZE)) > 0)
66 {
67 tmp = lo_write(conn, lobj_fd, buf, nbytes);
68 if (tmp < nbytes)
69 fprintf(stderr, "error while reading \"%s\"", filename);
70 }
71
72 close(fd);
73 lo_close(conn, lobj_fd);
74
75 return lobjId;
76}
int lo_write(int fd, const char *buf, int len)
Definition: be-fsstubs.c:182
Oid lo_creat(PGconn *conn, int mode)
Definition: fe-lobj.c:438
#define read(a, b, c)
Definition: win32.h:13
#define INV_WRITE
Definition: libpq-fs.h:21
unsigned int Oid
Definition: postgres_ext.h:32

References buf, BUFSIZE, close, conn, fd(), filename, fprintf, INV_READ, INV_WRITE, lo_close(), lo_creat(), lo_open(), lo_write(), and read.

◆ main()

int main ( int  argc,
char **  argv 
)

Definition at line 223 of file testlo64.c.

224{
225 char *in_filename,
226 *out_filename,
227 *out_filename2;
228 char *database;
229 Oid lobjOid;
230 PGconn *conn;
231 PGresult *res;
232
233 if (argc != 5)
234 {
235 fprintf(stderr, "Usage: %s database_name in_filename out_filename out_filename2\n",
236 argv[0]);
237 exit(1);
238 }
239
240 database = argv[1];
241 in_filename = argv[2];
242 out_filename = argv[3];
243 out_filename2 = argv[4];
244
245 /*
246 * set up the connection
247 */
248 conn = PQsetdb(NULL, NULL, NULL, NULL, database);
249
250 /* check to see that the backend connection was successfully made */
252 {
253 fprintf(stderr, "%s", PQerrorMessage(conn));
255 }
256
257 /* Set always-secure search path, so malicious users can't take control. */
258 res = PQexec(conn,
259 "SELECT pg_catalog.set_config('search_path', '', false)");
261 {
262 fprintf(stderr, "SET failed: %s", PQerrorMessage(conn));
263 PQclear(res);
265 }
266 PQclear(res);
267
268 res = PQexec(conn, "begin");
269 PQclear(res);
270 printf("importing file \"%s\" ...\n", in_filename);
271/* lobjOid = importFile(conn, in_filename); */
272 lobjOid = lo_import(conn, in_filename);
273 if (lobjOid == 0)
274 fprintf(stderr, "%s\n", PQerrorMessage(conn));
275 else
276 {
277 printf("\tas large object %u.\n", lobjOid);
278
279 printf("picking out bytes 4294967000-4294968000 of the large object\n");
280 pickout(conn, lobjOid, 4294967000U, 1000);
281
282 printf("overwriting bytes 4294967000-4294968000 of the large object with X's\n");
283 overwrite(conn, lobjOid, 4294967000U, 1000);
284
285 printf("exporting large object to file \"%s\" ...\n", out_filename);
286/* exportFile(conn, lobjOid, out_filename); */
287 if (lo_export(conn, lobjOid, out_filename) < 0)
288 fprintf(stderr, "%s\n", PQerrorMessage(conn));
289
290 printf("truncating to 3294968000 bytes\n");
291 my_truncate(conn, lobjOid, 3294968000U);
292
293 printf("exporting truncated large object to file \"%s\" ...\n", out_filename2);
294 if (lo_export(conn, lobjOid, out_filename2) < 0)
295 fprintf(stderr, "%s\n", PQerrorMessage(conn));
296 }
297
298 res = PQexec(conn, "end");
299 PQclear(res);
300 PQfinish(conn);
301 return 0;
302}
ConnStatusType PQstatus(const PGconn *conn)
Definition: fe-connect.c:7616
char * PQerrorMessage(const PGconn *conn)
Definition: fe-connect.c:7679
PGresult * PQexec(PGconn *conn, const char *query)
Definition: fe-exec.c:2273
Oid lo_import(PGconn *conn, const char *filename)
Definition: fe-lobj.c:626
int lo_export(PGconn *conn, Oid lobjId, const char *filename)
Definition: fe-lobj.c:748
#define PQclear
Definition: libpq-be-fe.h:245
#define PQresultStatus
Definition: libpq-be-fe.h:247
@ CONNECTION_OK
Definition: libpq-fe.h:84
@ PGRES_TUPLES_OK
Definition: libpq-fe.h:128
#define PQsetdb(M_PGHOST, M_PGPORT, M_PGOPT, M_PGTTY, M_DBNAME)
Definition: libpq-fe.h:339
#define printf(...)
Definition: port.h:245
Definition: libpq-int.h:372
Definition: libpq-int.h:165
static void exit_nicely(PGconn *conn)
Definition: testlo64.c:216
static void overwrite(PGconn *conn, Oid lobjId, int64_t start, int len)
Definition: testlo64.c:114
static void my_truncate(PGconn *conn, Oid lobjId, int64_t len)
Definition: testlo64.c:152
static void pickout(PGconn *conn, Oid lobjId, int64_t start, int len)
Definition: testlo64.c:79

References conn, CONNECTION_OK, exit_nicely(), fprintf, lo_export(), lo_import(), my_truncate(), overwrite(), PGRES_TUPLES_OK, pickout(), PQclear, PQerrorMessage(), PQexec(), PQfinish(), PQresultStatus, PQsetdb, PQstatus(), and printf.

◆ my_truncate()

static void my_truncate ( PGconn *  conn,
Oid  lobjId,
int64_t  len 
)
static

Definition at line 152 of file testlo64.c.

153{
154 int lobj_fd;
155
156 lobj_fd = lo_open(conn, lobjId, INV_READ | INV_WRITE);
157 if (lobj_fd < 0)
158 fprintf(stderr, "cannot open large object %u", lobjId);
159
160 if (lo_truncate64(conn, lobj_fd, len) < 0)
161 fprintf(stderr, "error in lo_truncate64: %s", PQerrorMessage(conn));
162
163 lo_close(conn, lobj_fd);
164}
int lo_truncate64(PGconn *conn, int fd, int64_t len)
Definition: fe-lobj.c:195
const void size_t len

References conn, fprintf, INV_READ, INV_WRITE, len, lo_close(), lo_open(), lo_truncate64(), and PQerrorMessage().

Referenced by main().

◆ overwrite()

static void overwrite ( PGconn *  conn,
Oid  lobjId,
int64_t  start,
int  len 
)
static

Definition at line 114 of file testlo64.c.

115{
116 int lobj_fd;
117 char *buf;
118 int nbytes;
119 int nwritten;
120 int i;
121
122 lobj_fd = lo_open(conn, lobjId, INV_WRITE);
123 if (lobj_fd < 0)
124 fprintf(stderr, "cannot open large object %u", lobjId);
125
126 if (lo_lseek64(conn, lobj_fd, start, SEEK_SET) < 0)
127 fprintf(stderr, "error in lo_lseek64: %s", PQerrorMessage(conn));
128
129 buf = malloc(len + 1);
130
131 for (i = 0; i < len; i++)
132 buf[i] = 'X';
133 buf[i] = '0円';
134
135 nwritten = 0;
136 while (len - nwritten > 0)
137 {
138 nbytes = lo_write(conn, lobj_fd, buf + nwritten, len - nwritten);
139 nwritten += nbytes;
140 if (nbytes <= 0)
141 {
142 fprintf(stderr, "\nWRITE FAILED!\n");
143 break;
144 }
145 }
146 free(buf);
147 fprintf(stderr, "\n");
148 lo_close(conn, lobj_fd);
149}
int64_t lo_lseek64(PGconn *conn, int fd, int64_t offset, int whence)
Definition: fe-lobj.c:385
return str start
#define free(a)
Definition: header.h:65
#define malloc(a)
Definition: header.h:50
i
int i
Definition: isn.c:77

References buf, conn, fprintf, free, i, INV_WRITE, len, lo_close(), lo_lseek64(), lo_open(), lo_write(), malloc, PQerrorMessage(), and start.

Referenced by main().

◆ pickout()

static void pickout ( PGconn *  conn,
Oid  lobjId,
int64_t  start,
int  len 
)
static

Definition at line 79 of file testlo64.c.

80{
81 int lobj_fd;
82 char *buf;
83 int nbytes;
84 int nread;
85
86 lobj_fd = lo_open(conn, lobjId, INV_READ);
87 if (lobj_fd < 0)
88 fprintf(stderr, "cannot open large object %u", lobjId);
89
90 if (lo_lseek64(conn, lobj_fd, start, SEEK_SET) < 0)
91 fprintf(stderr, "error in lo_lseek64: %s", PQerrorMessage(conn));
92
93 if (lo_tell64(conn, lobj_fd) != start)
94 fprintf(stderr, "error in lo_tell64: %s", PQerrorMessage(conn));
95
96 buf = malloc(len + 1);
97
98 nread = 0;
99 while (len - nread > 0)
100 {
101 nbytes = lo_read(conn, lobj_fd, buf, len - nread);
102 buf[nbytes] = '0円';
103 fprintf(stderr, ">>> %s", buf);
104 nread += nbytes;
105 if (nbytes <= 0)
106 break; /* no more data? */
107 }
108 free(buf);
109 fprintf(stderr, "\n");
110 lo_close(conn, lobj_fd);
111}
int64_t lo_tell64(PGconn *conn, int fd)
Definition: fe-lobj.c:548

References buf, conn, fprintf, free, INV_READ, len, lo_close(), lo_lseek64(), lo_open(), lo_read(), lo_tell64(), malloc, PQerrorMessage(), and start.

Referenced by main().

AltStyle によって変換されたページ (->オリジナル) /